Update app.py

app.py

@@ -274,10 +274,10 @@ async def generate_image(
     image: UploadFile = File(...),
     background_tasks: BackgroundTasks = BackgroundTasks()
 ):
-    async def generate_image_task():
+    async def generate_image_task(position_in_queue):
         try:
             # Save the uploaded image to a temporary file
-            temp_image_path = f"/tmp/{int(time.time())}_{image.filename}"
+            temp_image_path = f"/tmp/{position_in_queue}_{image.filename}"
             with open(temp_image_path, "wb") as temp_image:
                 temp_image.write(image.file.read())
 
@@ -302,8 +302,8 @@ async def generate_image(
             return "Failed to generate image"
 
     try:
-        background_tasks.add_task(generate_image_task)
-        position_in_queue = queue_manager.queue.qsize()
+        position_in_queue = queue_manager.queue.qsize() + 1
+        queue_manager.add_task(lambda _: generate_image_task(position_in_queue))
         # Total queue size is still async
         total_queue_size = await queue_manager.get_total_queue_size()  # Implement this function