From 1779d3898d82843159ae48db2cff64ff3ee8d5cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Wed, 26 Nov 2025 23:35:15 +0100
Subject: [PATCH] Fix service init params and simplify VRAM monitor
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Fix GenerationService params: registry, database
- Simplify VRAM display to basic Markdown for now

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 main.py       |  3 ++-
 src/ui/app.py | 11 ++++-------
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/main.py b/main.py
index 0b00bb7..90aad08 100644
--- a/main.py
+++ b/main.py
@@ -59,8 +59,9 @@ async def initialize_services():
     # Initialize services
     logger.info("Initializing services...")
     generation_service = GenerationService(
-        model_registry=model_registry,
+        registry=model_registry,
         gpu_manager=gpu_manager,
+        database=db,
         output_dir=settings.output_dir,
     )
 
diff --git a/src/ui/app.py b/src/ui/app.py
index 449ac9f..4a53aac 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -7,7 +7,6 @@ from pathlib import Path
 
 from src.ui.theme import get_custom_css
 from src.ui.state import UIState, DEFAULT_PRESETS, PROMPT_SUGGESTIONS
-from src.ui.components.vram_monitor import create_vram_monitor
 from src.ui.tabs import (
     create_dashboard_tab,
     create_musicgen_tab,
@@ -193,15 +192,13 @@ class AudioCraftApp:
         # In Gradio 6.x, theme/css moved to launch()
         # Create Blocks with no styling params
         with gr.Blocks() as app:
-            # Header with VRAM monitor
+            # Header
             with gr.Row():
                 with gr.Column(scale=4):
                     gr.Markdown("# AudioCraft Studio")
                 with gr.Column(scale=1):
-                    vram_monitor = create_vram_monitor(
-                        get_status_fn=self._get_gpu_status,
-                        update_interval=5,
-                    )
+                    # Simple GPU status display
+                    gpu_status = gr.Markdown("GPU: Loading...")
 
             # Main tabs
             with gr.Tabs() as main_tabs:
@@ -274,7 +271,7 @@ class AudioCraftApp:
 
             # Store component references
             self.components = {
-                "vram_monitor": vram_monitor,
+                "gpu_status": gpu_status,
                 "dashboard": dashboard,
                 "musicgen": musicgen,
                 "audiogen": audiogen,