Update app.py

app.py

@@ -12,6 +12,7 @@ from dotenv import load_dotenv
 
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
 import uvicorn
 
 # Load environment variables
@@ -168,6 +169,9 @@ def predict(audio_file_pth, agree):
 
 app = FastAPI()
 
+# Mount the 'outputs' directory to serve static files
+app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")
+
 @app.post("/predict")
 async def predict_endpoint(agree: bool = Form(...), audio_file: UploadFile = File(...)):
     temp_dir = "temp"
@@ -178,9 +182,10 @@ async def predict_endpoint(agree: bool = Form(...), audio_file: UploadFile = Fil
 
     info, audio_output_path = predict(audio_path, agree)
     if audio_output_path:
-        return JSONResponse(content={"info": info, "audio_path": audio_output_path})
+        audio_url = f"/outputs/{os.path.basename(audio_output_path)}"
+        return {"info": info, "audio_path": audio_url}
     else:
-        return JSONResponse(content={"info": info, "audio_path": None}, status_code=400)
+        return {"info": info, "audio_path": None}, 400
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))