Update backend_server.py

backend_server.py

@@ -0,0 +1,64 @@
+import gradio as gr
+import requests
+from PIL import Image
+import io
+
+# Replace with your friend's laptop backend URL (use ngrok if needed)
+BACKEND_URL = "http://<FRIEND_LAPTOP_IP>:5000/run_vton"
+
+def virtual_try_on_frontend(person_img, cloth_img):
+    """
+    Sends person and cloth images to the backend server and returns the try-on result.
+    """
+    if person_img is None or cloth_img is None:
+        return None
+
+    # Convert PIL Images to bytes
+    person_bytes = io.BytesIO()
+    person_img.save(person_bytes, format="PNG")
+    person_bytes = person_bytes.getvalue()
+
+    cloth_bytes = io.BytesIO()
+    cloth_img.save(cloth_bytes, format="PNG")
+    cloth_bytes = cloth_bytes.getvalue()
+
+    try:
+        # Send POST request with images as files
+        response = requests.post(
+            BACKEND_URL,
+            files={
+                "person": ("person.png", person_bytes, "image/png"),
+                "cloth": ("cloth.png", cloth_bytes, "image/png")
+            },
+            timeout=60
+        )
+        response.raise_for_status()
+
+        # Convert returned bytes to PIL image
+        result_bytes = io.BytesIO(response.content)
+        result_img = Image.open(result_bytes).convert("RGB")
+        return result_img
+
+    except Exception as e:
+        print("[ERROR] Backend request failed:", e)
+        return None
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## 👕 Virtual Try-On Demo (Frontend → Backend)")
+
+    with gr.Row():
+        person_input = gr.Image(type="pil", label="Upload Person Image")
+        cloth_input = gr.Image(type="pil", label="Upload Garment")
+
+    output = gr.Image(type="pil", label="Result")
+
+    run_btn = gr.Button("Run Virtual Try-On")
+    run_btn.click(
+        fn=virtual_try_on_frontend,
+        inputs=[person_input, cloth_input],
+        outputs=output
+    )
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)