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

monitor actually displays and sticks across navigation

Browse source
This commit is contained in:
phact 2025-11-26 06:17:20 -05:00
parent 67e31dbbe4
commit 71eaa1c1e2
1 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/tui/screens/monitor.py
View file

@ -33,13 +33,14 @@ class MonitorScreen(Screen):
("u", "upgrade", "Upgrade"), ("u", "upgrade", "Upgrade"),
("x", "reset", "Reset"), ("x", "reset", "Reset"),
("l", "logs", "View Logs"), ("l", "logs", "View Logs"),
("g", "toggle_mode", "Toggle GPU/CPU"),
("j", "cursor_down", "Move Down"), ("j", "cursor_down", "Move Down"),
("k", "cursor_up", "Move Up"), ("k", "cursor_up", "Move Up"),
] ]
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.container_manager = ContainerManager() self._container_manager = None # Use app's shared instance
self.docling_manager = DoclingManager() self.docling_manager = DoclingManager()
self.services_table = None self.services_table = None
self.docling_table = None self.docling_table = None
@ -52,6 +53,13 @@ class MonitorScreen(Screen):
# Track which table was last selected for mutual exclusion # Track which table was last selected for mutual exclusion
self._last_selected_table = None self._last_selected_table = None
@property
def container_manager(self) -> ContainerManager:
"""Get the shared container manager from the app."""
if self._container_manager is None:
self._container_manager = self.app.container_manager
return self._container_manager
def on_unmount(self) -> None: def on_unmount(self) -> None:
"""Clean up when the screen is unmounted.""" """Clean up when the screen is unmounted."""
if hasattr(self, 'docling_manager'): if hasattr(self, 'docling_manager'):
@ -69,10 +77,10 @@ class MonitorScreen(Screen):
def _create_services_tab(self) -> ComposeResult: def _create_services_tab(self) -> ComposeResult:
"""Create the services monitoring tab.""" """Create the services monitoring tab."""
# Current mode indicator + toggle # GPU/CPU mode section
yield Static("GPU Mode", id="mode-indicator", classes="tab-header")
yield Horizontal( yield Horizontal(
Static("", id="mode-indicator"), Button("Switch to CPU Mode", id="toggle-mode-btn"),
Button("Toggle Mode", id="toggle-mode-btn"),
classes="button-row", classes="button-row",
id="mode-row", id="mode-row",
) )
@ -583,8 +591,7 @@ class MonitorScreen(Screen):
try: try:
use_gpu = getattr(self.container_manager, "use_gpu_compose", False) use_gpu = getattr(self.container_manager, "use_gpu_compose", False)
indicator = self.query_one("#mode-indicator", Static) indicator = self.query_one("#mode-indicator", Static)
mode_text = "Mode: GPU" if use_gpu else "Mode: CPU (no GPU detected)" indicator.update("GPU Mode" if use_gpu else "CPU Mode")
indicator.update(mode_text)
toggle_btn = self.query_one("#toggle-mode-btn", Button) toggle_btn = self.query_one("#toggle-mode-btn", Button)
toggle_btn.label = "Switch to CPU Mode" if use_gpu else "Switch to GPU Mode" toggle_btn.label = "Switch to CPU Mode" if use_gpu else "Switch to GPU Mode"
except Exception: except Exception: