Update app.py

app.py

@@ -9,7 +9,6 @@ from gradio_client import Client, file
 
 # --- CONFIGURATION ---
 # 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):
@@ -59,13 +58,12 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
         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
+            False,             # Do not remove background
+            0.85,              # Foreground ratio
+            api_name="/generate"
         )
         
         print(f"-> Success! Received 3D model at: {result_path}")
@@ -88,12 +86,14 @@ 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.
+            # Note: TripoSR is purely image-to-3D, so these dropdowns 
+            # are currently just placeholders in this specific workflow, 
+            # but we keep them to prevent the code from crashing.
             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")
 
@@ -102,7 +102,9 @@ 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], outputs=[view3d, download])
+    
+    # --- FIXED LINE BELOW: Added 'b' back to inputs ---
+    btn2.click(generate_3d_avatar, inputs=[out_sketch, h, w, m, g, b], outputs=[view3d, download])
 
 if __name__ == "__main__":
     demo.launch(