Update app.py

app.py

@@ -2,16 +2,22 @@ import gradio as gr
 import cv2
 import numpy as np
 from gradio_client import Client
-from io import BytesIO
-from PIL import Image
+import getpass  # for secure token input
 
 # -----------------------------
-# Hugging Face API Key
+# Prompt for Hugging Face API key
 # -----------------------------
-API_KEY = "HT_Token"  # <-- Replace with your Hugging Face token
+API_KEY = getpass.getpass("Enter your Hugging Face API key: ").strip()
+
+try:
+    hunyuan_client = Client("tencent/Hunyuan3D-2.1", hf_token=API_KEY)
+    print("Hunyuan3D client loaded successfully!")
+except Exception as e:
+    print(f"Failed to load Hunyuan3D client: {e}")
+    hunyuan_client = None
 
 # -----------------------------
-# 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:
@@ -41,39 +47,26 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     return overlay, seed, info
 
 # -----------------------------
-# Initialize Hunyuan3D API client
-# -----------------------------
-hunyuan_client = Client("tencent/Hunyuan3D-2.1", hf_token=API_KEY)
-
-# -----------------------------
-# 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
+# Hunyuan3D API call
 # -----------------------------
 def tryon_to_3d(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, "Local try-on failed"
+    overlay_img, seed_used, info = tryon_local(person_img, garment_img, seed, randomize_seed)
+    if overlay_img is None:
+        return None, info
+
+    if hunyuan_client is None:
+        return overlay_img, "Hunyuan3D client not loaded"
 
     try:
-        person_bytes = np_to_bytes(person_img)
-        garment_bytes = np_to_bytes(garment_img)
+        # Convert image to bytes to send via Gradio client
+        _, img_encoded = cv2.imencode(".png", overlay_img)
+        img_bytes = img_encoded.tobytes()
 
-        # 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"
+        # Predict with the client
+        result = hunyuan_client.predict(img_bytes)
+        return result, "Success (3D try-on)"
     except Exception as e:
-        return None, f"Hunyuan3D API error: {e}"
+        return overlay_img, f"Hunyuan3D API error: {e}"
 
 # -----------------------------
 # Gradio UI