Create app.py

app.py

@@ -0,0 +1,65 @@
+import torch
+import gradio as gr
+from fastapi import FastAPI, HTTPException
+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
+MODEL_ID = "facebook/musicgen-small"
+processor = AutoProcessor.from_pretrained(MODEL_ID)
+model = MusicgenForConditionalGeneration.from_pretrained(MODEL_ID)
+
+def generate_core(prompt, duration):
+    """Shared generation logic for both Gradio and API"""
+    if duration > 30:
+        duration = 30
+    
+    inputs = processor(text=[prompt], padding=True, return_tensors="pt")
+    max_tokens = int(duration * 50)
+    
+    with torch.no_grad():
+        audio_values = model.generate(**inputs, max_new_tokens=max_tokens)
+
+    sampling_rate = model.config.audio_encoder.sampling_rate
+    audio_data = audio_values[0, 0].cpu().numpy()
+    return sampling_rate, audio_data
+
+# --- API Endpoint for n8n ---
+@app.post("/generate")
+async def api_generate(prompt: str, duration: int = 10):
+    try:
+        sr, audio = generate_core(prompt, duration)
+        byte_io = io.BytesIO()
+        scipy.io.wavfile.write(byte_io, rate=sr, data=audio)
+        return Response(content=byte_io.getvalue(), media_type="audio/wav")
+    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.")
+    
+    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")
+        
+        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
+    )
+
+# Mount Gradio into FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")