Update app.py

app.py

@@ -1,9 +1,7 @@
-# app.py
 import gradio as gr
 import cv2
 import numpy as np
 from gradio_client import Client
-import os
 
 # -----------------------------
 # Local "try-on" function
@@ -15,7 +13,6 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     h_person, w_person = person_img.shape[:2]
     h_garment, w_garment = garment_img.shape[:2]
 
-    # Resize garment to fit person width
     scale = w_person / w_garment
     new_w = int(w_garment * scale)
     new_h = int(h_garment * scale)
@@ -25,12 +22,12 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
     y_offset = 0
     x_offset = max(0, (w_person - new_w) // 2)
 
-    if garment_resized.shape[2] == 4:  # RGBA
+    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:  # RGB
+    else:
         overlay[y_offset:y_offset+new_h, x_offset:x_offset+new_w] = garment_resized
 
     info = "Success (local simulation)"
@@ -42,18 +39,14 @@ def tryon_local(person_img, garment_img, seed, randomize_seed):
 hunyuan_client = Client("tencent/Hunyuan3D-2.1")
 
 def tryon_to_3d(person_img, garment_img, seed, randomize_seed):
-    # Run local try-on simulation first
+    # Run local simulation 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 bytes for API
-        _, img_encoded = cv2.imencode(".png", tryon_img)
-        img_bytes = img_encoded.tobytes()
-
-        # Call Hunyuan3D API
-        result_3d = hunyuan_client.predict(img_bytes, api_name="/predict")
+        # Call the Hunyuan3D predict function directly
+        result_3d = hunyuan_client.predict(person_img, garment_img)
         return result_3d, tryon_info
     except Exception as e:
         return None, f"Hunyuan3D API error: {e}"
@@ -92,5 +85,4 @@ with gr.Blocks(css=css) as app:
         outputs=[output_img, result_info]
     )
 
-# Launch Gradio app
-app.launch(server_name="0.0.0.0", server_port=7860, share=False)
+app.launch()