Update app.py

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI
+from fastapi.responses import FileResponse
 import gradio as gr
 from generate import generate_music
 
@@ -12,11 +13,15 @@ api = FastAPI(title="AI Background Music Generator")
 def health():
     return {"status": "running"}
 
-@api.post("/generate")
-def api_generate(prompt: str, duration: int = 10):
-    file = generate_music(prompt, duration)
-    return {"audio": file}
+@app.post("/generate")
+async def api_generate(prompt: str, duration: int = 10):
+    file_path = generate_music(prompt, duration)
 
+    return FileResponse(
+        path=file_path,
+        media_type="audio/wav",
+        filename="music.wav"
+    )
 
 # ======================
 # GRADIO UI