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
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,
)

View File

@@ -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,