Fix slider performance and version string

app.py

@@ -80,6 +80,7 @@ MIDLINE_BOTTOM = 152
 # Image preprocessing helpers
 # ---------------------------------------------------------------------------
 
+
 def _normalize_to_bgr(image: np.ndarray) -> np.ndarray:
     """Convert any input image format (RGBA, grayscale, etc.) to BGR uint8."""
     if image is None:
@@ -163,8 +164,7 @@ def _detect_face_with_hints(image_bgr: np.ndarray) -> tuple[FaceLandmarks | None
     # Check if image is mostly black (too dark even after auto-adjust)
     if float(np.mean(gray)) < 30:
         return None, (
-            "No face detected -- the image appears very dark. "
-            "Try a photo with better lighting."
+            "No face detected -- the image appears very dark. Try a photo with better lighting."
         )
 
     # Check for very low contrast (washed out)
@@ -178,8 +178,7 @@ def _detect_face_with_hints(image_bgr: np.ndarray) -> tuple[FaceLandmarks | None
     aspect = w / max(h, 1)
     if aspect > 2.5 or aspect < 0.4:
         return None, (
-            "No face detected -- unusual aspect ratio. "
-            "Use a standard portrait or headshot photo."
+            "No face detected -- unusual aspect ratio. Use a standard portrait or headshot photo."
         )
 
     return None, (
@@ -193,6 +192,7 @@ def _detect_face_with_hints(image_bgr: np.ndarray) -> tuple[FaceLandmarks | None
 # Symmetry analysis
 # ---------------------------------------------------------------------------
 
+
 def compute_symmetry_score(
     face: FaceLandmarks,
 ) -> tuple[float, dict[str, float]]:
@@ -212,7 +212,7 @@ def compute_symmetry_score(
     coords = face.pixel_coords  # (478, 2) -- property, not method
 
     # Compute facial midline from forehead top (10) and chin bottom (152)
-    mid_top = coords[MIDLINE_TOP]    # (2,)
+    mid_top = coords[MIDLINE_TOP]  # (2,)
     mid_bot = coords[MIDLINE_BOTTOM]  # (2,)
 
     # Midline direction vector and unit normal
@@ -297,18 +297,24 @@ def render_symmetry_overlay(
 
     # Small label at midline top
     cv2.putText(
-        canvas, "midline", (int(mid_top[0]) + 5, int(mid_top[1]) - 5),
-        cv2.FONT_HERSHEY_SIMPLEX, 0.4, (255, 200, 0), 1, cv2.LINE_AA,
+        canvas,
+        "midline",
+        (int(mid_top[0]) + 5, int(mid_top[1]) - 5),
+        cv2.FONT_HERSHEY_SIMPLEX,
+        0.4,
+        (255, 200, 0),
+        1,
+        cv2.LINE_AA,
     )
 
     def _score_color(score: float) -> tuple[int, int, int]:
         """BGR color based on symmetry score."""
         if score >= 80:
-            return (0, 200, 0)    # green
+            return (0, 200, 0)  # green
         elif score >= 50:
             return (0, 200, 220)  # yellow (BGR)
         else:
-            return (0, 0, 220)    # red
+            return (0, 0, 220)  # red
 
     for region, pairs in SYMMETRY_PAIRS.items():
         score = region_scores.get(region, 0.0)
@@ -351,8 +357,14 @@ def render_symmetry_overlay(
 
         label = f"{region}: {score:.0f}"
         cv2.putText(
-            canvas, label, label_pos,
-            cv2.FONT_HERSHEY_SIMPLEX, 0.4, color, 1, cv2.LINE_AA,
+            canvas,
+            label,
+            label_pos,
+            cv2.FONT_HERSHEY_SIMPLEX,
+            0.4,
+            color,
+            1,
+            cv2.LINE_AA,
         )
 
     return canvas
@@ -379,6 +391,7 @@ def _format_symmetry_text(
 # Symmetry tab callbacks
 # ---------------------------------------------------------------------------
 
+
 def analyze_symmetry(image_rgb: np.ndarray):
     """Analyze facial symmetry from an uploaded photo."""
     if image_rgb is None:
@@ -461,6 +474,7 @@ def analyze_symmetry_comparison(
 # Core pipeline functions
 # ---------------------------------------------------------------------------
 
+
 def warp_image_tps(image, src_pts, dst_pts):
     """Thin-plate spline warp (CPU only)."""
     from landmarkdiff.synthetic.tps_warp import warp_image_tps as _warp
@@ -511,7 +525,10 @@ def process_image(image_rgb, procedure, intensity):
         if hint:
             return image_rgb_512, image_rgb_512, image_rgb_512, image_rgb_512, hint
         return (
-            image_rgb_512, image_rgb_512, image_rgb_512, image_rgb_512,
+            image_rgb_512,
+            image_rgb_512,
+            image_rgb_512,
+            image_rgb_512,
             "No face detected. Try a clearer, well-lit frontal photo.",
         )
 
@@ -527,9 +544,7 @@ def process_image(image_rgb, procedure, intensity):
         composited = mask_composite(warped, image_bgr, mask)
         composited_rgb = cv2.cvtColor(composited, cv2.COLOR_BGR2RGB)
 
-        displacement = np.mean(
-            np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1)
-        )
+        displacement = np.mean(np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1))
         elapsed = time.monotonic() - t0
 
         # Compute symmetry for original and predicted result
@@ -615,8 +630,7 @@ def intensity_sweep(image_rgb, procedure):
 # ---------------------------------------------------------------------------
 
 _proc_table = "\n".join(
-    f"| {name.replace('_', ' ').title()} | {desc} |"
-    for name, desc in PROCEDURE_INFO.items()
+    f"| {name.replace('_', ' ').title()} | {desc} |" for name, desc in PROCEDURE_INFO.items()
 )
 
 with gr.Blocks(
@@ -639,10 +653,16 @@ with gr.Blocks(
             with gr.Column(scale=1):
                 input_image = gr.Image(label="Face Photo", type="numpy", height=350)
                 procedure = gr.Radio(
-                    choices=PROCEDURES, value="rhinoplasty", label="Procedure",
+                    choices=PROCEDURES,
+                    value="rhinoplasty",
+                    label="Procedure",
                 )
                 intensity = gr.Slider(
-                    0, 100, 50, step=1, label="Intensity (%)",
+                    0,
+                    100,
+                    50,
+                    step=1,
+                    label="Intensity (%)",
                     info="0 = no change, 100 = maximum",
                 )
                 run_btn = gr.Button("Generate", variant="primary", size="lg")
@@ -666,8 +686,12 @@ with gr.Blocks(
         outputs = [out_wireframe, out_mask, out_result, out_original, info_box]
         _inputs = [input_image, procedure, intensity]
         run_btn.click(fn=process_image, inputs=_inputs, outputs=outputs)
-        for trigger in _inputs:
-            trigger.change(fn=process_image, inputs=_inputs, outputs=outputs)
+        # Auto-trigger on image upload and procedure change, but not on every
+        # slider tick during drag (each tick would re-run TPS on free CPU,
+        # causing severe lag). Use .release so it fires once on mouse-up.
+        input_image.change(fn=process_image, inputs=_inputs, outputs=outputs)
+        procedure.change(fn=process_image, inputs=_inputs, outputs=outputs)
+        intensity.release(fn=process_image, inputs=_inputs, outputs=outputs)
 
     # -- Tab 2: Compare Procedures --
     with gr.Tab("Compare All"):
@@ -694,7 +718,8 @@ with gr.Blocks(
         if EXAMPLE_IMAGES:
             gr.Examples(
                 examples=[[str(p)] for p in EXAMPLE_IMAGES],
-                inputs=[cmp_image], label="Examples",
+                inputs=[cmp_image],
+                label="Examples",
             )
 
         cmp_btn.click(fn=compare_procedures, inputs=[cmp_image, cmp_intensity], outputs=cmp_outputs)
@@ -713,7 +738,8 @@ with gr.Blocks(
         if EXAMPLE_IMAGES:
             gr.Examples(
                 examples=[[str(p)] for p in EXAMPLE_IMAGES],
-                inputs=[sweep_image], label="Examples",
+                inputs=[sweep_image],
+                label="Examples",
             )
 
         sweep_btn.click(
@@ -741,22 +767,29 @@ with gr.Blocks(
                 with gr.Row():
                     with gr.Column(scale=1):
                         sym_image = gr.Image(
-                            label="Face Photo", type="numpy", height=350,
+                            label="Face Photo",
+                            type="numpy",
+                            height=350,
                         )
                         sym_btn = gr.Button(
-                            "Analyze Symmetry", variant="primary", size="lg",
+                            "Analyze Symmetry",
+                            variant="primary",
+                            size="lg",
                         )
                     with gr.Column(scale=1):
                         sym_overlay = gr.Image(label="Symmetry Overlay", height=350)
 
                 sym_scores_box = gr.Textbox(
-                    label="Symmetry Scores", lines=8, interactive=False,
+                    label="Symmetry Scores",
+                    lines=8,
+                    interactive=False,
                 )
 
                 if EXAMPLE_IMAGES:
                     gr.Examples(
                         examples=[[str(p)] for p in EXAMPLE_IMAGES],
-                        inputs=[sym_image], label="Examples",
+                        inputs=[sym_image],
+                        label="Examples",
                     )
 
                 sym_btn.click(
@@ -773,23 +806,33 @@ with gr.Blocks(
                 )
                 with gr.Row():
                     sym_pre_image = gr.Image(
-                        label="Pre-Procedure", type="numpy", height=300,
+                        label="Pre-Procedure",
+                        type="numpy",
+                        height=300,
                     )
                     sym_post_image = gr.Image(
-                        label="Post-Procedure", type="numpy", height=300,
+                        label="Post-Procedure",
+                        type="numpy",
+                        height=300,
                     )
                 sym_cmp_btn = gr.Button(
-                    "Compare Symmetry", variant="primary", size="lg",
+                    "Compare Symmetry",
+                    variant="primary",
+                    size="lg",
                 )
                 with gr.Row():
                     sym_pre_overlay = gr.Image(
-                        label="Pre Symmetry Overlay", height=300,
+                        label="Pre Symmetry Overlay",
+                        height=300,
                     )
                     sym_post_overlay = gr.Image(
-                        label="Post Symmetry Overlay", height=300,
+                        label="Post Symmetry Overlay",
+                        height=300,
                     )
                 sym_cmp_box = gr.Textbox(
-                    label="Comparison", lines=14, interactive=False,
+                    label="Comparison",
+                    lines=14,
+                    interactive=False,
                 )
 
                 sym_cmp_btn.click(
@@ -800,7 +843,7 @@ with gr.Blocks(
 
     gr.Markdown(
         f"<div style='text-align:center;color:#999;font-size:0.8em;padding:8px'>"
-        f"LandmarkDiff v0.2.3 | TPS on CPU | MediaPipe 478-point mesh | "
+        f"LandmarkDiff v0.2.2 | TPS on CPU | MediaPipe 478-point mesh | "
         f"<a href='{GITHUB_URL}'>GitHub</a> | MIT License</div>"
     )