Update app.py

app.py

@@ -1,23 +1,16 @@
 import gradio as gr
-import io
 from PIL import Image
-import tempfile
-import base64
-import time
-import requests
-import json
 import numpy as np
 import cv2
 import os
+import tempfile
+import time
+from gradio_client import Client, file
 
 # --- CONFIGURATION ---
-# SECURE SETUP: Read token from Environment Variables.
-HF_TOKEN = os.getenv("HF_TOKEN") 
-TRIPOSR_MODEL = "stabilityai/TripoSR"
-
-# NEW URL: The old "api-inference" URL is deprecated (Error 410). 
-# We use the new Router URL.
-API_URL = f"https://router.huggingface.co/hf-inference/models/{TRIPOSR_MODEL}"
+# We use the Gradio Client to connect to the official GPU-powered Space
+# This bypasses the "404" API error.
+REMOTE_MODEL_ID = "stabilityai/TripoSR"
 
 def photo_to_sketch(image):
     """Instant local sketch – always returns PIL Image"""
@@ -28,94 +21,64 @@ 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 from sketch"""
+    """Generate rigged 3D model using the remote TripoSR Space"""
     print("-> Starting 3D Generation...")
     
     if sketch_image is None:
         raise gr.Error("Please upload an image and generate a sketch first!")
 
+    # 1. Save Sketch to a temporary file (Required for Client)
     if isinstance(sketch_image, np.ndarray):
         sketch_image = Image.fromarray(sketch_image.astype('uint8'))
-
-    # Build prompt
-    body = f"{gender}, {height} height, {weight} build, {muscle} muscles"
-    if gender == "female":
-        body += f", {breast} bust"
-    prompt = f"clean line drawing of {body}, full body standing pose, black and white sketch, 3D model, rigged"
-
-    print(f"-> Prompt: {prompt}")
-
-    buffer = io.BytesIO()
-    sketch_image.save(buffer, format="PNG")
-    img_bytes = buffer.getvalue()
-
-    files = {"inputs": ("sketch.png", img_bytes, "image/png")}
-    data = {"parameters": json.dumps({"prompt": prompt})}
+        
+    temp_dir = tempfile.gettempdir()
+    sketch_path = os.path.join(temp_dir, f"sketch_{int(time.time())}.png")
+    sketch_image.save(sketch_path)
     
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
-
-    # Retry Loop
-    for attempt in range(1, 4):
-        try:
-            print(f"-> Attempt {attempt}/3: Sending request to {API_URL}...")
-            r = requests.post(
-                API_URL,
-                headers=headers,
-                files=files,
-                data=data,
-                timeout=120
-            )
-            
-            print(f"-> Status Code: {r.status_code}")
-            
-            if r.status_code == 200:
-                print("-> Success! Processing 3D file...")
-                glb = r.content
-                b64 = base64.b64encode(glb).decode()
-
-                tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".glb")
-                tmp.write(glb)
-                tmp.close()
-
-                return f"data:model/gltf-binary;base64,{b64}", tmp.name
-            
-            elif r.status_code == 503:
-                print("-> Model is loading (503). Waiting 20s...")
-                time.sleep(20)
-            elif r.status_code == 410:
-                print("-> Error 410: Endpoint deprecated. Trying fallback...")
-                # Fallback to standard inference if router fails, though router is the fix.
-                time.sleep(5)
-            elif r.status_code == 401:
-                 raise gr.Error("Authentication Failed. Check your HF_TOKEN secret.")
-            else:
-                print(f"-> API Error: {r.text}")
-
-        except Exception as e:
-            print(f"-> Connection Error: {e}")
-            time.sleep(5)
-
-    print("-> All attempts failed.")
-    raise gr.Error("Generation Failed. The model is busy. If this persists, the free API tier for TripoSR might be overloaded.")
+    print(f"-> Saved temporary sketch to: {sketch_path}")
+
+    # 2. Connect to the Remote Space
+    try:
+        print(f"-> Connecting to {REMOTE_MODEL_ID}...")
+        client = Client(REMOTE_MODEL_ID)
+        
+        # 3. Send the image to the remote model
+        # Note: TripoSR inputs are usually [Image, Remove Background Bool, Foreground Ratio]
+        print("-> Sending request to remote GPU...")
+        result_path = client.predict(
+            file(sketch_path), # The image file
+            False,             # Do not remove background (our sketch is already clean)
+            0.85,              # Foreground ratio (default)
+            api_name="/generate" # Standard endpoint name for TripoSR
+        )
+        
+        print(f"-> Success! Received 3D model at: {result_path}")
+        return result_path, result_path
+
+    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}")
 
 # =============== UI ===============
 with gr.Blocks(title="SketchToLife") as demo:
     gr.Markdown("# SketchToLife – Photo → Sketch → Animated 3D Avatar")
-    gr.Markdown("Works with **humans, dogs, cats, anything** → Instant sketch + Rigged 3D download")
+    gr.Markdown("**Status:** Connected to `stabilityai/TripoSR` via Gradio Client.")
 
     with gr.Row():
         with gr.Column():
@@ -125,12 +88,13 @@ with gr.Blocks(title="SketchToLife") as demo:
 
         with gr.Column():
             gr.Markdown("### Customize 3D Body")
+            # These dropdowns are placeholders for the prompt logic (if we were using text-to-3d), 
+            # but TripoSR is purely Image-to-3D. We keep them for UI layout.
             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")
             g = gr.Radio(["male", "female", "neutral"], value="neutral", label="Gender")
-            b = gr.Dropdown(["small", "medium", "large"], value="medium", label="Breast/Form")
-
+            
             btn2 = gr.Button("Generate 3D Model", variant="primary", size="lg")
 
     with gr.Row():
@@ -138,7 +102,7 @@ with gr.Blocks(title="SketchToLife") as demo:
         download = gr.File(label="Download .GLB")
 
     btn1.click(photo_to_sketch, inputs=inp, outputs=out_sketch)
-    btn2.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g, b], outputs=[view3d, download])
+    btn2.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g], outputs=[view3d, download])
 
 if __name__ == "__main__":
     demo.launch(