Update app/api/image_router.py

app/api/image_router.py

@@ -119,44 +119,47 @@ async def update_image(
         raise HTTPException(status_code=500, detail=str(e))
     
     
-# --- New Virtual Try-On Endpoint with File Uploads ---
+# --- Corrected Virtual Try-On Endpoint with 3 Inputs ---
 @router.post("/virtual-try-on")
 async def virtual_try_on(
-    dress_image: UploadFile = File(..., description="The dress image (Input 1)"),
-    person_image: UploadFile = File(..., description="The person image (Input 2)"),
-    shoes_image: UploadFile = File(None, description="The shoes image (Input 3) - Optional")
-) -> Response: 
+    dress_image: UploadFile = File(..., description="The dress image for try-on (PNG, JPG)"),
+    person_image: UploadFile = File(..., description="The person image for try-on (PNG, JPG)"),
+    shoes_image: Optional[UploadFile] = File(None, description="The shoes image for try-on (PNG, JPG)")
+) -> Response:
     """
-    Performs a virtual try-on. 
-    Accepts: Dress, Person, and optionally Shoes.
+    Performs a virtual try-on using dress, person, and optional shoes.
+    Returns the try-on image directly or a JSON response with a summary if no image.
     """
     logger.info("Received request to /virtual-try-on with image uploads.")
 
-    # Validate mandatory images
+    # 1. Validate mandatory images
     if not dress_image.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="Dress image must be an image type.")
     if not person_image.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="Person image must be an image type.")
 
-    # Validate shoes only if provided
+    # 2. Validate optional shoes
+    # Note: We check 'shoes_image' is not None before checking content_type
     if shoes_image and not shoes_image.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="Shoes image must be an image type.")
 
     try:
+        # 3. Read bytes
         dress_image_bytes = await dress_image.read()
         person_image_bytes = await person_image.read()
         
         shoes_image_bytes = None
         if shoes_image:
-            shoes_image_bytes = await shoes_image.read()
+             shoes_image_bytes = await shoes_image.read()
 
-        # Service returns JSON summary (dict) and BytesIO object
+        # 4. Call Service
         summary, try_on_image_bytes_io = image_service.virtual_try_on(
             dress_image_bytes=dress_image_bytes,
             person_image_bytes=person_image_bytes,
             shoes_image_bytes=shoes_image_bytes
         )
         
+        # 5. Return Result
         if try_on_image_bytes_io:
             logger.info("Virtual try-on successful. Streaming image response.")
             return create_image_streaming_response(try_on_image_bytes_io)
@@ -170,6 +173,7 @@ async def virtual_try_on(
         logger.error(f"Error in /virtual-try-on: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=str(e))
 
+        
 @router.post("/generate-shoe", response_class=StreamingResponse)
 async def generate_shoe(request: ShoeGenerateRequest):
     """