Update main.py

main.py

@@ -1,16 +1,17 @@
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse
-from fastapi.middleware.cors import CORSMiddleware
 import os
 import io
 
 app = FastAPI()
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=['*']
-)
+# Define the directory to store uploaded files
+UPLOAD_DIR = "/content/"
 
+@app.on_event("startup")
+async def startup_event():
+    # Check and create the directory if it doesn't exist
+    os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 @app.get("/")
 def read_root():
@@ -18,14 +19,21 @@ def read_root():
 
 @app.post("/uploadfile/")
 async def create_upload_file(file: UploadFile = File(...)):
-    # Save the file with a specific name
-    file_path = "inputvoice.mp3"
-    with open(file_path, "wb") as f:
-        f.write(file.file.read())
-    
-    # Read the content of the saved file
-    with open(file_path, "rb") as f:
-        file_content = f.read()
-    
-    # Return the content as a streaming response
-    return StreamingResponse(io.BytesIO(file_content), media_type="audio/mpeg", headers={"Content-Disposition": "inline; filename=inputvoice.mp3"})
+    try:
+        # Save the file with a specific name
+        file_path = os.path.join(UPLOAD_DIR, "inputvoice.mp3")
+
+        with open(file_path, "wb") as f:
+            f.write(file.file.read())
+
+        # Read the content of the saved file
+        with open(file_path, "rb") as f:
+            file_content = f.read()
+
+        # Return the content as a streaming response
+        return StreamingResponse(io.BytesIO(file_content), media_type="audio/mpeg", headers={"Content-Disposition": "inline; filename=inputvoice.mp3"})
+
+    except PermissionError as e:
+        return {"error": f"PermissionError: {str(e)}"}
+
+