Create # app.py

# app.py

@@ -0,0 +1,39 @@
+# app.py
+
+import uvicorn
+from fastapi import FastAPI, Query
+from fastapi.responses import JSONResponse
+from test_music_generator_only import music_model, DEVICE, generate_music
+
+app = FastAPI(title="MusicGenerator API")
+
+# ================================
+# Health check endpoint
+# ================================
+@app.get("/health")
+def health_check():
+    return {"status": "ok", "device": DEVICE}
+
+# ================================
+# Generate music endpoint
+# ================================
+@app.get("/generate")
+def generate(prompt: str = Query(None, description="Optional text prompt for music generation"),
+             seq_len: int = Query(64, description="Length of music sequence to generate")):
+    """
+    Generate a music sequence using MusicGenerator.
+    Returns the output as a JSON array.
+    """
+    try:
+        output = generate_music(prompt=prompt, seq_len=seq_len)
+        # convert tensor to list
+        output_list = output.squeeze(0).cpu().tolist()
+        return JSONResponse(content={"prompt": prompt, "seq_len": seq_len, "output": output_list})
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+
+# ================================
+# Run app
+# ================================
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)