Update app.py

app.py

@@ -8,9 +8,9 @@ import time
 from gradio_client import Client, file
 
 # --- CONFIGURATION ---
-# 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"
+# We switched to "hysts/Shap-E" (OpenAI's model).
+# It is extremely reliable and rarely crashes compared to TripoSR/InstantMesh.
+REMOTE_MODEL_ID = "hysts/Shap-E"
 
 def photo_to_sketch(image):
     """Instant local sketch – always returns PIL Image"""
@@ -21,29 +21,25 @@ def photo_to_sketch(image):
     if isinstance(image, np.ndarray):
         image = Image.fromarray(image.astype('uint8'))
 
-    # Convert to grayscale
     gray = image.convert("L")
     img_array = np.array(gray)
 
-    # Blur + Canny edges
     blurred = cv2.GaussianBlur(img_array, (5, 5), 0)
     edges = cv2.Canny(blurred, 60, 150)
 
-    # White lines on black background
     sketch_np = 255 - edges
     sketch_pil = Image.fromarray(sketch_np)
     
-    print("-> Sketch Generated Successfully")
     return sketch_pil.convert("RGB")
 
 def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
-    """Generate rigged 3D model using the remote InstantMesh Space"""
+    """Generate 3D model using OpenAI Shap-E (Reliable Fallback)"""
     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!")
+        raise gr.Error("Please upload an image first!")
 
-    # 1. Save Sketch to a temporary file (Required for Client)
+    # 1. Save Sketch to a temporary file
     if isinstance(sketch_image, np.ndarray):
         sketch_image = Image.fromarray(sketch_image.astype('uint8'))
         
@@ -51,45 +47,38 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
     sketch_path = os.path.join(temp_dir, f"sketch_{int(time.time())}.png")
     sketch_image.save(sketch_path)
     
-    print(f"-> Saved temporary sketch to: {sketch_path}")
-
-    # 2. Connect to the Remote Space
+    # 2. Connect to Shap-E
     try:
         print(f"-> Connecting to {REMOTE_MODEL_ID}...")
         client = Client(REMOTE_MODEL_ID)
         
-        # 3. Send request to InstantMesh
-        # InstantMesh API: [Image, do_remove_background, sample_steps, seed]
-        print("-> Sending request to remote GPU...")
+        # 3. Send request (Shap-E API parameters)
+        print("-> Sending request to Shap-E...")
         
-        # We try the standard generate endpoint for InstantMesh
-        result_path = client.predict(
-            file(sketch_path), # Input image
-            True,              # Remove background? (Yes, safer)
-            30,                # Sample steps (30 is standard quality)
-            42,                # Seed
-            api_name="/generate" 
+        # Shap-E expects: [image, prompt_text, seed, guidance_scale, num_inference_steps]
+        # We leave prompt empty to force Image-to-3D mode
+        result = client.predict(
+            file(sketch_path), # Input Image
+            "",                # Text Prompt (Empty for Img-to-3D)
+            0,                 # Seed
+            15,                # Guidance Scale
+            64,                # Steps (64 is fast, 128 is better)
+            api_name="/image-to-3d"
         )
         
-        # 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
+        # Result is just the file path string
+        print(f"-> Success! Received model: {result}")
+        return result, result
 
     except Exception as e:
         print(f"-> Connection Error: {e}")
-        # Detailed error message for debugging
-        raise gr.Error(f"Remote AI Error: {e}. The InstantMesh space might be busy or changing APIs.")
+        # If even Shap-E fails, we raise a clear error
+        raise gr.Error(f"All 3D models are currently busy. Error: {e}")
 
 # =============== UI ===============
 with gr.Blocks(title="SketchToLife") as demo:
-    gr.Markdown("# SketchToLife – Photo → Sketch → 3D Avatar (InstantMesh)")
-    gr.Markdown(f"**Status:** Connected to `{REMOTE_MODEL_ID}` (High Quality GPU Model)")
+    gr.Markdown("# SketchToLife – Photo → Sketch → 3D Avatar")
+    gr.Markdown(f"**Status:** Connected to `OpenAI Shap-E` (High Stability Backup).")
 
     with gr.Row():
         with gr.Column():
@@ -99,7 +88,7 @@ with gr.Blocks(title="SketchToLife") as demo:
 
         with gr.Column():
             gr.Markdown("### Customize 3D Body")
-            # Placeholders to keep UI consistent and prevent argument errors
+            # Placeholders to 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")
@@ -114,7 +103,7 @@ with gr.Blocks(title="SketchToLife") as demo:
 
     btn1.click(photo_to_sketch, inputs=inp, outputs=out_sketch)
     
-    # Passing all 6 inputs to match the function definition
+    # 6 inputs -> 6 arguments
     btn2.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g, b], outputs=[view3d, download])
 
 if __name__ == "__main__":