Update app.py

app.py

@@ -7,9 +7,13 @@ from gradio_client import Client
 # Local "try-on" function
 # -----------------------------
 def tryon_local(person_img, garment_img, seed, randomize_seed):
+    """
+    Simulates try-on by resizing the garment to fit the person and overlaying.
+    """
     if person_img is None or garment_img is None:
         return None, None, "Empty image"
 
+    # Resize garment to width of person image
     h_person, w_person = person_img.shape[:2]
     h_garment, w_garment = garment_img.shape[:2]
 
@@ -18,10 +22,12 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     new_h = int(h_garment * scale)
     garment_resized = cv2.resize(garment_img, (new_w, new_h))
 
+    # Simple overlay: top of garment onto top of person
     overlay = person_img.copy()
     y_offset = 0
     x_offset = max(0, (w_person - new_w) // 2)
 
+    # Handle transparency if 4-channel
     if garment_resized.shape[2] == 4:
         alpha = garment_resized[:, :, 3] / 255.0
         for c in range(3):
@@ -39,14 +45,18 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
 hunyuan_client = Client("tencent/Hunyuan3D-2.1")
 
 def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
-    tryon_img, seed_used, info = tryon_local(person_img, garment_img, seed, randomize_seed)
+    # Step 1: Run local try-on
+    tryon_img, seed_used, tryon_info = tryon_local(person_img, garment_img, seed, randomize_seed)
     if tryon_img is None:
         return None, "Try-on failed"
 
+    # Step 2: Send 2D try-on result to Hunyuan3D
     try:
-        # Direct predict call; do NOT specify api_name
-        result_3d = hunyuan_client.predict(tryon_img)
-        return result_3d, info
+        result_3d = hunyuan_client.predict(
+            tryon_img,   # input image
+            api_name="/process"   # correct endpoint
+        )
+        return result_3d, tryon_info
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"
 
@@ -54,8 +64,14 @@ def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
 # Gradio UI
 # -----------------------------
 css = """
-#col-left, #col-mid, #col-right {margin: 0 auto; max-width: 430px;}
-#col-showcase {margin: 0 auto; max-width: 1100px;}
+#col-left, #col-mid, #col-right {
+    margin: 0 auto;
+    max-width: 430px;
+}
+#col-showcase {
+    margin: 0 auto;
+    max-width: 1100px;
+}
 #button { color: blue; }
 """