fix timeouts

main.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from fastapi import FastAPI, UploadFile, File, HTTPException, Depends
+from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, BackgroundTasks
 import httpx
 from contextlib import asynccontextmanager
 from dotenv import load_dotenv
@@ -11,7 +11,7 @@ import logging
 import random
 from io import BytesIO
 from urllib.parse import unquote
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 import stripe
 from typing import Optional
 from routers import user_models
@@ -63,6 +63,7 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
+    expose_headers=["*"],  # Ensure all headers are exposed to frontend
 )
 
 # Pydantic models
@@ -902,6 +903,37 @@ async def debug_environment():
     
     return debug_info
 
+@app.post("/debug/test-background", tags=["Debug"])
+async def test_background_response(background_tasks: BackgroundTasks):
+    """Test endpoint to debug background task response issues"""
+    from fastapi.responses import JSONResponse
+    
+    def dummy_background_task():
+        import time
+        time.sleep(1)  # Simulate some work
+        logging.info("Background task completed")
+    
+    # Add a background task
+    background_tasks.add_task(dummy_background_task)
+    
+    response_data = {
+        "message": "Background task started",
+        "test_id": "test-123",
+        "status": "processing"
+    }
+    
+    logging.info(f"Returning test response: {response_data}")
+    
+    return JSONResponse(
+        content=response_data,
+        status_code=200,
+        headers={
+            "Content-Type": "application/json",
+            "Connection": "close",
+            "Cache-Control": "no-cache"
+        }
+    )
+
 @app.get("/proxy/model", tags=["Proxy"])
 async def proxy_model(url: str):
     """

routers/user_models.py

@@ -1,4 +1,5 @@
 from fastapi import APIRouter, Depends, HTTPException, Request, BackgroundTasks, UploadFile, File, Form
+from fastapi.responses import JSONResponse
 from auth import get_current_active_user, User, supabase
 import logging
 import httpx
@@ -334,14 +335,26 @@ async def text_to_3d(prompt: TextPrompt, background_tasks: BackgroundTasks, curr
         # Add background task for processing
         background_tasks.add_task(_process_text_to_3d_background, generated_model_id, current_user.id, prompt.text)
         
-        # Return immediately
-        return {
+        # Return immediately with explicit headers
+        response_data = {
             "generated_model_id": generated_model_id,
             "status": "initializing",
             "original_prompt": prompt.text,
             "message": "Generation started. Use the progress_update endpoint to check status."
         }
         
+        logging.info(f"Returning response for text-to-3d: {response_data}")
+        
+        return JSONResponse(
+            content=response_data,
+            status_code=200,
+            headers={
+                "Content-Type": "application/json",
+                "Connection": "close",
+                "Cache-Control": "no-cache"
+            }
+        )
+        
     except Exception as ex:
         logging.error(f"Failed to create initial model DB record: {ex}")
         raise HTTPException(status_code=500, detail=f"Failed to start generation: {ex}")
@@ -383,14 +396,26 @@ async def image_to_3d(request: ImageTo3DRequest, background_tasks: BackgroundTas
         # Add background task for Meshy API processing
         background_tasks.add_task(_process_image_to_3d_background, generated_model_id, payload, "image_to_3d")
 
-        # Return immediately
-        return {
+        # Return immediately with explicit headers
+        response_data = {
             "generated_model_id": generated_model_id,
             "status": "initializing",
             "input_image_url": request.image_url,
             "message": "Generation started. Use the progress_update endpoint to check status."
         }
         
+        logging.info(f"Returning response for image-to-3d: {response_data}")
+        
+        return JSONResponse(
+            content=response_data,
+            status_code=200,
+            headers={
+                "Content-Type": "application/json",
+                "Connection": "close", 
+                "Cache-Control": "no-cache"
+            }
+        )
+        
     except Exception as ex:
         logging.error(f"Failed to create initial model DB record: {ex}")
         raise HTTPException(status_code=500, detail=f"Failed to start generation: {ex}")
@@ -449,14 +474,26 @@ async def image_to_3d_upload(
         # Add background task for Meshy API processing
         background_tasks.add_task(_process_image_to_3d_background, generated_model_id, payload, "image_to_3d")
 
-        # Return immediately
-        return {
+        # Return immediately with explicit headers
+        response_data = {
             "generated_model_id": generated_model_id,
             "status": "initializing",
             "input_image_filename": file.filename,
             "message": "Generation started. Use the progress_update endpoint to check status."
         }
         
+        logging.info(f"Returning response for image-to-3d/upload: {response_data}")
+        
+        return JSONResponse(
+            content=response_data,
+            status_code=200,
+            headers={
+                "Content-Type": "application/json",
+                "Connection": "close",
+                "Cache-Control": "no-cache"
+            }
+        )
+        
     except Exception as ex:
         logging.error(f"Failed to create initial model DB record: {ex}")
         raise HTTPException(status_code=500, detail=f"Failed to start generation: {ex}")
@@ -524,14 +561,26 @@ async def multi_image_to_3d_upload(
         # Add background task for Meshy API processing
         background_tasks.add_task(_process_image_to_3d_background, generated_model_id, payload, "multi_image_to_3d")
 
-        # Return immediately
-        return {
+        # Return immediately with explicit headers
+        response_data = {
             "generated_model_id": generated_model_id,
             "status": "initializing",
             "input_image_filenames": filenames,
             "message": "Generation started. Use the progress_update endpoint to check status."
         }
         
+        logging.info(f"Returning response for multi-image-to-3d: {response_data}")
+        
+        return JSONResponse(
+            content=response_data,
+            status_code=200,
+            headers={
+                "Content-Type": "application/json",
+                "Connection": "close",
+                "Cache-Control": "no-cache"
+            }
+        )
+        
     except Exception as ex:
         logging.error(f"Failed to create initial model DB record: {ex}")
         raise HTTPException(status_code=500, detail=f"Failed to start generation: {ex}")