Update app.py

app.py

@@ -1,45 +1,100 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
+from pydantic import BaseModel
+
 import gradio as gr
-from generate import generate_music
+
+from generate import generate_music, build_prompt, load_model
+
+# preload model (removes cold start)
+load_model()
+
+api = FastAPI(title="Basyx Music Generator")
+
+# serve local outputs
+api.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")
+
 
 # ======================
-# FASTAPI BACKEND
+# API MODEL
 # ======================
 
-api = FastAPI(title="AI Background Music Generator")
+class GenerateRequest(BaseModel):
+    prompt: str
+    duration: int = 10
+    style: str | None = "nasheed"
+
+
+BASE_URL = "https://basyx-musicgen.hf.space"
+
+
+# ======================
+# HEALTH CHECK
+# ======================
 
 @api.get("/health")
 def health():
     return {"status": "running"}
 
+
+# ======================
+# REST API (n8n Ready)
+# ======================
+
 @api.post("/generate")
-def api_generate(prompt: str, duration: int = 10):
-    file = generate_music(prompt, duration)
-    return {"audio": file}
+async def generate(req: GenerateRequest):
+
+    if req.duration > 30:
+        raise HTTPException(400, "Duration too long")
+
+    if len(req.prompt) > 200:
+        raise HTTPException(400, "Prompt too long")
+
+    final_prompt = build_prompt(req.prompt, req.style)
+
+    local_path, public_url = generate_music(
+        final_prompt,
+        req.duration
+    )
+
+    filename = local_path.split("/")[-1]
+
+    return JSONResponse({
+        "status": "success",
+        "local_url": f"{BASE_URL}/outputs/{filename}",
+        "public_url": public_url
+    })
 
 
 # ======================
 # GRADIO UI
 # ======================
 
-def ui_generate(prompt, duration):
-    audio = generate_music(prompt, duration)
-    return audio
+def ui_generate(prompt, duration, style):
+
+    final_prompt = build_prompt(prompt, style)
 
+    local_path, _ = generate_music(
+        final_prompt,
+        duration
+    )
+
+    return local_path
 
-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.
-""")
+with gr.Blocks(title="AI Background Music Generator") as demo:
+
+    gr.Markdown("# 🎵 AI Background Music Generator")
 
     prompt = gr.Textbox(
-        label="Music Description",
-        placeholder="Soft cinematic nasheed style..."
+        label="Describe your music"
+    )
+
+    style = gr.Dropdown(
+        choices=["nasheed", "cinematic", "lofi"],
+        value="nasheed",
+        label="Style"
     )
 
     duration = gr.Slider(
@@ -47,7 +102,7 @@ Generate royalty-free background music instantly.
         maximum=30,
         value=10,
         step=1,
-        label="Duration (seconds)"
+        label="Duration"
     )
 
     btn = gr.Button("Generate")
@@ -56,14 +111,10 @@ Generate royalty-free background music instantly.
 
     btn.click(
         fn=ui_generate,
-        inputs=[prompt, duration],
+        inputs=[prompt, duration, style],
         outputs=output
     )
 
 
-# ======================
-# IMPORTANT FIX
-# ======================
-# Mount GRADIO as ROOT APP
-
+# mount UI at root
 app = gr.mount_gradio_app(api, demo, path="/")