From cdb420d0f7ba3326358f973144150980bd53f2e4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Wed, 26 Nov 2025 23:28:41 +0100
Subject: [PATCH] Fix ModelRegistry params and Gradio Blocks compatibility
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Fix ModelRegistry init params: config_path, max_cached_models, idle_timeout_minutes
- Make gr.Blocks creation compatible with different Gradio versions

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

Co-Authored-By: Claude <noreply@anthropic.com>
---
 main.py       |  5 +++--
 src/ui/app.py | 22 ++++++++++++++--------
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/main.py b/main.py
index 63bc4cf..f312c68 100644
--- a/main.py
+++ b/main.py
@@ -50,9 +50,10 @@ async def initialize_services():
     # Initialize model registry
     logger.info("Initializing model registry...")
     model_registry = ModelRegistry(
+        config_path=settings.models_config,
         gpu_manager=gpu_manager,
-        max_loaded=settings.max_loaded_models,
-        idle_timeout_seconds=settings.idle_unload_minutes * 60,
+        max_cached_models=settings.max_loaded_models,
+        idle_timeout_minutes=settings.idle_unload_minutes,
     )
 
     # Initialize services
diff --git a/src/ui/app.py b/src/ui/app.py
index 9f0d6ca..8ecc82f 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -5,7 +5,7 @@ import gradio as gr
 from typing import Any, Optional
 from pathlib import Path
 
-from src.ui.theme import create_audiocraft_theme, get_custom_css
+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 (
@@ -190,15 +190,21 @@ class AudioCraftApp:
 
     def build(self) -> gr.Blocks:
         """Build the Gradio application."""
-        theme = create_audiocraft_theme()
         css = get_custom_css()
 
-        with gr.Blocks(
-            theme=theme,
-            css=css,
-            title="AudioCraft Studio",
-            analytics_enabled=False,
-        ) as app:
+        # Create Blocks with compatible parameters
+        try:
+            # Gradio 4.x
+            app = gr.Blocks(
+                theme=gr.themes.Soft(),
+                css=css,
+                title="AudioCraft Studio",
+            )
+        except TypeError:
+            # Fallback for older Gradio versions
+            app = gr.Blocks(css=css)
+
+        with app:
             # Header with VRAM monitor
             with gr.Row():
                 with gr.Column(scale=4):