Update app.py

app.py

@@ -41,6 +41,11 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
     if sketch_image is None:
         return None, None
 
+    # --- FIX: Convert NumPy array (from Gradio) back to PIL Image ---
+    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":
@@ -76,13 +81,14 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
 
         except Exception as e:
             if attempt == 2:
-                raise gr.Error("3D generation failed – retry in 30s (TripoSR is busy)")
+                # Raise Gradio error so it appears in the UI
+                raise gr.Error(f"3D generation failed: {e}")
             time.sleep(15)
 
     return None, None
 
 
-# =============== UI (Theme in launch() – Gradio 6+ fix) ===============
+# =============== UI ===============
 with gr.Blocks(title="SketchToLife") as demo:
     gr.Markdown("# SketchToLife – Photo → Sketch → Animated 3D Avatar (100% Free!)")
     gr.Markdown("Works with **humans, dogs, cats, anything** → Instant sketch + Rigged 3D download")
@@ -91,7 +97,9 @@ with gr.Blocks(title="SketchToLife") as demo:
         with gr.Column():
             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)
+            # Explicitly setting type="pil" here is good practice, 
+            # but the function fix above handles it regardless.
+            out_sketch = gr.Image(label="Your Sketch", height=420, type="pil")
 
         with gr.Column():
             gr.Markdown("### Customize 3D Body")
@@ -110,9 +118,9 @@ with gr.Blocks(title="SketchToLife") as demo:
     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",
-    server_port=7860
-)
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        ssr_mode=False  # Disabled SSR as hinted in your logs
+    )