Update main.py

main.py

@@ -559,7 +559,12 @@ def process_heavy_files_background(space_id: str, saved_file_paths: List[Dict]):
         print(f"Background processing failed: {e}")
 
 @app.post("/api/spaces/{space_id}/upload")
-async def upload_files(space_id: str, background_tasks: BackgroundTasks, files: List[UploadFile] = File(...)):
+@app.post("/api/spaces/{space_id}/upload")
+async def upload_files(
+    space_id: str, 
+    background_tasks: BackgroundTasks, 
+    files: List[UploadFile] = File(...)
+):
     """Accepts files quickly and processes them in the background"""
     try:
         space_dir = get_space_dir(space_id)
@@ -568,7 +573,7 @@ async def upload_files(space_id: str, background_tasks: BackgroundTasks, files:
         
         saved_files = []
         
-        # 1. Save files to hard drive
+        # 1. Save files to hard drive (Extremely Fast)
         for file in files:
             file_path = uploads_dir / file.filename
             with open(file_path, "wb") as f:
@@ -583,6 +588,7 @@ async def upload_files(space_id: str, background_tasks: BackgroundTasks, files:
         # 2. Hand heavy math and Mongo saving to background task
         background_tasks.add_task(process_heavy_files_background, space_id, saved_files)
         
+        # 3. Reply instantly to prevent timeouts
         return {
             "status": "processing",
             "message": f"Successfully received {len(files)} files. Processing in the background."