Update app.py

app.py

@@ -1,26 +1,30 @@
 import torch
 import gradio as gr
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import Response
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import scipy.io.wavfile
 import io
 import numpy as np
 
-# Initialize FastAPI
-app = FastAPI()
-
-# Model Setup
+# 1. Resource-Optimized Model Loading
 MODEL_ID = "facebook/musicgen-small"
+device = "cpu"
+
+print(f"Loading model {MODEL_ID}...")
 processor = AutoProcessor.from_pretrained(MODEL_ID)
-model = MusicgenForConditionalGeneration.from_pretrained(MODEL_ID)
+model = MusicgenForConditionalGeneration.from_pretrained(MODEL_ID, torch_dtype=torch.float32)
+model.to(device)
+print("Model loaded successfully.")
 
 def generate_core(prompt, duration):
-    """Shared generation logic for both Gradio and API"""
-    if duration > 30:
-        duration = 30
+    if not prompt:
+        return None
     
-    inputs = processor(text=[prompt], padding=True, return_tensors="pt")
+    duration = min(int(duration), 30)
+    inputs = processor(text=[prompt], padding=True, return_tensors="pt").to(device)
+    
+    # 50 tokens = 1 second
     max_tokens = int(duration * 50)
     
     with torch.no_grad():
@@ -30,9 +34,11 @@ def generate_core(prompt, duration):
     audio_data = audio_values[0, 0].cpu().numpy()
     return sampling_rate, audio_data
 
-# --- API Endpoint for n8n ---
+# 2. FastAPI Engine
+app = FastAPI(title="MusicGen Automation API")
+
 @app.post("/generate")
-async def api_generate(prompt: str, duration: int = 10):
+async def api_generate(prompt: str = Query(...), duration: int = Query(10)):
     try:
         sr, audio = generate_core(prompt, duration)
         byte_io = io.BytesIO()
@@ -41,25 +47,18 @@ async def api_generate(prompt: str, duration: int = 10):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# --- Gradio UI Interface ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🎵 MusicGen Automation Hub")
-    gr.Markdown("Use this UI for manual testing or hit the `/generate` endpoint for n8n.")
-    
+# 3. Gradio Interface
+with gr.Blocks(theme=gr.themes.Default()) as demo:
+    gr.Markdown("# 🎵 MusicGen Automation Hub")
     with gr.Row():
         with gr.Column():
-            prompt_input = gr.Textbox(label="Text Prompt", placeholder="Cyberpunk synthwave with heavy bass...")
-            duration_slider = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Duration (Seconds)")
-            generate_btn = gr.Button("Generate Music", variant="primary")
-        
+            p_in = gr.Textbox(label="Prompt", placeholder="Upbeat synthwave...")
+            d_in = gr.Slider(1, 30, value=10, label="Duration (sec)")
+            run_btn = gr.Button("Generate", variant="primary")
         with gr.Column():
-            audio_output = gr.Audio(label="Generated Audio", type="numpy")
-
-    generate_btn.click(
-        fn=generate_core,
-        inputs=[prompt_input, duration_slider],
-        outputs=audio_output
-    )
+            a_out = gr.Audio(label="Output")
+    
+    run_btn.click(generate_core, [p_in, d_in], a_out)
 
-# Mount Gradio into FastAPI
-app = gr.mount_gradio_app(app, demo, path="/")
+# Mount Gradio and KILL the API schema generator that causes the 500 error
+app = gr.mount_gradio_app(app, demo, path="/", show_api=False)