Fix upside-down mesh rotation, add 3D preview

app.py

@@ -112,6 +112,10 @@ def reconstruct_body(image: np.ndarray) -> tuple:
         # Create trimesh mesh
         mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
 
+        # Rotate 180° around X-axis to fix upside-down orientation
+        rotation = trimesh.transformations.rotation_matrix(np.pi, [1, 0, 0])
+        mesh.apply_transform(rotation)
+
         # Save GLB
         output_dir = tempfile.mkdtemp()
         glb_path = f"{output_dir}/body_{uuid.uuid4().hex[:8]}.glb"
@@ -128,17 +132,19 @@ def reconstruct_body(image: np.ndarray) -> tuple:
 # Gradio Interface
 with gr.Blocks(title="SAM 3D Body MCP") as demo:
     gr.Markdown("# 🧍 SAM 3D Body MCP Server\n**Image → 3D Human Mesh (GLB)**")
-    
+
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(label="Input Image", type="numpy")
             btn = gr.Button("🎯 Reconstruct", variant="primary")
-        
+
         with gr.Column():
-            output_file = gr.File(label="3D Mesh (GLB)")
+            output_model = gr.Model3D(label="3D Preview")
+            output_file = gr.File(label="Download GLB")
             status = gr.Textbox(label="Status")
-    
-    btn.click(reconstruct_body, inputs=[input_image], outputs=[output_file, status])
+
+    btn.click(reconstruct_body, inputs=[input_image], outputs=[output_model, status])
+    output_model.change(lambda x: x, inputs=[output_model], outputs=[output_file])
     
     gr.Markdown("""
     ---