Update app.py

app.py

@@ -2,10 +2,16 @@ import gradio as gr
 import cv2
 import numpy as np
 from gradio_client import Client
-import base64
+from io import BytesIO
+from PIL import Image
 
 # -----------------------------
-# Local "try-on" function
+# Hugging Face API Key
+# -----------------------------
+API_KEY = "HF_Token"  # <-- Replace with your Hugging Face token
+
+# -----------------------------
+# Local "try-on" simulation
 # -----------------------------
 def tryon_local(person_img, garment_img, seed, randomize_seed):
     if person_img is None or garment_img is None:
@@ -35,33 +41,37 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     return overlay, seed, info
 
 # -----------------------------
-# Hunyuan3D API client
+# Initialize Hunyuan3D API client
 # -----------------------------
-# Set your API key here
-API_KEY = "YOUR_HUNYUAN3D_API_KEY"
 hunyuan_client = Client("tencent/Hunyuan3D-2.1", hf_token=API_KEY)
 
-def image_to_bytes(img):
-    """Convert numpy image to PNG bytes for API upload"""
-    _, buf = cv2.imencode(".png", img)
-    return buf.tobytes()
+# -----------------------------
+# Convert NumPy image to bytes for API
+# -----------------------------
+def np_to_bytes(img: np.ndarray):
+    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    pil_img = Image.fromarray(img_rgb)
+    buf = BytesIO()
+    pil_img.save(buf, format="PNG")
+    buf.seek(0)
+    return buf.read()
 
+# -----------------------------
+# Try-on using Hunyuan3D
+# -----------------------------
 def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
-    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"
+        return None, "Local try-on failed"
 
     try:
-        # 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
+        person_bytes = np_to_bytes(person_img)
+        garment_bytes = np_to_bytes(garment_img)
+
+        # Use the correct endpoint without keyword args
+        result_bytes = hunyuan_client.predict(person_bytes, garment_bytes)
+        result_img = np.array(Image.open(BytesIO(result_bytes)))
+        return result_img, info + " + Hunyuan3D processed"
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"