Update largest face status (#116)

* add largest face status

* add bbox info and have face order by the bbox area

* update

---------

Co-authored-by: openminddev <147775420+openminddev@users.noreply.github.com>

Author: WenjinFu

src/om1_vlm/anonymizationSys/face_recog_stream/face_tracker.py

@@ -120,6 +120,9 @@ def __init__(
         # Track IDs seen this frame (for cleanup)
         self._active_ids: set = set()
 
+        # Current frame faces (for status queries)
+        self._current_faces: list = []
+
     @staticmethod
     def _init_tracker(track_buffer: int, det_conf: float = 0.5):
         """Initialize BoTSORT tracker with thresholds aligned to detection confidence."""
@@ -284,6 +287,22 @@ def update(
         # Cleanup stale tracks
         self._cleanup_stale()
 
+        # Build faces sorted by bbox area (largest = closest first)
+        faces = []
+        for r in results:
+            x1, y1, x2, y2 = r.bbox
+            area = (x2 - x1) * (y2 - y1)
+            ident = self._identities.get(r.track_id)
+            name = ident.name if ident and ident.status == "identified" else "unknown"
+            faces.append(
+                {
+                    "name": name,
+                    "bbox": r.bbox,
+                    "area": area,
+                }
+            )
+        self._current_faces = sorted(faces, key=lambda f: f["area"], reverse=True)
+
         return results
 
     def _get_display_name(
@@ -525,6 +544,17 @@ def get_track_identities(self) -> Dict[int, dict]:
             if tid in self._active_ids
         }
 
+    def get_faces(self) -> list:
+        """Get all faces sorted by bbox area (largest first).
+
+        Returns
+        -------
+        list of dict
+            Each dict has 'name' (str), 'bbox' (tuple), 'area' (int).
+            Empty list if no faces in current frame.
+        """
+        return self._current_faces
+
     def reset(self) -> None:
         """Reset all tracking state."""
         self._identities.clear()

src/om1_vlm/anonymizationSys/face_recog_stream/http_api.py

@@ -85,6 +85,7 @@ def __init__(
         frame_lock: threading.Lock,
         run_job_sync: Callable[[Callable[[], Any]], Any],
         logger: Optional[logging.Logger] = None,
+        face_tracker=None,
     ):
         """Initialize the HTTP API wrapper."""
         self.who = who
@@ -100,6 +101,7 @@ def __init__(
         self.run_job_sync = run_job_sync
         self.log = logger or logging.getLogger("http_api")
         self.server = None
+        self.face_tracker = face_tracker
 
     def stop(self) -> None:
         """Stop an attached HTTP server if present."""
@@ -148,7 +150,10 @@ def _handle(self, payload: Dict[str, Any], path: str) -> Dict[str, Any]:
                     if payload
                     else self.who.lookback_sec
                 )
-                return self.who.snapshot(sec)
+                result = self.who.snapshot(sec)
+                if self.face_tracker is not None:
+                    result["faces"] = self.face_tracker.get_faces()
+                return result
 
             if path == "/config":
                 return self._handle_config(payload)

src/om1_vlm/anonymizationSys/face_recog_stream/run.py

@@ -480,7 +480,7 @@ def main() -> None:
     if args.recognition and arc is not None:
         face_tracker = FaceTracker(
             arc=arc,
-            recog_interval=0.2,  # run AdaFace every 0.3s per unidentified track
+            recog_interval=0.2,  # run AdaFace every 0.2s per unidentified track
             vote_frames=3,  # collect 3 votes before deciding
             vote_threshold=0.5,  # majority vote (2/3)
             max_recog_attempts=10,
@@ -684,6 +684,7 @@ def run_job_sync(fn):
         frame_lock=frame_lock,
         run_job_sync=run_job_sync,
         logger=logger,
+        face_tracker=face_tracker,
     )
 
     # Run the server here