From 406b783a1aafb28378a8c266229b0b43772460d0 Mon Sep 17 00:00:00 2001
From: phact <estevezsebastian@gmail.com>
Date: Fri, 21 Nov 2025 17:14:05 -0500
Subject: [PATCH] conditional buttons and copy

---
 src/tui/screens/config.py  | 14 ++++++++++----
 src/tui/screens/welcome.py | 32 ++++++++++++++++++++++++--------
 2 files changed, 34 insertions(+), 12 deletions(-)

diff --git a/src/tui/screens/config.py b/src/tui/screens/config.py
index df8d6e98..51662964 100644
--- a/src/tui/screens/config.py
+++ b/src/tui/screens/config.py
@@ -145,6 +145,9 @@ class ConfigScreen(Screen):
         self.mode = mode  # "no_auth" or "full"
         self.env_manager = EnvManager()
         self.inputs = {}
+        
+        # Check if .env file exists
+        self.has_env_file = self.env_manager.env_file.exists()
 
         # Load existing config if available
         self.env_manager.load_existing_env()
@@ -156,12 +159,15 @@ class ConfigScreen(Screen):
             with ScrollableContainer(id="config-scroll"):
                 with Vertical(id="config-form"):
                     yield from self._create_all_fields()
-            yield Horizontal(
+            # Create button row - conditionally include Back button
+            buttons = [
                 Button("Generate Passwords", variant="default", id="generate-btn"),
                 Button("Save Configuration", variant="success", id="save-btn"),
-                Button("Back", variant="default", id="back-btn"),
-                classes="button-row",
-            )
+            ]
+            # Only show Back button if .env file exists
+            if self.has_env_file:
+                buttons.append(Button("Back", variant="default", id="back-btn"))
+            yield Horizontal(*buttons, classes="button-row")
         yield Footer()
 
     def _create_header_text(self) -> Text:
diff --git a/src/tui/screens/welcome.py b/src/tui/screens/welcome.py
index 66321010..64ad888a 100644
--- a/src/tui/screens/welcome.py
+++ b/src/tui/screens/welcome.py
@@ -22,13 +22,6 @@ class WelcomeScreen(Screen):
 
     BINDINGS = [
         ("q", "quit", "Quit"),
-        ("enter", "default_action", "Continue"),
-        ("1", "no_auth_setup", "Basic Setup"),
-        ("2", "full_setup", "Advanced Setup"),
-        ("3", "monitor", "Status"),
-        ("4", "diagnostics", "Diagnostics"),
-        ("5", "start_stop_services", "Start/Stop Services"),
-        ("6", "open_app", "Open App"),
     ]
 
     def __init__(self):
@@ -41,6 +34,9 @@ class WelcomeScreen(Screen):
         self.has_oauth_config = False
         self.default_button_id = "basic-setup-btn"
         self._state_checked = False
+        
+        # Check if .env file exists
+        self.has_env_file = self.env_manager.env_file.exists()
 
         # Load .env file if it exists
         load_dotenv()
@@ -161,6 +157,23 @@ class WelcomeScreen(Screen):
 
         buttons = []
 
+        # If no .env file exists, only show setup buttons
+        if not self.has_env_file:
+            if has_oauth:
+                # If OAuth is configured, only show advanced setup
+                buttons.append(
+                    Button("Advanced Setup", variant="success", id="advanced-setup-btn")
+                )
+            else:
+                # If no OAuth, show both options with basic as primary
+                buttons.append(
+                    Button("Basic Setup", variant="success", id="basic-setup-btn")
+                )
+                buttons.append(
+                    Button("Advanced Setup", variant="default", id="advanced-setup-btn")
+                )
+            return Horizontal(*buttons, classes="button-row")
+
         # Check if all services (native + container) are running
         all_services_running = self.services_running and self.docling_running
 
@@ -189,7 +202,7 @@ class WelcomeScreen(Screen):
                 )
 
             buttons.append(
-                Button("Start All Services", variant="primary", id="start-all-services-btn")
+                Button("Start OpenRAG", variant="primary", id="start-all-services-btn")
             )
 
         # Always show status option
@@ -253,6 +266,9 @@ class WelcomeScreen(Screen):
 
     async def on_screen_resume(self) -> None:
         """Called when returning from another screen (e.g., config screen)."""
+        # Check if .env file exists (may have been created)
+        self.has_env_file = self.env_manager.env_file.exists()
+        
         # Reload environment variables
         load_dotenv(override=True)