feat: async lock

run_fastapi.py

@@ -5,7 +5,7 @@ from fastapi.templating import Jinja2Templates
 import uuid
 import shutil
 import os
-import cv2
+import asyncio
 
 from pipeline.process_session import process_session_image
 from database.db import init_db
@@ -15,6 +15,7 @@ from models.dino import DinoWrapper
 
 from huggingface_hub import hf_hub_download
 
+process_lock = asyncio.Lock()
 
 app = FastAPI()
 
@@ -32,13 +33,9 @@ print("WRITE to /tmp:", os.access("/tmp", os.W_OK))
 BASE_DIR = "/tmp" if os.access("/tmp", os.W_OK) else "."
 
 WEIGHTS_DIR = os.path.join(BASE_DIR, "weights")
-UPLOADS_DIR = os.path.join(BASE_DIR, "uploads")
-OUTPUTS_DIR = os.path.join(BASE_DIR, "outputs")
 
 # Create directories if not exist
 os.makedirs(WEIGHTS_DIR, exist_ok=True)
-os.makedirs(UPLOADS_DIR, exist_ok=True)
-os.makedirs(OUTPUTS_DIR, exist_ok=True)
 
 # --- Initialize database
 init_db()
@@ -117,27 +114,28 @@ def show_results(request: Request, session_id: str):
 
 
 @app.post("/process")
-def process_image(request: Request, image: UploadFile = Form(...), prompt: str = Form(...)):
-    # 1. Save uploaded image
-    session_id = uuid.uuid4().hex
-    save_dir = "uploads"
-    os.makedirs(save_dir, exist_ok=True)
-
-    image_path = os.path.join(save_dir, f"{session_id}_{image.filename}")
-    with open(image_path, "wb") as buffer:
-        shutil.copyfileobj(image.file, buffer)
-
-    # 2. Run main pipeline
-    process_session_image(
-        session_id=session_id,
-        image_path=image_path,
-        prompt_text=prompt,
-        sam_wrapper=sam,
-        dino_wrapper=dino
-    )
+async def process_image(request: Request, image: UploadFile = Form(...), prompt: str = Form(...)):
+    async with process_lock:
+        # 1. Save uploaded image
+        session_id = uuid.uuid4().hex
+        save_dir = "uploads"
+        os.makedirs(save_dir, exist_ok=True)
+
+        image_path = os.path.join(save_dir, f"{session_id}_{image.filename}")
+        with open(image_path, "wb") as buffer:
+            shutil.copyfileobj(image.file, buffer)
+
+        # 2. Run main pipeline
+        process_session_image(
+            session_id=session_id,
+            image_path=image_path,
+            prompt_text=prompt,
+            sam_wrapper=sam,
+            dino_wrapper=dino
+        )
 
-    # 3. Redirect to results page
-    return RedirectResponse(f"/results/{session_id}", status_code=303)
+        # 3. Redirect to results page
+        return RedirectResponse(f"/results/{session_id}", status_code=303)
 
 
 @app.post("/finalize", response_class=HTMLResponse)