Update app.py

app.py

@@ -1,93 +1,69 @@
-import os
+from fastapi import FastAPI
 import gradio as gr
-from generate import generate_music, load_model
-
-# ------------------------------------------------
-# Load Model (Runs Once)
-# ------------------------------------------------
-print("Loading MusicGen model...")
-model = load_model()
-print("Model ready.")
-
-# ------------------------------------------------
-# Generate Handler
-# ------------------------------------------------
-def create_music(
-    prompt,
-    duration,
-    temperature
-):
-    """
-    Generates music from prompt.
-    Returns path to generated audio file.
-    """
-
-    if not prompt or prompt.strip() == "":
-        return None, "Prompt cannot be empty"
-
-    output_file = generate_music(
-        model=model,
-        prompt=prompt,
-        duration=int(duration),
-        temperature=float(temperature),
+from generate import generate_music
+
+# ======================
+# FASTAPI BACKEND
+# ======================
+
+api = FastAPI(title="AI Background Music Generator")
+
+@api.get("/health")
+def health():
+    return {"status": "running"}
+
+@api.post("/generate")
+def api_generate(prompt: str, duration: int = 10):
+    file = generate_music(prompt, duration)
+    return {"audio": file}
+
+
+# ======================
+# GRADIO UI
+# ======================
+
+def ui_generate(prompt, duration):
+    audio = generate_music(prompt, duration)
+    return audio
+
+
+with gr.Blocks(
+    title="AI Background Music Generator",
+    theme=gr.themes.Soft()
+) as demo:
+
+    gr.Markdown("""
+# 🎵 AI Background Music Generator
+Generate royalty-free background music instantly.
+""")
+
+    prompt = gr.Textbox(
+        label="Music Description",
+        placeholder="Soft cinematic nasheed style..."
+    )
+
+    duration = gr.Slider(
+        minimum=5,
+        maximum=30,
+        value=10,
+        step=1,
+        label="Duration (seconds)"
     )
 
-    return output_file, "Generation Complete ✅"
-
-
-# ------------------------------------------------
-# UI
-# ------------------------------------------------
-with gr.Blocks(title="Tier 3 AI Music Studio") as demo:
-
-    gr.Markdown("# 🎵 Tier 3 AI Music Generator")
-
-    with gr.Row():
-        prompt = gr.Textbox(
-            label="Music Prompt",
-            placeholder="Epic cinematic orchestra, emotional, cinematic buildup..."
-        )
-
-    with gr.Row():
-        duration = gr.Slider(
-            minimum=5,
-            maximum=30,
-            value=10,
-            step=1,
-            label="Duration (seconds)"
-        )
-
-        temperature = gr.Slider(
-            minimum=0.1,
-            maximum=1.5,
-            value=1.0,
-            step=0.1,
-            label="Creativity"
-        )
-
-    generate_btn = gr.Button("Generate Music")
-
-    audio_output = gr.Audio(label="Generated Audio")
-    status = gr.Textbox(label="Status")
-
-    generate_btn.click(
-        fn=create_music,
-        inputs=[
-            prompt,
-            duration,
-            temperature
-        ],
-        outputs=[
-            audio_output,
-            status
-        ],
+    btn = gr.Button("Generate")
+
+    output = gr.Audio(type="filepath")
+
+    btn.click(
+        fn=ui_generate,
+        inputs=[prompt, duration],
+        outputs=output
     )
 
-# ------------------------------------------------
-# Launch
-# ------------------------------------------------
-if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+
+# ======================
+# IMPORTANT FIX
+# ======================
+# Mount GRADIO as ROOT APP
+
+app = gr.mount_gradio_app(api, demo, path="/")