Update app.py

app.py

@@ -4,13 +4,15 @@ import numpy as np
 import base64
 from gradio_client import Client
 
-def np_to_base64(img_np):
-    _, buffer = cv2.imencode('.png', img_np)
-    img_bytes = buffer.tobytes()
-    return "data:image/png;base64," + base64.b64encode(img_bytes).decode()
+# -----------------------------
+# Helper: Convert NumPy image to base64
+# -----------------------------
+def np_to_base64(img):
+    _, buffer = cv2.imencode(".png", img)
+    return base64.b64encode(buffer).decode("utf-8")
 
 # -----------------------------
-# Local "try-on" function
+# Local "try-on" simulation
 # -----------------------------
 def tryon_local(person_img, garment_img, seed, randomize_seed):
     if person_img is None or garment_img is None:
@@ -43,25 +45,23 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
 # Hunyuan3D API client
 # -----------------------------
 hunyuan_client = Client("tencent/Hunyuan3D-2.1")
+
+# Get default fn_index
 fn_index_default = hunyuan_client.endpoints[0].fn_index if hunyuan_client.endpoints else 0
 
 def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
+    # Run local try-on first
     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"
 
     try:
-        # Convert images to base64 strings
-        person_base64 = np_to_base64(tryon_img)
-        garment_base64 = np_to_base64(garment_img)
-
-        # Send base64 images to Hunyuan3D API
-        result_3d = hunyuan_client.predict(
-            person_base64,
-            garment_base64,
-            fn_index=fn_index_default
-        )
-        return result_3d, tryon_info
+        # Convert image to base64 for API
+        img_b64 = np_to_base64(tryon_img)
+
+        # Hunyuan3D API only accepts positional arguments (list)
+        result = hunyuan_client.predict([img_b64], fn_index=fn_index_default)
+        return result, tryon_info
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"