Update app.py

app.py

@@ -3,59 +3,48 @@ import requests
 from PIL import Image
 from io import BytesIO
 
-# -------------------------
-# CONFIG
-# -------------------------
-BACKEND_URL = "http://10.10.11.115:5000/run_vton"  # your friend's laptop URL
+# Backend server URL (your friend’s laptop)
+BACKEND_URL = "http://10.10.11.115:5000/run_vton"
 
-# -------------------------
-# FUNCTION TO CALL BACKEND
-# -------------------------
-def vton_interface(person_image, cloth_image):
-    try:
-        # Convert PIL images to bytes
-        person_bytes = BytesIO()
-        person_image.save(person_bytes, format="PNG")
-        person_bytes = person_bytes.getvalue()
-
-        cloth_bytes = BytesIO()
-        cloth_image.save(cloth_bytes, format="PNG")
-        cloth_bytes = cloth_bytes.getvalue()
-
-        # Send POST request to backend server
-        response = requests.post(
-            BACKEND_URL,
-            files={
-                "person_image": ("person.png", person_bytes, "image/png"),
-                "cloth_image": ("cloth.png", cloth_bytes, "image/png")
-            },
-        )
-
-        # Check response
-        if response.status_code != 200:
-            return None
-
-        # Convert response bytes to image
-        result_image = Image.open(BytesIO(response.content))
-        return result_image
+def vton_via_backend(person_img, garment_img):
+    if person_img is None or garment_img is None:
+        return None
 
+    # Convert PIL images to bytes
+    person_bytes = BytesIO()
+    garment_bytes = BytesIO()
+    person_img.save(person_bytes, format="PNG")
+    garment_img.save(garment_bytes, format="PNG")
+    person_bytes.seek(0)
+    garment_bytes.seek(0)
+
+    # POST request to backend
+    files = {
+        "person": ("person.png", person_bytes, "image/png"),
+        "garment": ("garment.png", garment_bytes, "image/png")
+    }
+    try:
+        response = requests.post(BACKEND_URL, files=files)
+        response.raise_for_status()  # Raise error if bad status
+        result_img = Image.open(BytesIO(response.content))
+        return result_img
     except Exception as e:
-        print("[ERROR] Frontend request failed:", e)
+        print("[ERROR] Backend request failed:", e)
         return None
 
-# -------------------------
-# GRADIO INTERFACE
-# -------------------------
-demo = gr.Interface(
-    fn=vton_interface,
-    inputs=[
-        gr.Image(type="pil", label="Upload Person Image"),
-        gr.Image(type="pil", label="Upload Cloth Image"),
-    ],
-    outputs=gr.Image(label="Result"),
-    title="Virtual Try-On Demo",
-    description="Upload a person image and a garment image to see the virtual try-on result."
-)
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## 👕 Virtual Try-On via Remote Backend")
+    with gr.Row():
+        person_input = gr.Image(type="pil", label="Upload Person Image")
+        garment_input = gr.Image(type="pil", label="Upload Garment Image")
+    output = gr.Image(type="pil", label="Result")
+    run_btn = gr.Button("Run Virtual Try-On")
+    run_btn.click(
+        fn=vton_via_backend,
+        inputs=[person_input, garment_input],
+        outputs=output
+    )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)