Update app.py

app.py

@@ -1,68 +1,61 @@
-import os
 import gradio as gr
 import cv2
 import numpy as np
 from gradio_client import Client
+import io
+from PIL import Image
 
 # -----------------------------
-# Local "try-on" function
+# Hunyuan3D API setup
 # -----------------------------
-def tryon_local(person_img, garment_img, seed, randomize_seed):
-    if person_img is None or garment_img is None:
-        return None, None, "Empty image"
-
-    h_person, w_person = person_img.shape[:2]
-    h_garment, w_garment = garment_img.shape[:2]
-
-    scale = w_person / w_garment
-    new_w = int(w_garment * scale)
-    new_h = int(h_garment * scale)
-    garment_resized = cv2.resize(garment_img, (new_w, new_h))
-
-    overlay = person_img.copy()
-    y_offset = 0
-    x_offset = max(0, (w_person - new_w) // 2)
-
-    if garment_resized.shape[2] == 4:
-        alpha = garment_resized[:, :, 3] / 255.0
-        for c in range(3):
-            overlay[y_offset:y_offset+new_h, x_offset:x_offset+new_w, c] = \
-                alpha * garment_resized[:, :, c] + (1 - alpha) * overlay[y_offset:y_offset+new_h, x_offset:x_offset+new_w, c]
-    else:
-        overlay[y_offset:y_offset+new_h, x_offset:x_offset+new_w] = garment_resized
-
-    info = "Success (local simulation)"
-    return overlay, seed, info
-
-# -----------------------------
-# Hunyuan3D API client
-# -----------------------------
-API_KEY = os.getenv("HF_API_KEY", "hf_YOUR_TOKEN_HERE")  # Replace with your token if not using env var
-hunyuan_client = None
+API_KEY = "HT_Token"  # Replace with your valid key
 try:
     hunyuan_client = Client("tencent/Hunyuan3D-2.1", hf_token=API_KEY)
-    print("Hunyuan3D client loaded successfully!")
+    # Grab first endpoint index for the 3D try-on function
+    fn_index_default = hunyuan_client.endpoints[0].fn_index if hunyuan_client.endpoints else 0
+    client_loaded = True
 except Exception as e:
     print(f"Failed to load Hunyuan3D client: {e}")
     hunyuan_client = None
+    fn_index_default = 0
+    client_loaded = False
 
+# -----------------------------
+# Convert numpy image to bytes
+# -----------------------------
+def np_to_bytes(img_np):
+    img_pil = Image.fromarray(cv2.cvtColor(img_np, cv2.COLOR_BGR2RGB))
+    buf = io.BytesIO()
+    img_pil.save(buf, format="PNG")
+    return buf.getvalue()
+
+# -----------------------------
+# Try-on function using Hunyuan3D API
+# -----------------------------
 def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
-    # If Hunyuan3D client is not loaded, fall back to local
-    if hunyuan_client is None:
-        result_img, seed_used, info = tryon_local(person_img, garment_img, seed, randomize_seed)
-        return result_img, info
+    if not client_loaded:
+        return None, "Hunyuan3D client not loaded. Please check your API key."
+
+    if person_img is None or garment_img is None:
+        return None, "Empty image provided."
 
-    # Call Hunyuan3D API
     try:
-        # Convert images to bytes for API
-        _, person_bytes = cv2.imencode(".png", person_img)
-        _, garment_bytes = cv2.imencode(".png", garment_img)
-        result_3d = hunyuan_client.predict(person_bytes.tobytes(), garment_bytes.tobytes())
-        return result_3d, "Success (Hunyuan3D)"
+        # Convert to bytes
+        person_bytes = np_to_bytes(person_img)
+        garment_bytes = np_to_bytes(garment_img)
+
+        # Call the 3D try-on endpoint
+        result = hunyuan_client.predict(
+            person_bytes,
+            garment_bytes,
+            fn_index=fn_index_default
+        )
+
+        # Convert returned bytes to image
+        result_img = np.array(Image.open(io.BytesIO(result)))
+        return cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR), "3D try-on successful!"
     except Exception as e:
-        # Fallback to local try-on on API error
-        result_img, seed_used, info = tryon_local(person_img, garment_img, seed, randomize_seed)
-        return result_img, f"Hunyuan3D API error: {e}. Used local simulation instead."
+        return None, f"Hunyuan3D API error: {e}"
 
 # -----------------------------
 # Gradio UI