Update app.py

app.py

@@ -1,18 +1,11 @@
 import gradio as gr
 import cv2
 import numpy as np
-import base64
 from gradio_client import Client
+import base64
 
 # -----------------------------
-# 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" simulation
+# Local "try-on" function
 # -----------------------------
 def tryon_local(person_img, garment_img, seed, randomize_seed):
     if person_img is None or garment_img is None:
@@ -44,24 +37,31 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
 # -----------------------------
 # Hunyuan3D API client
 # -----------------------------
-hunyuan_client = Client("tencent/Hunyuan3D-2.1")
+# Set your API key here
+API_KEY = "YOUR_HUNYUAN3D_API_KEY"
+hunyuan_client = Client("tencent/Hunyuan3D-2.1", hf_token=API_KEY)
 
-# Get default fn_index
-fn_index_default = hunyuan_client.endpoints[0].fn_index if hunyuan_client.endpoints else 0
+def image_to_bytes(img):
+    """Convert numpy image to PNG bytes for API upload"""
+    _, buf = cv2.imencode(".png", img)
+    return buf.tobytes()
 
 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 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
+        # Convert images to bytes to avoid JSON serialization errors
+        person_bytes = image_to_bytes(person_img)
+        garment_bytes = image_to_bytes(garment_img)
+
+        # Call Hunyuan3D API
+        result_bytes = hunyuan_client.predict([person_bytes, garment_bytes])
+        # Convert returned bytes back to numpy image
+        nparr = np.frombuffer(result_bytes, np.uint8)
+        result_img = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
+        return result_img, tryon_info
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"