lock via async

app.py

@@ -14,6 +14,7 @@ import uuid
 import logging
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+from asyncio import Lock
 
 logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO"))
 
@@ -26,6 +27,7 @@ DB_PATH.mkdir(exist_ok=True, parents=True)
 IMGS_PATH.mkdir(exist_ok=True, parents=True)
 
 database = Database(DB_PATH)
+generate_lock = Lock()
 
 
 model_id = "segmind/Segmind-Vega"
@@ -98,14 +100,15 @@ async def generate_image(
         return StreamingResponse(open(cached_img[0], "rb"), media_type="image/jpeg")
 
     logging.info(f"Image not found in cache, generating new image")
-    pil_image = generate(prompt, negative_prompt, seed)
-    img_id = str(uuid.uuid4())
-    img_path = IMGS_PATH / f"{img_id}.jpg"
-    pil_image.save(img_path)
-    img_io = io.BytesIO()
-    pil_image.save(img_io, "JPEG")
-    img_io.seek(0)
-    database.insert(prompt, negative_prompt, str(img_path), seed)
+    async with generate_lock:
+        pil_image = generate(prompt, negative_prompt, seed)
+        img_id = str(uuid.uuid4())
+        img_path = IMGS_PATH / f"{img_id}.jpg"
+        pil_image.save(img_path)
+        img_io = io.BytesIO()
+        pil_image.save(img_io, "JPEG")
+        img_io.seek(0)
+        database.insert(prompt, negative_prompt, str(img_path), seed)
 
     return StreamingResponse(img_io, media_type="image/jpeg")