working on file input

app.py

@@ -12,6 +12,7 @@ import uuid
 from time import sleep
 import json
 from functions import combine_images_side_by_side
+from typing import Optional
 
 from dotenv import load_dotenv
 load_dotenv()
@@ -33,35 +34,43 @@ def greet_json():
 
 @app.post("/virtualTryOn", summary="Virtual try on single call", 
     description="Virtual try on single call for complete outfit try on")
-async def virtual_try_on(background_tasks: BackgroundTasks,
-                        talent_trigger_word: str = Form(...),
-                        talent_lora_url: str = Form(...),
-                        num_images: int = 4,
-                        top_garment_image: UploadFile = File(None), 
-                        bottom_garment_image: UploadFile = File(None),
-                        top_back_garment_image: UploadFile = File(None), 
-                        bottom_back_garment_image: UploadFile = File(None)):
+async def virtual_try_on(
+    background_tasks: BackgroundTasks,
+    talent_trigger_word: str = Form(...),
+    talent_lora_url: str = Form(...),
+    num_images: int = 4,
+    top_garment_image: UploadFile = File(...), 
+    bottom_garment_image: UploadFile = File( None),
+    top_back_garment_image: UploadFile = File(None), 
+    bottom_back_garment_image: UploadFile = File(None)):
     """
     Pass in the top garment image, bottom garment image, top back garment image, bottom back garment image, talent trigger word, talent lora url and number of images to generate.
     Only one front garment image is required, the rest are optional.
     Default number of images is 4.
     """
-
+    print(top_garment_image)
     request_id = str(uuid.uuid4())
     r.set(request_id, "pending")
-    # Launch background task
+    
+    # Read all files first
+    top_garment_data = await safe_read_file(top_garment_image)
+    bottom_garment_data = await safe_read_file(bottom_garment_image)
+    top_back_garment_data = await safe_read_file(top_back_garment_image)
+    bottom_back_garment_data = await safe_read_file(bottom_back_garment_image)
+    
+    # Launch background task with the actual data
     background_tasks.add_task(
         run_virtual_tryon_pipeline,
         request_id,
-        top_garment_image,
-        bottom_garment_image,
-        top_back_garment_image,
-        bottom_back_garment_image,
+        top_garment_data,
+        bottom_garment_data,
+        top_back_garment_data,
+        bottom_back_garment_data,
         talent_trigger_word,
         talent_lora_url,
         num_images
     )
-    return(request_id)
+    return {"request_id": request_id}
 
 @app.get("/status/{request_id}")
 async def check_status(request_id: str):
@@ -95,10 +104,10 @@ async def safe_read_file(file: UploadFile):
 
 async def run_virtual_tryon_pipeline(
     request_id,
-    top_garment_image,
-    bottom_garment_image,
-    top_back_garment_image,
-    bottom_back_garment_image,
+    top_garment_data: bytes,
+    bottom_garment_data: bytes,
+    top_back_garment_data: bytes,
+    bottom_back_garment_data: bytes,
     talent_trigger_word,
     talent_lora_url,
     num_images
@@ -107,23 +116,9 @@ async def run_virtual_tryon_pipeline(
         r.set(request_id, "checking incoming data...")
         front_type = ""
         back_type = ""
-        print("checking incoming data")
-        # Read all files once at the start
-        try:
-            print("reading top front")
-            top_garment_data = await safe_read_file(top_garment_image)
-            print("reading bottom front")
-            bottom_garment_data = await safe_read_file(bottom_garment_image)
-            print("reading top back")
-            top_back_garment_data = await safe_read_file(top_back_garment_image)
-            print("reading bottom back")
-            bottom_back_garment_data = await safe_read_file(bottom_back_garment_image)
-        except Exception as e:
-            print(f"Error reading garment images: {e}")
-            #return {"error": "Error reading garment images"}
-
+        
+        # Now we can use the data directly
         if top_garment_data is None and bottom_garment_data is None:
-            print("Missing garment images from front")
             return {"error": "Missing garment images"}
             
         if top_garment_data and bottom_garment_data:
@@ -368,3 +363,15 @@ async def describe_garment(image: UploadFile = File(...)):
 
 
 
+@app.post("/upload")
+async def upload_files(
+    name: str = Form(...),
+    image1: UploadFile = File(...),
+    image2: UploadFile = File(None)  # Optional!
+):
+    result = {
+        "name": name,
+        "image1_filename": image1.filename,
+        "image2_filename": image2.filename if image2 else "Not provided"
+    }
+    return result