conditional buttons and copy

This commit is contained in:
phact 2025-11-21 17:14:05 -05:00
parent 9e4502d07b
commit 406b783a1a
2 changed files with 34 additions and 12 deletions

View file

@ -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:

View file

@ -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)