diff --git a/frontend/src/app/api/queries/useDoclingHealthQuery.ts b/frontend/src/app/api/queries/useDoclingHealthQuery.ts index 88c0a39b..01441f4b 100644 --- a/frontend/src/app/api/queries/useDoclingHealthQuery.ts +++ b/frontend/src/app/api/queries/useDoclingHealthQuery.ts @@ -56,8 +56,13 @@ export const useDoclingHealthQuery = ( queryKey: ["docling-health"], queryFn: checkDoclingHealth, retry: 1, - refetchInterval: 30000, // Check every 30 seconds - staleTime: 25000, // Consider data stale after 25 seconds + refetchInterval: (query) => { + // If healthy, check every 30 seconds; otherwise check every 3 seconds + return query.state.data?.status === "healthy" ? 30000 : 3000; + }, + refetchOnWindowFocus: true, + refetchOnMount: true, + staleTime: 30000, // Consider data stale after 25 seconds ...options, }, queryClient, diff --git a/frontend/src/app/api/queries/useProviderHealthQuery.ts b/frontend/src/app/api/queries/useProviderHealthQuery.ts index d4038cfc..82ca2db2 100644 --- a/frontend/src/app/api/queries/useProviderHealthQuery.ts +++ b/frontend/src/app/api/queries/useProviderHealthQuery.ts @@ -92,6 +92,13 @@ export const useProviderHealthQuery = ( queryKey: ["provider", "health"], queryFn: checkProviderHealth, retry: false, // Don't retry health checks automatically + refetchInterval: (query) => { + // If healthy, check every 30 seconds; otherwise check every 3 seconds + return query.state.data?.status === "healthy" ? 30000 : 3000; + }, + refetchOnWindowFocus: true, + refetchOnMount: true, + staleTime: 30000, // Consider data stale after 25 seconds enabled: !!settings?.edited && options?.enabled !== false, // Only run after onboarding is complete ...options, }, diff --git a/src/tui/managers/docling_manager.py b/src/tui/managers/docling_manager.py index e58a5b1e..109cb7c1 100644 --- a/src/tui/managers/docling_manager.py +++ b/src/tui/managers/docling_manager.py @@ -34,6 +34,7 @@ class DoclingManager: # Bind to all interfaces by default (can be overridden with DOCLING_BIND_HOST env var) self._host = os.getenv('DOCLING_BIND_HOST', '0.0.0.0') self._running = False + self._starting = False self._external_process = False # PID file to track docling-serve across sessions (in current working directory) @@ -126,6 +127,7 @@ class DoclingManager: if self._process is not None and self._process.poll() is None: self._running = True self._external_process = False + self._starting = False # Clear starting flag if service is running return True # Check if we have a PID from file @@ -133,6 +135,7 @@ class DoclingManager: if pid is not None and self._is_process_running(pid): self._running = True self._external_process = True + self._starting = False # Clear starting flag if service is running return True # No running process found @@ -142,6 +145,19 @@ class DoclingManager: def get_status(self) -> Dict[str, Any]: """Get current status of docling serve.""" + # Check for starting state first + if self._starting: + display_host = "localhost" if self._host == "0.0.0.0" else self._host + return { + "status": "starting", + "port": self._port, + "host": self._host, + "endpoint": None, + "docs_url": None, + "ui_url": None, + "pid": None + } + if self.is_running(): # Try to get PID from process handle first, then from PID file pid = None @@ -196,6 +212,9 @@ class DoclingManager: except Exception as e: self._add_log_entry(f"Error checking port availability: {e}") + # Set starting flag to show "Starting" status in UI + self._starting = True + # Clear log buffer when starting self._log_buffer = [] self._add_log_entry("Starting docling serve as external process...") @@ -261,6 +280,8 @@ class DoclingManager: if result == 0: self._add_log_entry(f"Docling-serve is now listening on {self._host}:{self._port}") + # Service is now running, clear starting flag + self._starting = False break except: pass @@ -294,16 +315,24 @@ class DoclingManager: self._add_log_entry(f"Error reading final output: {e}") self._running = False + self._starting = False return False, f"Docling serve process exited immediately (code: {return_code})" + # If we get here and the process is still running but not listening yet, + # clear the starting flag anyway (it's running, just not ready) + if self._process.poll() is None: + self._starting = False + display_host = "localhost" if self._host == "0.0.0.0" else self._host return True, f"Docling serve starting on http://{display_host}:{port}" except FileNotFoundError: + self._starting = False return False, "docling-serve not available. Please install: uv add docling-serve" except Exception as e: self._running = False self._process = None + self._starting = False return False, f"Error starting docling serve: {str(e)}" def _start_output_capture(self): diff --git a/src/tui/screens/monitor.py b/src/tui/screens/monitor.py index 91df51f6..01c243c6 100644 --- a/src/tui/screens/monitor.py +++ b/src/tui/screens/monitor.py @@ -206,10 +206,21 @@ class MonitorScreen(Screen): # Add docling serve to its own table docling_status = self.docling_manager.get_status() - docling_running = docling_status["status"] == "running" - docling_status_text = "running" if docling_running else "stopped" - docling_style = "bold green" if docling_running else "bold red" - docling_port = f"{docling_status['host']}:{docling_status['port']}" if docling_running else "N/A" + docling_status_value = docling_status["status"] + docling_running = docling_status_value == "running" + docling_starting = docling_status_value == "starting" + + if docling_running: + docling_status_text = "running" + docling_style = "bold green" + elif docling_starting: + docling_status_text = "starting" + docling_style = "bold yellow" + else: + docling_status_text = "stopped" + docling_style = "bold red" + + docling_port = f"{docling_status['host']}:{docling_status['port']}" if (docling_running or docling_starting) else "N/A" docling_pid = str(docling_status.get("pid")) if docling_status.get("pid") else "N/A" if self.docling_table: @@ -375,15 +386,25 @@ class MonitorScreen(Screen): """Start docling serve.""" self.operation_in_progress = True try: - success, message = await self.docling_manager.start() + # Start the service (this sets _starting = True internally at the start) + # Create task and let it begin executing (which sets the flag) + start_task = asyncio.create_task(self.docling_manager.start()) + # Give it a tiny moment to set the _starting flag + await asyncio.sleep(0.1) + # Refresh immediately to show "Starting" status + await self._refresh_services() + # Now wait for start to complete + success, message = await start_task if success: self.notify(message, severity="information") else: self.notify(f"Failed to start docling serve: {message}", severity="error") - # Refresh the services table to show updated status + # Refresh again to show final status (running or stopped) await self._refresh_services() except Exception as e: self.notify(f"Error starting docling serve: {str(e)}", severity="error") + # Refresh on error to clear starting status + await self._refresh_services() finally: self.operation_in_progress = False @@ -646,7 +667,11 @@ class MonitorScreen(Screen): suffix = f"-{random.randint(10000, 99999)}" # Add docling serve controls - docling_running = self.docling_manager.is_running() + docling_status = self.docling_manager.get_status() + docling_status_value = docling_status["status"] + docling_running = docling_status_value == "running" + docling_starting = docling_status_value == "starting" + if docling_running: docling_controls.mount( Button("Stop", variant="error", id=f"docling-stop-btn{suffix}") @@ -654,6 +679,11 @@ class MonitorScreen(Screen): docling_controls.mount( Button("Restart", variant="primary", id=f"docling-restart-btn{suffix}") ) + elif docling_starting: + # Show disabled button or no button when starting + start_btn = Button("Starting...", variant="warning", id=f"docling-start-btn{suffix}") + start_btn.disabled = True + docling_controls.mount(start_btn) else: docling_controls.mount( Button("Start", variant="success", id=f"docling-start-btn{suffix}")