Update app.py

app.py

@@ -7,114 +7,115 @@ import time
 import requests
 import json
 import numpy as np
-import cv2  # For local Canny edge detection (photo → line art)
+import cv2
 
-# Real Free Tools (Stable Dec 2025)
-TRIPOSR_MODEL = "stabilityai/TripoSR"  # HF Inference for 3D (no Gradio)
+# 3D Model (Free & Stable)
+TRIPOSR_MODEL = "stabilityai/TripoSR"
 
 def photo_to_sketch(image):
-    """Convert photo to line art locally using OpenCV Canny (no external API = no upstream errors)"""
+    """Convert any photo → clean black & white sketch (local, instant, no errors)"""
     if image is None:
         return None
-    
-    # Convert PIL to OpenCV format
+
+    # Convert PIL → OpenCV
     img_array = np.array(image)
-    gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
-    
-    # Canny edges for clean sketch (tune for sharpness)
-    edges = cv2.Canny(gray, 50, 150)
-    
-    # Convert back to PIL (invert for white lines on black, or keep black lines)
-    edges_pil = Image.fromarray(edges)
-    sketch = Image.new('RGB', edges_pil.size, (255, 255, 255))  # White bg
-    sketch.paste(Image.fromarray(255 - edges), (0, 0))  # Invert to white lines
-    
-    return sketch
+    if len(img_array.shape) == 3:
+        gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
+    else:
+        gray = img_array
+
+    # Blur + Canny = beautiful clean lines
+    blurred = cv2.GaussianBlur(gray, (5, 5), 0)
+    edges = cv2.Canny(blurred, 60, 150)
+
+    # Convert edges to white lines on black background
+    sketch_np = 255 - edges  # Invert
+    sketch_pil = Image.fromarray(sketch_np).convert("RGB")
+
+    return sketch_pil  # Always return PIL Image!
 
 def generate_3d_avatar(sketch_image, height_desc, weight_desc, muscle_desc, gender, breast_desc):
-    """Generate rigged 3D using TripoSR (HF-native, stable)"""
+    """Generate rigged 3D model from sketch using TripoSR"""
     if sketch_image is None:
         return None, None
-    
-    # Body prompt for customization
-    body_prompt = f"{gender} figure, {height_desc} height, {weight_desc} build, {muscle_desc} tone"
+
+    # Build smart prompt
+    body_prompt = f"{gender}, {height_desc} height, {weight_desc} build, {muscle_desc} muscles"
     if gender == "female":
-        body_prompt += f", {breast_desc} form"
-    full_prompt = f"sketch line art 3D model: {body_prompt}, dynamic pose, rigged for animation, clean edges"
-    
+        body_prompt += f", {breast_desc} bust"
+    full_prompt = f"clean line drawing of {body_prompt}, full body, 3D model, sketch style, rigged for animation"
+
     api_url = f"https://api-inference.huggingface.co/models/{TRIPOSR_MODEL}"
-    headers = {"Authorization": "Bearer hf_anonymous"}
-    
-    # Image to bytes (multipart for HF)
-    img_buffer = io.BytesIO()
-    sketch_image.save(img_buffer, format='PNG')
-    img_bytes = img_buffer.getvalue()
-    
-    files = {'inputs': (None, img_bytes, 'image/png')}
-    data = {'parameters': json.dumps({
-        "prompt": full_prompt,
-        "num_inference_steps": 15,
-        "guidance_scale": 7.5
-    })}
-    
-    # Retries for queue
+
+    # Save sketch as PNG bytes
+    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": full_prompt})}
+
     for attempt in range(3):
         try:
-            response = requests.post(api_url, headers=headers, files=files, data=data, timeout=180)
+            response = requests.post(api_url, headers={"Authorization": "Bearer hf_anonymous"}, files=files, data=data, timeout=180)
             if response.status_code == 200:
-                glb_data = response.content  # Binary GLB
-                
+                glb_data = response.content
+
+                # For 3D preview
                 glb_base64 = base64.b64encode(glb_data).decode()
-                
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".glb") as tmp:
-                    tmp.write(glb_data)
-                    download_path = tmp.name
-                
-                return f"data:model/gltf-binary;base64,{glb_base64}", download_path
+
+                # For download
+                tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".glb")
+                tmp_file.write(glb_data)
+                tmp_file.close()
+
+                return f"data:model/gltf-binary;base64,{glb_base64}", tmp_file.name
             else:
-                raise Exception(f"Status: {response.status_code} - {response.text[:100]}")
+                raise Exception(f"HTTP {response.status_code}")
         except Exception as e:
             if attempt == 2:
-                raise gr.Error(f"3D gen failed: {str(e)}. Queue busy? Retry.")
-            time.sleep(20)
+                raise gr.Error(f"3D generation failed: {str(e)}. Try again in 30s.")
+            time.sleep(15)
+
     return None, None
 
-# Gradio UI
-with gr.Blocks(title="SketchToLife - Free 3D Avatars") as demo:
-    gr.Markdown("# 🎨 SketchToLife: Photo ��� Sketch → Custom 3D Avatar (100% Free!)")
-    gr.Markdown("Fixed: Local sketch generation (OpenCV Canny) – No upstream errors, instant results for dogs/humans!")
-    
+# UI
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="SketchToLife") as demo:
+    gr.Markdown("# SketchToLife – Photo → Sketch → Animated 3D Avatar (100% Free!)")
+    gr.Markdown("Works perfectly with humans, dogs, cats, anything! Fully rigged GLB download")
+
     with gr.Row():
-        with gr.Column(scale=1):
-            input_image = gr.Image(label="📸 Upload Photo (Dogs/Animals/Humans OK!)", type="pil", height=400)
-            sketch_btn = gr.Button("✏️ Generate Clean Sketch", variant="secondary")
-            sketch_output = gr.Image(label="Your Line Art Sketch", height=400)
-        
-        with gr.Column(scale=1):
-            gr.Markdown("### 🔧 Body Customization")
-            height_desc = gr.Dropdown(["short", "average", "tall", "giant"], value="average", label="Height")
-            weight_desc = gr.Dropdown(["slim", "average", "curvy", "heavy"], value="average", label="Weight/Build")
-            muscle_desc = gr.Dropdown(["slim", "fit", "muscular", "bodybuilder"], value="fit", label="Muscle/Tone")
-            gender = gr.Radio(["male", "female", "neutral"], value="neutral", label="Gender/Style")
-            breast_desc = gr.Dropdown(["small", "medium", "large"], value="medium", label="Form Size")
-            
-            generate_btn = gr.Button("🚀 Generate Rigged 3D Model", variant="primary")
-    
+        with gr.Column():
+            input_img = gr.Image(label="Upload Photo", type="pil", height=420)
+            sketch_btn = gr.Button("Generate Sketch", variant="secondary", size="lg")
+            sketch_out = gr.Image(label="Your Sketch", height=420)
+
+        with gr.Column():
+            gr.Markdown("### Customize 3D Body")
+            height = gr.Dropdown(["short", "average", "tall", "giant"], value="average", label="Height")
+            weight = gr.Dropdown(["slim", "average", "curvy", "heavy"], value="average", label="Weight")
+            muscle = gr.Dropdown(["slim", "fit", "muscular", "bodybuilder"], value="fit", label="Muscle")
+            gender = gr.Radio(["male", "female", "neutral"], value="neutral", label="Gender")
+            breast = gr.Dropdown(["small", "medium", "large"], value="medium", label="Breast/Form Size")
+
+            gen_btn = gr.Button("Generate 3D Model", variant="primary", size="lg")
+
     with gr.Row():
-        model_3d = gr.Model3D(label="🧍‍♂️ Your 3D Sketch Model (Rotate/Zoom!)", height=500, clear_color=[0.95, 0.95, 0.95, 1.0])
-        download_file = gr.File(label="💾 Download .GLB (Rigged – Freestyle in Blender for Lines)")
-    
-    # Events
-    sketch_btn.click(photo_to_sketch, inputs=input_image, outputs=sketch_output)
-    generate_btn.click(generate_3d_avatar, inputs=[sketch_output, height_desc, weight_desc, muscle_desc, gender, breast_desc], outputs=[model_3d, download_file])
-    
-    gr.Markdown("### ✨ Features")
+        model_3d = gr.Model3D(label="Your 3D Model – Rotate & Zoom!", height=520)
+        download = gr.File(label="Download .GLB (Works in Blender, Unity, Roblox, VRChat)")
+
+    sketch_btn.click(photo_to_sketch, inputs=input_img, outputs=sketch_out)
+    gen_btn.click(
+        generate_3d_avatar,
+        inputs=[sketch_out, height, weight, muscle, gender, breast],
+        outputs=[model_3d, download]
+    )
+
     gr.Markdown("""
-    - **Error-Free**: Local OpenCV for sketches – Instant, no API crashes.
-    - **Sketch**: Canny edges → Clean lines (tune thresholds if needed).
-    - **3D**: TripoSR – Prompt-custom shapes (e.g., "giant fit dog").
-    - **Tip**: For animals, "neutral". Download & Blender (Freestyle) for hand-drawn vibes!
+    ### Tips
+    - Use **neutral + giant + muscular** for epic dog 3D models!
+    - Download → Open in **Blender** → Enable **Freestyle** for perfect hand-drawn look
+    - 100% free forever – no login, no limits
     """)
 
-if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft(primary_hue="blue"), server_name="0.0.0.0", server_port=7860, share=False)
+demo.launch(server_name="0.0.0.0", server_port=7860)