"""
@app.get("/", response_class=HTMLResponse)
async def index():
return HTML_PAGE
@app.post("/api/generate")
async def generate(request: Request):
import time
body = await request.json()
prompt = body.get("prompt", "").strip()
if not prompt:
return JSONResponse({"detail": "Prompt is required"}, status_code=400)
negative_prompt = body.get("negative_prompt", "Low quality, distorted.")
duration = min(max(int(body.get("duration", 3)), 1), 10)
steps = min(max(int(body.get("steps", 20)), 10), 50)
start = time.time()
audio = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
num_inference_steps=steps,
audio_length_in_s=float(duration),
guidance_scale=2.5,
num_waveforms_per_prompt=1,
).audios[0]
gen_time = time.time() - start
filename = f"sfx_{uuid.uuid4().hex[:8]}.wav"
out_path = OUTPUT_DIR / filename
scipy.io.wavfile.write(str(out_path), rate=16000, data=audio)
return {"url": f"/files/{filename}", "generation_time": gen_time}
@app.get("/files/{filename}")
async def get_file(filename: str):
filepath = OUTPUT_DIR / filename
if not filepath.exists():
return JSONResponse({"detail": "File not found"}, status_code=404)
return FileResponse(str(filepath), media_type="audio/wav", filename=filename)
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=7860)