Slider

app.py

@@ -123,7 +123,7 @@ def compute_faces_overlay(image):
 # -----------------------------
 # New Facemesh Functions (with connected red lines and mask output)
 # -----------------------------
-def compute_facemesh_overlay(image):
+def compute_facemesh_overlay(image, confidence=0.5):
     """
     Uses MediaPipe Face Mesh to detect and draw facial landmarks.
     Draws green dots for landmarks and connects them with thin red lines.
@@ -137,9 +137,9 @@ def compute_facemesh_overlay(image):
     annotated = frame_bgr.copy()
     mask = np.zeros_like(frame_bgr)
     
-    # Initialize Face Mesh in static mode
+    # Initialize Face Mesh in static mode with adjustable confidence
     face_mesh = mp.solutions.face_mesh.FaceMesh(
-        static_image_mode=True, max_num_faces=1, refine_landmarks=True, min_detection_confidence=0.5
+        static_image_mode=True, max_num_faces=1, refine_landmarks=True, min_detection_confidence=confidence
     )
     results = face_mesh.process(cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB))
     
@@ -169,8 +169,8 @@ def compute_facemesh_overlay(image):
     face_mesh.close()
     return annotated, mask, text
 
-def analyze_facemesh(image):
-    annotated_image, mask_image, text = compute_facemesh_overlay(image)
+def analyze_facemesh(image, confidence):
+    annotated_image, mask_image, text = compute_facemesh_overlay(image, confidence)
     return (
         annotated_image,
         mask_image,
@@ -286,14 +286,22 @@ faces_interface = gr.Interface(
 
 facemesh_interface = gr.Interface(
     fn=analyze_facemesh,
-    inputs=gr.Image(label="Upload an Image for Facemesh"),
+    inputs=[
+        gr.Image(label="Upload an Image for Facemesh"),
+        gr.Slider(0.0, 1.0, value=0.5, label="Detection Confidence", elem_id="confidence_slider")
+    ],
     outputs=[
         gr.Image(type="numpy", label="Annotated Output"),
         gr.Image(type="numpy", label="Mask Output"),
         gr.HTML(label="Facemesh Analysis")
     ],
     title="<div style='color:#00ff00;'>Facemesh",
-    description="<div style='color:#00ff00;'>Detects facial landmarks using MediaPipe Face Mesh.</div>",
+    description="""
+    <div style='color:#00ff00;'>
+        Detects facial landmarks using MediaPipe Face Mesh.
+        <button onclick="document.getElementById('confidence_slider').value = 0.5; document.getElementById('confidence_slider').dispatchEvent(new Event('change'))" style="margin-left:10px;">Reset to Default</button>
+    </div>
+    """,
     examples=SAMPLE_IMAGES,
     live=False
 )