Fix service init params and simplify VRAM monitor

- 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>
This commit is contained in:
2025-11-26 23:35:15 +01:00
parent 5080eeab08
commit 1779d3898d
2 changed files with 6 additions and 8 deletions

View File

@@ -59,8 +59,9 @@ async def initialize_services():
# Initialize services # Initialize services
logger.info("Initializing services...") logger.info("Initializing services...")
generation_service = GenerationService( generation_service = GenerationService(
model_registry=model_registry, registry=model_registry,
gpu_manager=gpu_manager, gpu_manager=gpu_manager,
database=db,
output_dir=settings.output_dir, output_dir=settings.output_dir,
) )

View File

@@ -7,7 +7,6 @@ from pathlib import Path
from src.ui.theme import get_custom_css from src.ui.theme import get_custom_css
from src.ui.state import UIState, DEFAULT_PRESETS, PROMPT_SUGGESTIONS 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 ( from src.ui.tabs import (
create_dashboard_tab, create_dashboard_tab,
create_musicgen_tab, create_musicgen_tab,
@@ -193,15 +192,13 @@ class AudioCraftApp:
# In Gradio 6.x, theme/css moved to launch() # In Gradio 6.x, theme/css moved to launch()
# Create Blocks with no styling params # Create Blocks with no styling params
with gr.Blocks() as app: with gr.Blocks() as app:
# Header with VRAM monitor # Header
with gr.Row(): with gr.Row():
with gr.Column(scale=4): with gr.Column(scale=4):
gr.Markdown("# AudioCraft Studio") gr.Markdown("# AudioCraft Studio")
with gr.Column(scale=1): with gr.Column(scale=1):
vram_monitor = create_vram_monitor( # Simple GPU status display
get_status_fn=self._get_gpu_status, gpu_status = gr.Markdown("GPU: Loading...")
update_interval=5,
)
# Main tabs # Main tabs
with gr.Tabs() as main_tabs: with gr.Tabs() as main_tabs:
@@ -274,7 +271,7 @@ class AudioCraftApp:
# Store component references # Store component references
self.components = { self.components = {
"vram_monitor": vram_monitor, "gpu_status": gpu_status,
"dashboard": dashboard, "dashboard": dashboard,
"musicgen": musicgen, "musicgen": musicgen,
"audiogen": audiogen, "audiogen": audiogen,