Update app.py

app.py

@@ -41,6 +41,9 @@ stream_handler.setFormatter(formatter)
 logger.addHandler(file_handler)
 logger.addHandler(stream_handler)
 
+next_id = 0
+next_id_lock = threading.Lock()
+
 class ImageGenerationQueue:
     def __init__(self):
         self.queue = asyncio.Queue()
@@ -327,10 +330,12 @@ async def generate_image(
             return "Failed to generate image"
 
     try:
-        id = queue_manager.next_id
-        queue_manager.next_id += 1
+        with next_id_lock:
+            id = next_id
+            next_id += 1
 
-        queue_manager.add_task(lambda _: generate_image_task(id))
+        background_tasks.add_task(lambda _: generate_image_task(id))
+        # queue_manager.add_task(lambda _: generate_image_task(id))
         position_in_queue = queue_manager.queue.qsize()
 
         # Total queue size is still async