diff --git a/src/main.py b/src/main.py index fac49be..41a2471 100644 --- a/src/main.py +++ b/src/main.py @@ -44,7 +44,7 @@ def create_layout(current_route='/'): @ui.page('/') async def index_page(): create_layout('/') - DashboardPage(system_monitor, gpu_monitor) + DashboardPage(system_monitor, gpu_monitor, ollama_monitor) @ui.page('/system') diff --git a/src/pages/dashboard.py b/src/pages/dashboard.py index fa7d6e6..0b79fd8 100644 --- a/src/pages/dashboard.py +++ b/src/pages/dashboard.py @@ -1,7 +1,7 @@ from typing import Literal from nicegui import ui from components.circular_progress import MetricCircle, LargeMetricCircle, ColorfulMetricCard, MetricCircleAdv -from utils import SystemMonitor, GPUMonitor +from utils import SystemMonitor, GPUMonitor, OllamaMonitor """ with ui.element('div').classes('main-content w-full'): @@ -49,7 +49,7 @@ with ui.element('div').classes('main-content w-full'): class DashboardPage(ui.column): - def __init__(self, system_monitor: SystemMonitor, gpu_monitor: GPUMonitor, *, wrap: bool = False, align_items: None | Literal['start'] | Literal['end'] | Literal['center'] | Literal['baseline'] | Literal['stretch'] = None) -> None: + def __init__(self, system_monitor: SystemMonitor, gpu_monitor: GPUMonitor, ollama_monitor: OllamaMonitor, *, wrap: bool = False, align_items: None | Literal['start'] | Literal['end'] | Literal['center'] | Literal['baseline'] | Literal['stretch'] = None) -> None: super().__init__(wrap=wrap, align_items=align_items) self.system_monitor = system_monitor self.gpu_monitor = gpu_monitor