Made process work in background for better interaction

app.py

@@ -9,6 +9,7 @@ from openai    import OpenAI
 import base64
 import redis
 import uuid
+from time import sleep
 
 app = FastAPI()
 openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
@@ -42,24 +43,18 @@ async def virtual_try_on(talent_trigger_word: str = Form(...),
 
     request_id = str(uuid.uuid4())
     r.set(request_id, "pending")
-    sleep(5)
-    # step 1, make the base image
-    r.set(request_id, "Creating Base Images")
-    sleep(5)
-    # step 2, combine the garment images
-    r.set(request_id, "Combining Garment Images")
-    sleep(5)
-    # step 3, style the talent
-    r.set(request_id, "Styling Talent")
-    sleep(5)
-    # step 4, create variations
-    r.set(request_id, "Creating Variations")
-    sleep(5)
-    #step 5. Upscale the images
-    r.set(request_id, "Upscaling Images")
-    sleep(5)
-    # step 6, return the result
-    r.set(request_id, "Completed")
+    # Launch background task
+    background_tasks.add_task(
+        run_virtual_tryon_pipeline,
+        request_id,
+        top_garment_image,
+        bottom_garment_image,
+        top_back_garment_image,
+        bottom_back_garment_image,
+        talent_trigger_word,
+        talent_lora_url,
+        num_images
+    )
     return(request_id)
 
 @app.get("/status/{request_id}")
@@ -84,6 +79,52 @@ async def make_base_image(character_lora: str, character_keyword: str, outfit_de
 
 # Function related to virtual outfit try on
 
+async def run_virtual_tryon_pipeline(
+    request_id,
+    top,
+    bottom,
+    top_back,
+    bottom_back,
+    trigger,
+    lora_url,
+    num_images
+):
+    try:
+        r.set(request_id, "Creating base image")
+        sleep(5)
+        # STEP 1: make base image
+        # ... your logic here
+
+        r.set(request_id, "Combining garment images")
+        sleep(5)
+
+        # STEP 2: combine garments
+        # ...
+
+        r.set(request_id, "Styling talent")
+        sleep(5)
+        # STEP 3: style the talent
+        # ...
+
+        r.set(request_id, "Creating variations")
+        sleep(5)
+        # STEP 4: generate variations
+        # ...
+
+        r.set(request_id, "Upscaling images")
+        sleep(5)
+        # STEP 5: upscale
+        # ...
+
+        # STEP 6: Final result
+        final_result = {"images": ["url1", "url2"], "notes": "done"}
+        r.set(f"{request_id}_result", json.dumps(final_result))
+        r.set(request_id, "completed")
+    except Exception as e:
+        r.set(request_id, "error")
+        r.set(f"{request_id}_error", str(e))
+
+
 #figure out how to deal with front vs back images of garments