Update main.py

main.py

@@ -98,12 +98,27 @@ async def set_api_key(api_key: APIKey):
 ## POST - /upload_file_on_server
 @app.post("/upload_file_on_server")
 async def upload_file_on_server(file: UploadFile = File(...)):
-    # Save the uploaded file to a temporary location
-    with open(file.filename, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
+    file_content = file.read()  # Load the document
+    
+    # Create a directory if it doesn't exist
+    data_dir = "/data"
+    # os.makedirs(data_dir, exist_ok=True)
+    
+    # Create a temporary file in the data directory
+    with tempfile.NamedTemporaryFile(delete=False, dir=data_dir) as temp_file:
+        temp_file.write(file_content)  # Write the uploaded file content to the temporary file
+        temp_file_path = temp_file.name  # Get the path of the temporary file
+        return temp_file_path
+    # # Save the uploaded file to the temporary directory
+    # file_path = os.path.join(temp_dir, file.filename)
+    
+    # # Save the uploaded file to a temporary location
+    # with open(file.filename, "wb") as buffer:
+    #     shutil.copyfileobj(file.file, buffer)
+    
+    # # Return a response
+    # return {"filename": file.filename, "message": "File uploaded successfully"}
     
-    # Return a response
-    return {"filename": file.filename, "message": "File uploaded successfully"}
 
 ## POST - /load_file
 # Load the file, split it into document chunks, and upload the document embeddings into a vectorstore