Update server.py

server.py

@@ -7,9 +7,11 @@ 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')
+@app.on_event("startup")
+async def startup_event():
+    # Create directories for uploaded and processed files
+    os.makedirs('uploaded', exist_ok=True)
+    os.makedirs('processed', exist_ok=True)
 
 # Mount the static files directory
 app.mount("/processed", StaticFiles(directory="processed"), name="processed")
@@ -29,7 +31,7 @@ def process_audio(audio_file_path, text):
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...), text: str = Form(...)):
     contents = await file.read()
-    # Save the audio file
+    # Save the uploaded audio file
     input_file_path = os.path.join('uploaded', file.filename)
     with open(input_file_path, "wb") as f:
         f.write(contents)
@@ -38,4 +40,5 @@ async def upload_file(file: UploadFile = File(...), text: str = Form(...)):
     processed_audio_file_path = process_audio(input_file_path, text)
 
     # 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)}"})
+    processed_file_url = f"/processed/{os.path.basename(processed_audio_file_path)}"
+    return JSONResponse(status_code=200, content={"message": "Audio file received and processed.", "processed_audio_file": processed_file_url})