Update app.py

app.py

@@ -12,27 +12,42 @@ import cv2
 TRIPOSR_MODEL = "stabilityai/TripoSR"
 
 def photo_to_sketch(image):
+    """Instant local sketch – always returns PIL Image"""
     if image is None:
         return None
-    img_array = np.array(image)
-    if len(img_array.shape) == 3:
-        gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
-    else:
-        gray = img_array
-    blurred = cv2.GaussianBlur(gray, (5, 5), 0)
+
+    # Ensure we have a PIL 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
-    return Image.fromarray(sketch_np).convert("RGB")
+    sketch_pil = Image.fromarray(sketch_np)
+
+    # Return as RGB (required for TripoSR)
+    return sketch_pil.convert("RGB")
+
 
 def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
+    """Generate rigged 3D model from sketch"""
     if sketch_image is None:
         return None, None
 
+    # Build prompt
     body = f"{gender}, {height} height, {weight} build, {muscle} muscles"
     if gender == "female":
         body += f", {breast} bust"
-    prompt = f"clean black and white line drawing of {body}, full body, 3D model, rigged, sketch style"
+    prompt = f"clean line drawing of {body}, full body standing pose, black and white sketch, 3D model, rigged"
 
+    # Save sketch to bytes
     buffer = io.BytesIO()
     sketch_image.save(buffer, format="PNG")
     img_bytes = buffer.getvalue()
@@ -58,21 +73,24 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
                 tmp.close()
 
                 return f"data:model/gltf-binary;base64,{b64}", tmp.name
-        except:
+
+        except Exception as e:
             if attempt == 2:
-                raise gr.Error("3D generation failed – retry in 30s")
+                raise gr.Error("3D generation failed – retry in 30s (TripoSR is busy)")
             time.sleep(15)
+
     return None, None
 
-# ——— UI (theme goes in launch(), NOT in Blocks!) ———
+
+# =============== UI (Theme in launch() – Gradio 6+ fix) ===============
 with gr.Blocks(title="SketchToLife") as demo:
     gr.Markdown("# SketchToLife – Photo → Sketch → Animated 3D Avatar (100% Free!)")
-    gr.Markdown("Instant local sketch + TripoSR 3D → rigged GLB download")
+    gr.Markdown("Works with **humans, dogs, cats, anything** → Instant sketch + Rigged 3D download")
 
     with gr.Row():
         with gr.Column():
-            inp = gr.Image(label="Upload any photo (people, dogs…)", type="pil", height=420)
-            btn_sketch = gr.Button("Generate Sketch", variant="secondary")
+            inp = gr.Image(label="Upload Photo", type="pil", height=420)
+            btn1 = gr.Button("Generate Clean Sketch", variant="secondary", size="lg")
             out_sketch = gr.Image(label="Your Sketch", height=420)
 
         with gr.Column():
@@ -83,15 +101,16 @@ with gr.Blocks(title="SketchToLife") as demo:
             g = gr.Radio(["male", "female", "neutral"], value="neutral", label="Gender")
             b = gr.Dropdown(["small", "medium", "large"], value="medium", label="Breast/Form")
 
-            btn_3d = gr.Button("Generate 3D Model", variant="primary", size="lg")
+            btn2 = gr.Button("Generate 3D Model", variant="primary", size="lg")
 
     with gr.Row():
-        model3d = gr.Model3D(label="Your 3D Model – rotate & zoom", height=520)
+        view3d = gr.Model3D(label="Your 3D Model – Rotate & Zoom!", height=520)
         download = gr.File(label="Download .GLB (Blender / Unity / Roblox / VRChat)")
 
-    btn_sketch.click(photo_to_sketch, inputs=inp, outputs=out_sketch)
-    btn_3d.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g, b], outputs=[model3d, download])
+    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])
 
+# Theme goes HERE in Gradio 6+
 demo.launch(
     theme=gr.themes.Soft(primary_hue="blue"),
     server_name="0.0.0.0",