Update app.py

app.py

@@ -8,8 +8,9 @@ import time
 from gradio_client import Client, file
 
 # --- CONFIGURATION ---
-# We use the Gradio Client to connect to the official GPU-powered Space
-REMOTE_MODEL_ID = "stabilityai/TripoSR"
+# The official TripoSR space is down (Runtime Error). 
+# We are switching to "InstantMesh" which is newer, better quality, and currently ONLINE.
+REMOTE_MODEL_ID = "TencentARC/InstantMesh"
 
 def photo_to_sketch(image):
     """Instant local sketch – always returns PIL Image"""
@@ -36,8 +37,8 @@ def photo_to_sketch(image):
     return sketch_pil.convert("RGB")
 
 def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
-    """Generate rigged 3D model using the remote TripoSR Space"""
-    print("-> Starting 3D Generation...")
+    """Generate rigged 3D model using the remote InstantMesh Space"""
+    print(f"-> Starting 3D Generation using {REMOTE_MODEL_ID}...")
     
     if sketch_image is None:
         raise gr.Error("Please upload an image and generate a sketch first!")
@@ -57,26 +58,38 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
         print(f"-> Connecting to {REMOTE_MODEL_ID}...")
         client = Client(REMOTE_MODEL_ID)
         
-        # 3. Send the image to the remote model
+        # 3. Send request to InstantMesh
+        # InstantMesh API: [Image, do_remove_background, sample_steps, seed]
         print("-> Sending request to remote GPU...")
+        
+        # We try the standard generate endpoint for InstantMesh
         result_path = client.predict(
-            file(sketch_path), # The image file
-            False,             # Do not remove background
-            0.85,              # Foreground ratio
-            api_name="/generate"
+            file(sketch_path), # Input image
+            True,              # Remove background? (Yes, safer)
+            30,                # Sample steps (30 is standard quality)
+            42,                # Seed
+            api_name="/generate" 
         )
         
-        print(f"-> Success! Received 3D model at: {result_path}")
-        return result_path, result_path
+        # InstantMesh returns a tuple, usually the GLB is the first item or named path
+        # If it returns a tuple/list, we take the first file path.
+        if isinstance(result_path, (list, tuple)):
+            final_glb = result_path[0]
+        else:
+            final_glb = result_path
+            
+        print(f"-> Success! Received 3D model at: {final_glb}")
+        return final_glb, final_glb
 
     except Exception as e:
         print(f"-> Connection Error: {e}")
-        raise gr.Error(f"Failed to connect to remote model. The StabilityAI space might be busy. Error: {e}")
+        # Detailed error message for debugging
+        raise gr.Error(f"Remote AI Error: {e}. The InstantMesh space might be busy or changing APIs.")
 
 # =============== UI ===============
 with gr.Blocks(title="SketchToLife") as demo:
-    gr.Markdown("# SketchToLife – Photo → Sketch → Animated 3D Avatar")
-    gr.Markdown("**Status:** Connected to `stabilityai/TripoSR` via Gradio Client.")
+    gr.Markdown("# SketchToLife – Photo → Sketch → 3D Avatar (InstantMesh)")
+    gr.Markdown(f"**Status:** Connected to `{REMOTE_MODEL_ID}` (High Quality GPU Model)")
 
     with gr.Row():
         with gr.Column():
@@ -86,9 +99,7 @@ with gr.Blocks(title="SketchToLife") as demo:
 
         with gr.Column():
             gr.Markdown("### Customize 3D Body")
-            # Note: TripoSR is purely image-to-3D, so these dropdowns 
-            # are currently just placeholders in this specific workflow, 
-            # but we keep them to prevent the code from crashing.
+            # Placeholders to keep UI consistent and prevent argument errors
             h = gr.Dropdown(["short", "average", "tall", "giant"], value="average", label="Height")
             w = gr.Dropdown(["slim", "average", "curvy", "heavy"], value="average", label="Weight")
             m = gr.Dropdown(["slim", "fit", "muscular", "bodybuilder"], value="fit", label="Muscle")
@@ -103,7 +114,7 @@ with gr.Blocks(title="SketchToLife") as demo:
 
     btn1.click(photo_to_sketch, inputs=inp, outputs=out_sketch)
     
-    # --- FIXED LINE BELOW: Added 'b' back to inputs ---
+    # Passing all 6 inputs to match the function definition
     btn2.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g, b], outputs=[view3d, download])
 
 if __name__ == "__main__":