Add parameter validation logic in /vton/

app.py

@@ -307,8 +307,27 @@ async def vton_run(
     denoise_steps: int = 30,
     seed: int = 42
 ):
-    target_human = Image.open(io.BytesIO(await upload_human.read()))
-    target_cloth = Image.open(io.BytesIO(await upload_cloth.read()))
+    # Validate file parameters
+    if not upload_human or not upload_human.filename:
+        return {"error": "Human image file is required"}
+    if not upload_cloth or not upload_cloth.filename:
+        return {"error": "Cloth image file is required"}
+    
+    # Validate file contents
+    human_content = await upload_human.read()
+    cloth_content = await upload_cloth.read()
+    
+    if not human_content:
+        return {"error": "Human image file is empty"}
+    if not cloth_content:
+        return {"error": "Cloth image file is empty"}
+    
+    # Reset file pointers for reading
+    await upload_human.seek(0)
+    await upload_cloth.seek(0)
+    
+    target_human = Image.open(io.BytesIO(human_content))
+    target_cloth = Image.open(io.BytesIO(cloth_content))
 
     results = quick_tryon(target_human, target_cloth, input_prompt)