Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 import cv2
 import mediapipe as mp
@@ -40,45 +39,51 @@ def create_3d_plot(vertices):
         scene=dict(
             xaxis_title='X',
             yaxis_title='Y',
-            zaxis_title='Z'
+            zaxis_title='Z',
+            aspectmode='cube'
         ),
-        margin=dict(l=0, r=0, b=0, t=0)
+        margin=dict(l=0, r=0, b=0, t=0),
+        showlegend=False
     )
     
     return fig
 
 def save_obj_and_display(frame):
     """Save frame as OBJ and return both file path and 3D visualization"""
-    if frame is None:
-        return None, None, "No frame provided"
-    
-    img, faces = process_frame(frame)
-    
-    if not faces:
-        return None, None, "No face detected in the frame"
-    
-    # Scale vertices
-    vertices = np.array(faces[0]) * 150
-    
-    # Create temporary directory if it doesn't exist
-    temp_dir = Path('temp')
-    temp_dir.mkdir(exist_ok=True)
-    
-    # Generate timestamp
-    timestamp = time.strftime("%Y%m%d-%H%M%S")
-    filename = f"face_mesh_{timestamp}"
-    
-    # Save OBJ file
-    obj_path = temp_dir / f"{filename}.obj"
-    success = detector.save_obj_file(vertices, filename)
-    
-    if not success:
-        return None, None, "Failed to save OBJ file"
-    
-    # Create 3D visualization
-    fig = create_3d_plot(vertices)
+    try:
+        if frame is None:
+            return None, None, "No frame provided"
+        
+        img, faces = process_frame(frame)
+        
+        if not faces:
+            return None, None, "No face detected in the frame"
+        
+        # Scale vertices
+        vertices = np.array(faces[0]) * 150
+        
+        # Create temporary directory if it doesn't exist
+        temp_dir = Path('temp')
+        temp_dir.mkdir(exist_ok=True)
+        
+        # Generate timestamp
+        timestamp = time.strftime("%Y%m%d-%H%M%S")
+        filename = f"face_mesh_{timestamp}"
+        
+        # Save OBJ file
+        obj_path = temp_dir / f"{filename}.obj"
+        success = detector.save_obj_file(vertices, filename)
+        
+        if not success:
+            return None, None, "Failed to save OBJ file"
+        
+        # Create 3D visualization
+        fig = create_3d_plot(vertices)
+        
+        return str(obj_path), fig, "Model generated successfully"
     
-    return str(obj_path), fig, "Model generated successfully"
+    except Exception as e:
+        return None, None, f"Error: {str(e)}"
 
 def cleanup_old_files():
     """Remove files older than 1 hour"""
@@ -105,8 +110,20 @@ def create_interface():
         
         with gr.Row():
             with gr.Column():
-                # Input methods
-                input_image = gr.Image(source="webcam", type="numpy")
+                # Input methods - using webcam and image input
+                with gr.Tab("Webcam"):
+                    input_webcam = gr.Image(
+                        label="Webcam Input", 
+                        type="numpy",
+                        sources="webcam"
+                    )
+                
+                with gr.Tab("Upload"):
+                    input_image = gr.Image(
+                        label="Upload Image",
+                        type="numpy"
+                    )
+                
                 generate_button = gr.Button("Generate 3D Model")
             
             with gr.Column():
@@ -115,13 +132,29 @@ def create_interface():
                 plot_3d = gr.Plot(label="3D Preview")
                 status_text = gr.Textbox(label="Status")
         
-        # Set up event handler
+        # Set up event handlers for both webcam and uploaded image
+        generate_button.click(
+            fn=save_obj_and_display,
+            inputs=[input_webcam],
+            outputs=[obj_file, plot_3d, status_text]
+        )
+        
         generate_button.click(
             fn=save_obj_and_display,
             inputs=[input_image],
             outputs=[obj_file, plot_3d, status_text]
         )
         
+        # Example section
+        gr.Markdown("""
+        ## Tips for Best Results
+        - Ensure good lighting
+        - Face the camera directly
+        - Keep a neutral expression
+        - Avoid extreme head rotations
+        - Stay within arm's length of the camera
+        """)
+        
         # Cleanup old files periodically
         cleanup_old_files()