Update app.py

app.py

@@ -1,18 +1,15 @@
 import gradio as gr
 import cv2
 import numpy as np
+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]
 
@@ -21,12 +18,10 @@ 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,24 +34,27 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     return overlay, seed, info
 
 # -----------------------------
-# Hunyuan3D API call
+# Hunyuan3D API client
 # -----------------------------
-from gradio_client import Client
 hunyuan_client = Client("tencent/Hunyuan3D-2.1")
 
+# Auto-detect available API function
+available_functions = hunyuan_client.get_functions()
+if len(available_functions) == 0:
+    raise RuntimeError("No functions available in Hunyuan3D Space")
+api_name = available_functions[0]["api_name"]  # use first available function
+
 def tryon_to_3d(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)
+    tryon_img, seed_used, 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:
         result_3d = hunyuan_client.predict(
-            tryon_img,  # input image
-            api_name="/predict"
+            tryon_img,
+            api_name=api_name
         )
-        return result_3d, tryon_info
+        return result_3d, info
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"
 
@@ -64,14 +62,8 @@ 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; }
 """