Update server.py

server.py

@@ -1,10 +1,19 @@
 from fastapi import FastAPI, File, UploadFile, Form
-from starlette.responses import JSONResponse
+from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
 import torch
+import os
 from pipeline import build_audiosep, separate_audio
 
 app = FastAPI()
 
+# Create a directory to save processed audio files if it doesn't exist
+if not os.path.exists('processed'):
+    os.makedirs('processed')
+
+# Mount the static files directory
+app.mount("/processed", StaticFiles(directory="processed"), name="processed")
+
 def process_audio(audio_file_path, text):
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
@@ -13,7 +22,7 @@ def process_audio(audio_file_path, text):
         checkpoint_path='checkpoint/audiosep_base_4M_steps.ckpt',
         device=device)
 
-    output_file = 'SeparatedAudio.wav'
+    output_file = os.path.join('processed', 'SeparatedAudio.wav')
     separate_audio(model, audio_file_path, text, output_file, device)
     return output_file
 
@@ -21,11 +30,12 @@ def process_audio(audio_file_path, text):
 async def upload_file(file: UploadFile = File(...), text: str = Form(...)):
     contents = await file.read()
     # Save the audio file
-    with open(file.filename, "wb") as f:
+    input_file_path = os.path.join('uploaded', file.filename)
+    with open(input_file_path, "wb") as f:
         f.write(contents)
     
     # Process the audio file
-    processed_audio_file_path = process_audio(file.filename, text)
+    processed_audio_file_path = process_audio(input_file_path, text)
 
-    # Response with processed audio file path
-    return JSONResponse(status_code=200, content={"message": "Audio file received and processed.", "processed_audio_file": processed_audio_file_path})
+    # Response with processed audio file URL
+    return JSONResponse(status_code=200, content={"message": "Audio file received and processed.", "processed_audio_file": f"/processed/{os.path.basename(processed_audio_file_path)}"})