From bd82b65f5d081c8519a777836fb5419dca609c77 Mon Sep 17 00:00:00 2001
From: rajeevrajeshuni <rajeev@rajeshuni.com>
Date: Sat, 29 Nov 2025 22:02:52 +0530
Subject: [PATCH] UI finetuning

---
 cognee/cli/tui/add_screen.py     |  4 ++--
 cognee/cli/tui/cognify_screen.py |  6 +++---
 cognee/cli/tui/config_screen.py  |  2 +-
 cognee/cli/tui/delete_screen.py  |  3 ++-
 cognee/cli/tui/home_screen.py    |  2 +-
 cognee/cli/tui/search_screen.py  | 22 +++++++++++++++++++---
 6 files changed, 28 insertions(+), 11 deletions(-)

diff --git a/cognee/cli/tui/add_screen.py b/cognee/cli/tui/add_screen.py
index 60136ecf1..d647c5edf 100644
--- a/cognee/cli/tui/add_screen.py
+++ b/cognee/cli/tui/add_screen.py
@@ -47,7 +47,7 @@ class AddTUIScreen(BaseTUIScreen):
     .field-label {
         color: $text-muted;
         margin-top: 1;
-        margin-bottom: 0;
+        margin-bottom: 1;
     }
 
     Input {
@@ -68,7 +68,7 @@ class AddTUIScreen(BaseTUIScreen):
 
     #add-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
diff --git a/cognee/cli/tui/cognify_screen.py b/cognee/cli/tui/cognify_screen.py
index a13e14c71..24eae1e6b 100644
--- a/cognee/cli/tui/cognify_screen.py
+++ b/cognee/cli/tui/cognify_screen.py
@@ -47,7 +47,7 @@ class CognifyTUIScreen(BaseTUIScreen):
     .field-label {
         color: $text-muted;
         margin-top: 1;
-        margin-bottom: 0;
+        margin-bottom: 1;
     }
 
     Input {
@@ -78,7 +78,7 @@ class CognifyTUIScreen(BaseTUIScreen):
 
     #cognify-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
@@ -92,7 +92,7 @@ class CognifyTUIScreen(BaseTUIScreen):
 
     def compose_content(self) -> ComposeResult:
         with Container(id="cognify-container"):
-            yield Label("Cognify Data", id="form-title")
+            yield Label("⚡ Cognify Data", id="form-title")
             with Vertical(id="cognify-form"):
                 yield Label("Dataset Name (optional, leave empty for all):", classes="field-label")
                 yield Input(
diff --git a/cognee/cli/tui/config_screen.py b/cognee/cli/tui/config_screen.py
index a0d5064b8..f8a17c91f 100644
--- a/cognee/cli/tui/config_screen.py
+++ b/cognee/cli/tui/config_screen.py
@@ -129,7 +129,7 @@ class ConfigTUIScreen(BaseTUIScreen):
 
     #config-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
diff --git a/cognee/cli/tui/delete_screen.py b/cognee/cli/tui/delete_screen.py
index a2b2386b9..bfd204147 100644
--- a/cognee/cli/tui/delete_screen.py
+++ b/cognee/cli/tui/delete_screen.py
@@ -42,6 +42,7 @@ class DeleteTUIScreen(BaseTUIScreen):
         text-style: bold;
         color: $accent;
         margin-bottom: 2;
+        width: 100%;
     }
 
     .input-group {
@@ -77,7 +78,7 @@ class DeleteTUIScreen(BaseTUIScreen):
 
     #delete-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
diff --git a/cognee/cli/tui/home_screen.py b/cognee/cli/tui/home_screen.py
index 56a101f07..77a3bbc0c 100644
--- a/cognee/cli/tui/home_screen.py
+++ b/cognee/cli/tui/home_screen.py
@@ -147,7 +147,7 @@ class HomeScreen(BaseTUIScreen):
 
     #home-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
diff --git a/cognee/cli/tui/search_screen.py b/cognee/cli/tui/search_screen.py
index 87ee1dd1f..4bfc14341 100644
--- a/cognee/cli/tui/search_screen.py
+++ b/cognee/cli/tui/search_screen.py
@@ -20,6 +20,15 @@ class SearchTUIScreen(BaseTUIScreen):
     SearchTUIScreen {
         background: $surface;
     }
+    
+    #form-title {
+        text-align: center;
+        text-style: bold;
+        color: $accent;
+        margin-bottom: 2;
+        width: 100%;
+    }
+
 
     #search-container {
         height: 100%;
@@ -58,10 +67,16 @@ class SearchTUIScreen(BaseTUIScreen):
         height: 1fr;
         overflow-y: auto;
     }
+    
+    .field-label {
+        color: $text-muted;
+        margin-top: 1;
+        margin-bottom: 1;
+    }
 
     #search-footer {
         dock: bottom;
-        height: 3;
+        padding: 1 0;
         background: $boost;
         color: $text-muted;
         content-align: center middle;
@@ -75,10 +90,11 @@ class SearchTUIScreen(BaseTUIScreen):
 
     def compose_content(self) -> ComposeResult:
         with Container(id="search-container"):
+            yield Label("🔍 Search Data", id="form-title")
             with Vertical(id="search-form"):
-                yield Label("Query:")
+                yield Label("Query:", classes="field-label")
                 yield Input(placeholder="Enter your search query...", id="query-input")
-                yield Label("Search Type:")
+                yield Label("Search Type:", classes="field-label")
                 yield Select(
                     [
                         ("Graph Completion (Recommended)", "GRAPH_COMPLETION"),