gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

final polishing for the homepage

Browse source
This commit is contained in:
rajeevrajeshuni 2025-11-29 10:35:37 +05:30
parent 99e7c206af
commit aba6619d02
1 changed files with 49 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
cognee/cli/commands/tui_command.py
View file

@ -4,6 +4,23 @@ from cognee.cli.config import DEFAULT_DOCS_URL
import cognee.cli.echo as fmt import cognee.cli.echo as fmt
from cognee.cli.exceptions import CliCommandException from cognee.cli.exceptions import CliCommandException
from cognee.version import get_cognee_version from cognee.version import get_cognee_version
from textual.app import App, ComposeResult
from textual.widgets import ListView, ListItem, Static
from textual.containers import Container, Horizontal
from textual.binding import Binding
def make_item(icon, command, description):
# Compose a ListItem that contains a Horizontal container with 3 children
return ListItem(
Horizontal(
Static(icon, classes="cmd-icon"),
Static(command, classes="cmd-name"),
Static(description, classes="cmd-desc"),
classes="cmd-row",
)
)
class TuiCommand(SupportsCliCommand): class TuiCommand(SupportsCliCommand):
@property @property
@ -24,11 +41,6 @@ class TuiCommand(SupportsCliCommand):
def execute(self, args: argparse.Namespace) -> None: def execute(self, args: argparse.Namespace) -> None:
try: try:
from textual.app import App, ComposeResult
from textual.widgets import Header, Footer, ListView, ListItem, Static
from textual.containers import Container, Vertical
from textual.binding import Binding
class CommandItem(Static): class CommandItem(Static):
"""A custom widget for command items with icon and description.""" """A custom widget for command items with icon and description."""
@ -42,8 +54,6 @@ class TuiCommand(SupportsCliCommand):
return f"{self.icon} {self.command:<12} {self.description}" return f"{self.icon} {self.command:<12} {self.description}"
class CogneeTUI(App): class CogneeTUI(App):
"""A k9s-style TUI for cognee commands."""
CSS = """ CSS = """
Screen { Screen {
background: $surface; background: $surface;
@ -75,7 +85,7 @@ class TuiCommand(SupportsCliCommand):
width: auto; width: auto;
color: $accent; color: $accent;
text-style: bold; text-style: bold;
padding: 0 3; padding: 0 10;
border: solid $accent; border: solid $accent;
margin-bottom: 2; margin-bottom: 2;
} }
@ -90,7 +100,7 @@ class TuiCommand(SupportsCliCommand):
height: auto; height: auto;
background: $surface; background: $surface;
border: none; border: none;
padding: 0 2; padding: 0 0;
} }
ListItem { ListItem {
@ -118,6 +128,30 @@ class TuiCommand(SupportsCliCommand):
content-align: center middle; content-align: center middle;
border: solid $primary; border: solid $primary;
} }
.cmd-row {
width: 100%;
height: auto;
align-horizontal: left;
padding: 0 1;
}
.cmd-icon {
width: 4;
text-align: center;
}
.cmd-name {
width: 14;
padding-left: 1;
}
.cmd-desc {
width: 1fr;
overflow: auto;
padding-left: 1;
}
""" """
BINDINGS = [ BINDINGS = [
@ -141,11 +175,11 @@ class TuiCommand(SupportsCliCommand):
with Container(id="title-wrapper"): with Container(id="title-wrapper"):
yield Static("Select Command", id="title") yield Static("Select Command", id="title")
yield ListView( yield ListView(
ListItem(CommandItem("📥", "add", "Add data to cognee")), make_item("📥", "add", "Add data to cognee"),
ListItem(CommandItem("🔍", "search", "Search data in cognee")), make_item("🔍", "search", "Search data in cognee"),
ListItem(CommandItem("⚡", "cognify", "Process data in cognee")), make_item("⚡", "cognify", "Process data in cognee"),
ListItem(CommandItem("🗑️", "delete", "Delete data from cognee")), make_item("🗑️", "delete", "Delete data from cognee"),
ListItem(CommandItem("⚙️", "config", "Configure cognee settings")), make_item("⚙️", "config", "Configure cognee settings"),
) )
yield Static( yield Static(
@ -163,6 +197,7 @@ class TuiCommand(SupportsCliCommand):
def _apply_highlight(self) -> None: def _apply_highlight(self) -> None:
lv = self.lv lv = self.lv
children = list(lv.children) children = list(lv.children)
self.lv.index = self.current_index
for idx, item in enumerate(children): for idx, item in enumerate(children):
if idx == self.current_index: if idx == self.current_index:
item.add_class("highlighted") item.add_class("highlighted")