Update app.py

app.py

@@ -1,42 +1,98 @@
 import gradio as gr
 import requests
 import base64
-import numpy as np
 import io
 import soundfile as sf
 
+# 🔥 IMPORTANT: your ngrok API endpoint
 API_URL = "https://bustled-hertha-unprojective.ngrok-free.dev/generate"
 
+
 def generate(prompt, duration, steps, cfg):
-    res = requests.post(API_URL, json={
-        "prompt": prompt,
-        "duration": duration,
-        "steps": steps,
-        "cfg_scale": cfg
-    }, timeout=120)
+    try:
+        # 🚀 send request to backend
+        res = requests.post(
+            API_URL,
+            json={
+                "prompt": prompt,
+                "duration": duration,
+                "steps": steps,
+                "cfg_scale": cfg
+            },
+            timeout=180  # longer timeout for generation
+        )
+
+        print("STATUS:", res.status_code)
+
+        # ❌ backend error
+        if res.status_code != 200:
+            print("❌ Backend error:", res.text[:300])
+            return None
+
+        # ❌ empty response
+        if not res.text:
+            print("❌ Empty response")
+            return None
+
+        # ❌ invalid JSON
+        try:
+            data = res.json()
+        except Exception:
+            print("❌ Invalid JSON:", res.text[:300])
+            return None
+
+        # ❌ missing audio
+        if "audio" not in data:
+            print("❌ No audio key:", data)
+            return None
+
+        # ✅ decode base64 → audio
+        audio_bytes = base64.b64decode(data["audio"])
+        audio, sr = sf.read(io.BytesIO(audio_bytes))
+
+        return sr, audio
+
+    except Exception as e:
+        print("❌ Request failed:", str(e))
+        return None
 
-    data = res.json()
 
-    # ✅ decode base64 properly
-    audio_bytes = base64.b64decode(data["audio"])
+# 🎨 UI
+with gr.Blocks(title="AutoMix AI 🎵") as demo:
+    gr.Markdown("# 🎵 AutoMix AI Beat Generator")
+    gr.Markdown("Generate AI beats using diffusion model 🚀")
 
-    # convert to numpy using soundfile
-    audio, sr = sf.read(io.BytesIO(audio_bytes))
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="🎧 Prompt",
+                placeholder="Dark trap beat, 140 BPM, heavy 808..."
+            )
 
-    return sr, audio
+            duration = gr.Slider(
+                1, 10, value=5, step=1,
+                label="⏱ Duration (seconds)"
+            )
 
+            steps = gr.Slider(
+                20, 80, value=40, step=5,
+                label="⚙️ Diffusion Steps"
+            )
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 🎵 AutoMix AI Beat Generator")
+            cfg = gr.Slider(
+                1, 10, value=7, step=0.5,
+                label="🎛 CFG Scale"
+            )
 
-    prompt = gr.Textbox(label="Prompt", placeholder="Dark trap beat, 140 BPM...")
-    duration = gr.Slider(1, 10, value=5, label="Duration (seconds)")
-    steps = gr.Slider(20, 80, value=40, label="Steps")
-    cfg = gr.Slider(1, 10, value=7, label="CFG Scale")
+            btn = gr.Button("🚀 Generate Beat", variant="primary")
 
-    btn = gr.Button("Generate Beat 🚀")
-    out = gr.Audio(label="Output")
+        with gr.Column():
+            output = gr.Audio(label="🎵 Generated Beat")
 
-    btn.click(generate, [prompt, duration, steps, cfg], out)
+    btn.click(
+        fn=generate,
+        inputs=[prompt, duration, steps, cfg],
+        outputs=output
+    )
 
 demo.launch()