fix: eye gaze toggle now properly controls gaze line visibility

.gitignore

@@ -45,3 +45,4 @@ test_focus_guard.db
 __pycache__/
 docs/
 docs
+LOCAL_TESTING.md

main.py

@@ -65,8 +65,8 @@ def _draw_polyline(frame, lm, indices, w, h, color, thickness):
         cv2.line(frame, _lm_px(lm, indices[i], w, h), _lm_px(lm, indices[i + 1], w, h), color, thickness, cv2.LINE_AA)
 
 
-def _draw_face_mesh(frame, lm, w, h):
-    """Draw tessellation, contours, eyebrows, nose, lips, eyes, irises, gaze lines."""
+def _draw_face_mesh(frame, lm, w, h, draw_gaze_lines=False):
+    """Draw tessellation, contours, eyebrows, nose, lips, eyes, irises, and optionally gaze lines."""
     # Tessellation (gray triangular grid, semi-transparent)
     overlay = frame.copy()
     for s, e in _TESSELATION_CONNS:
@@ -92,7 +92,7 @@ def _draw_face_mesh(frame, lm, w, h):
     for indices in [_LEFT_EAR_POINTS, _RIGHT_EAR_POINTS]:
         for idx in indices:
             cv2.circle(frame, _lm_px(lm, idx, w, h), 3, (0, 255, 255), -1, cv2.LINE_AA)
-    # Irises + gaze lines
+    # Irises (always draw) + gaze lines (only when eye gaze is enabled)
     for iris_idx, eye_inner, eye_outer in [
         (FaceMeshDetector.LEFT_IRIS_INDICES, 133, 33),
         (FaceMeshDetector.RIGHT_IRIS_INDICES, 362, 263),
@@ -104,10 +104,11 @@ def _draw_face_mesh(frame, lm, w, h):
             radius = max(int(np.mean(radii)), 2)
             cv2.circle(frame, tuple(center), radius, _MAGENTA, 2, cv2.LINE_AA)
             cv2.circle(frame, tuple(center), 2, _WHITE, -1, cv2.LINE_AA)
-        eye_cx = int((lm[eye_inner, 0] + lm[eye_outer, 0]) / 2.0 * w)
-        eye_cy = int((lm[eye_inner, 1] + lm[eye_outer, 1]) / 2.0 * h)
-        dx, dy = center[0] - eye_cx, center[1] - eye_cy
-        cv2.line(frame, tuple(center), (int(center[0] + dx * 3), int(center[1] + dy * 3)), _RED, 1, cv2.LINE_AA)
+        if draw_gaze_lines:
+            eye_cx = int((lm[eye_inner, 0] + lm[eye_outer, 0]) / 2.0 * w)
+            eye_cy = int((lm[eye_inner, 1] + lm[eye_outer, 1]) / 2.0 * h)
+            dx, dy = center[0] - eye_cx, center[1] - eye_cy
+            cv2.line(frame, tuple(center), (int(center[0] + dx * 3), int(center[1] + dy * 3)), _RED, 1, cv2.LINE_AA)
 
 
 def _draw_hud(frame, result, model_name):
@@ -356,8 +357,9 @@ class VideoTransformTrack(VideoStreamTrack):
                 # Draw face mesh + HUD on the video frame
                 h_f, w_f = img.shape[:2]
                 lm = out.get("landmarks")
+                eye_gaze_enabled = _l2cs_boost_enabled or model_name == "l2cs"
                 if lm is not None:
-                    _draw_face_mesh(img, lm, w_f, h_f)
+                    _draw_face_mesh(img, lm, w_f, h_f, draw_gaze_lines=eye_gaze_enabled)
                 _draw_hud(img, out, model_name)
             else:
                 is_focused = False
@@ -959,6 +961,7 @@ async def websocket_endpoint(websocket: WebSocket):
                     "model": model_name,
                     "fc": frame_count,
                     "frame_count": frame_count,
+                    "eye_gaze_enabled": _l2cs_boost_enabled or model_name == "l2cs",
                 }
                 if out is not None:
                     if out.get("yaw") is not None:

src/utils/VideoManagerLocal.js

@@ -163,7 +163,7 @@ export class VideoManagerLocal {
                 this.ws.onclose = null;
                 try {
                     this.ws.close();
-                } catch (_) {}
+                } catch (_) { }
                 this.ws = null;
             }
 
@@ -460,6 +460,7 @@ export class VideoManagerLocal {
                     on_screen: data.on_screen,
                     gaze_yaw: data.gaze_yaw,
                     gaze_pitch: data.gaze_pitch,
+                    eye_gaze_enabled: data.eye_gaze_enabled || false,
                 };
                 this.drawDetectionResult(detectionData);
                 break;
@@ -572,15 +573,15 @@ export class VideoManagerLocal {
         152, 148, 176, 149, 150, 136, 172, 58, 132,
         93, 234, 127, 162, 21, 54, 103, 67, 109, 10,
     ];
-    static LEFT_EYE = [33,7,163,144,145,153,154,155,133,173,157,158,159,160,161,246];
-    static RIGHT_EYE = [362,382,381,380,374,373,390,249,263,466,388,387,386,385,384,398];
-    static LEFT_IRIS = [468,469,470,471,472];
-    static RIGHT_IRIS = [473,474,475,476,477];
-    static LEFT_EYEBROW = [70,63,105,66,107,55,65,52,53,46];
-    static RIGHT_EYEBROW = [300,293,334,296,336,285,295,282,283,276];
-    static NOSE_BRIDGE = [6,197,195,5,4,1,19,94,2];
-    static LIPS_OUTER = [61,146,91,181,84,17,314,405,321,375,291,409,270,269,267,0,37,39,40,185,61];
-    static LIPS_INNER = [78,95,88,178,87,14,317,402,318,324,308,415,310,311,312,13,82,81,80,191,78];
+    static LEFT_EYE = [33, 7, 163, 144, 145, 153, 154, 155, 133, 173, 157, 158, 159, 160, 161, 246];
+    static RIGHT_EYE = [362, 382, 381, 380, 374, 373, 390, 249, 263, 466, 388, 387, 386, 385, 384, 398];
+    static LEFT_IRIS = [468, 469, 470, 471, 472];
+    static RIGHT_IRIS = [473, 474, 475, 476, 477];
+    static LEFT_EYEBROW = [70, 63, 105, 66, 107, 55, 65, 52, 53, 46];
+    static RIGHT_EYEBROW = [300, 293, 334, 296, 336, 285, 295, 282, 283, 276];
+    static NOSE_BRIDGE = [6, 197, 195, 5, 4, 1, 19, 94, 2];
+    static LIPS_OUTER = [61, 146, 91, 181, 84, 17, 314, 405, 321, 375, 291, 409, 270, 269, 267, 0, 37, 39, 40, 185, 61];
+    static LIPS_INNER = [78, 95, 88, 178, 87, 14, 317, 402, 318, 324, 308, 415, 310, 311, 312, 13, 82, 81, 80, 191, 78];
     static LEFT_EAR_POINTS = [33, 160, 158, 133, 153, 145];
     static RIGHT_EAR_POINTS = [362, 385, 387, 263, 373, 380];
     // Iris/eye corners for gaze lines
@@ -676,11 +677,12 @@ export class VideoManagerLocal {
                 outer: VideoManagerLocal.RIGHT_EYE_OUTER,
             },
         ];
-        // Get L2CS gaze angles + on_screen state from latest detection data
+        // Get L2CS gaze angles + on_screen state + eye gaze toggle from latest detection data
         const detection = this._lastDetection;
         const gazeYaw = detection ? detection.gaze_yaw : undefined;
         const gazePitch = detection ? detection.gaze_pitch : undefined;
         const onScreen = detection ? detection.on_screen : undefined;
+        const eyeGazeEnabled = detection ? detection.eye_gaze_enabled : false;
         const hasL2CSGaze = gazeYaw !== undefined && gazePitch !== undefined;
         const gazeLineColor = (onScreen === false) ? '#FF0000' : '#00FF00';
         const gazeLineLength = 100;
@@ -712,57 +714,60 @@ export class VideoManagerLocal {
             ctx.lineWidth = 1;
             ctx.stroke();
 
-            // Gaze direction line — use L2CS angles when available, else geometric fallback
-            if (hasL2CSGaze) {
-                // L2CS pitch/yaw in radians -> pixel direction vector
-                // Matches upstream L2CS-Net vis.py draw_gaze formula:
-                //   dx = -length * sin(pitch) * cos(yaw)
-                //   dy = -length * sin(yaw)
-                const dx = -gazeLineLength * Math.sin(gazePitch) * Math.cos(gazeYaw);
-                const dy = -gazeLineLength * Math.sin(gazeYaw);
-                const ex = cx + dx;
-                const ey = cy + dy;
-
-                // Main gaze line (thick, color-coded)
-                ctx.beginPath();
-                ctx.moveTo(cx, cy);
-                ctx.lineTo(ex, ey);
-                ctx.strokeStyle = gazeLineColor;
-                ctx.lineWidth = 3;
-                ctx.stroke();
-
-                // Arrowhead
-                const angle = Math.atan2(ey - cy, ex - cx);
-                const arrowLen = 10;
-                ctx.beginPath();
-                ctx.moveTo(ex, ey);
-                ctx.lineTo(ex - arrowLen * Math.cos(angle - 0.4), ey - arrowLen * Math.sin(angle - 0.4));
-                ctx.moveTo(ex, ey);
-                ctx.lineTo(ex - arrowLen * Math.cos(angle + 0.4), ey - arrowLen * Math.sin(angle + 0.4));
-                ctx.strokeStyle = gazeLineColor;
-                ctx.lineWidth = 2;
-                ctx.stroke();
-            } else {
-                // Geometric fallback: iris displacement from eye center (scaled up)
-                const innerPt = _get(inner);
-                const outerPt = _get(outer);
-                if (innerPt && outerPt) {
-                    const eyeCx = (innerPt[0] + outerPt[0]) / 2.0 * w;
-                    const eyeCy = (innerPt[1] + outerPt[1]) / 2.0 * h;
-                    const fdx = cx - eyeCx;
-                    const fdy = cy - eyeCy;
-                    const flen = Math.hypot(fdx, fdy);
-                    if (flen > 0.5) {
-                        const scale = gazeLineLength / flen;
-                        ctx.beginPath();
-                        ctx.moveTo(cx, cy);
-                        ctx.lineTo(cx + fdx * scale, cy + fdy * scale);
-                        ctx.strokeStyle = '#00FFFF';
-                        ctx.lineWidth = 2;
-                        ctx.stroke();
+            // Gaze direction line — only draw when eye gaze toggle is ON
+            if (eyeGazeEnabled) {
+                if (hasL2CSGaze) {
+                    // L2CS pitch/yaw in radians -> pixel direction vector
+                    // Matches upstream L2CS-Net vis.py draw_gaze formula:
+                    //   dx = -length * sin(pitch) * cos(yaw)
+                    //   dy = -length * sin(yaw)
+                    const dx = -gazeLineLength * Math.sin(gazePitch) * Math.cos(gazeYaw);
+                    const dy = -gazeLineLength * Math.sin(gazeYaw);
+                    const ex = cx + dx;
+                    const ey = cy + dy;
+
+                    // Main gaze line (thick, color-coded)
+                    ctx.beginPath();
+                    ctx.moveTo(cx, cy);
+                    ctx.lineTo(ex, ey);
+                    ctx.strokeStyle = gazeLineColor;
+                    ctx.lineWidth = 3;
+                    ctx.stroke();
+
+                    // Arrowhead
+                    const angle = Math.atan2(ey - cy, ex - cx);
+                    const arrowLen = 10;
+                    ctx.beginPath();
+                    ctx.moveTo(ex, ey);
+                    ctx.lineTo(ex - arrowLen * Math.cos(angle - 0.4), ey - arrowLen * Math.sin(angle - 0.4));
+                    ctx.moveTo(ex, ey);
+                    ctx.lineTo(ex - arrowLen * Math.cos(angle + 0.4), ey - arrowLen * Math.sin(angle + 0.4));
+                    ctx.strokeStyle = gazeLineColor;
+                    ctx.lineWidth = 2;
+                    ctx.stroke();
+                } else {
+                    // Geometric fallback: iris displacement from eye center (scaled up)
+                    const innerPt = _get(inner);
+                    const outerPt = _get(outer);
+                    if (innerPt && outerPt) {
+                        const eyeCx = (innerPt[0] + outerPt[0]) / 2.0 * w;
+                        const eyeCy = (innerPt[1] + outerPt[1]) / 2.0 * h;
+                        const fdx = cx - eyeCx;
+                        const fdy = cy - eyeCy;
+                        const flen = Math.hypot(fdx, fdy);
+                        if (flen > 0.5) {
+                            const scale = gazeLineLength / flen;
+                            ctx.beginPath();
+                            ctx.moveTo(cx, cy);
+                            ctx.lineTo(cx + fdx * scale, cy + fdy * scale);
+                            ctx.strokeStyle = '#00FFFF';
+                            ctx.lineWidth = 2;
+                            ctx.stroke();
+                        }
                     }
                 }
             }
+            // When eye gaze is OFF, no gaze lines are drawn
         }
     }