TiUdZddlmZddlZddlZddlZddlmZmZddl m Z m Z eje ZeGddZGdd Zy) a>FR client for the Face Recognition server. Sends a single JPEG frame to the FR WebSocket server and returns the list of detections. The server runs SCRFD + ViT + ByteTrack; per-connection tracking state resets when the connection closes, which is fine for the demo pattern of one ``IDENTIFY_PERSON`` call per scene. ) annotationsN) dataclassfield)AnyOptionalc>eZdZUdZedZded<dZded<y ) FRConfigzConfiguration for the FR client. Attributes: endpoint: WebSocket URL for the FR server. timeout_s: Connection + response timeout in seconds. c.tjddS)N FR_ENDPOINTzws://kluster.klass.dev:42067/)osgetenvF/home/nelsen/Projects/kognitive/orchestrator/src/services/fr_client.pyzFRConfig.s :! r)default_factorystrendpoint int timeout_sN)__name__ __module__ __qualname____doc__rr__annotations__rrrrr r s+ Hc Isrr c"eZdZdZdddZddZy)FRClientuAsync WebSocket client for the Face Recognition server. Usage:: client = FRClient() detections = await client.identify(jpeg_bytes) # [{"identity": "alice", "confidence": 0.92, "bbox": [x1, y1, x2, y2]}, ...] The FR server expects: - **In:** Raw binary JPEG/PNG bytes (one frame per message, any resolution) - **Out:** ``{"timestamp": ..., "detections": [{"identity": str|null, "confidence": float, "bbox": [x1, y1, x2, y2]}]}`` A new WebSocket connection is opened per call. This is adequate for the demo frequency of 1–2 calls per scene and avoids long-lived connection management complexity. Nc*|xs t|_yN)r config)selfr!s r__init__zFRClient.__init__8s*  rcTK|stjdgS ddl} |j |j j|j j|j j4d{}|j|d{ddl }|j|j|j jd{}dddd{tj}|jdg}tj!dt#|t#||S#t$rtj dgcYSwxYw7777#1d{7swYxYw#t$$r"}tj d |gcYd}~Sd}~wwxYww) aSend a JPEG frame to the FR server and return detections. Args: jpeg_bytes: Raw JPEG bytes of the frame to analyse. Returns: List of detection dicts, each with keys: - ``identity``: str name or ``None`` if unrecognised - ``confidence``: float in [0, 1] - ``bbox``: ``[x1, y1, x2, y2]`` pixel coordinates Returns an empty list on any error (connection failure, timeout, malformed response). u6FRClient.identify called with empty bytes — skippingrNu^websockets is not installed — cannot call FR server. Install it with: pip install websockets) open_timeout close_timeout)timeout detectionsz0FR identified %d detection(s) from %d-byte framezFR identify request failed: %s)loggerwarning websockets ImportErrorerrorconnectr!rrsendasynciowait_forrecvjsonloadsgetinfolen Exception) r" jpeg_bytesr+wsr0rawdatar(excs ridentifyzFRClient.identify;s NNS TI   !)) $$![[22"kk33* W Wggj)))#,,RWWY @U@U,VV W W::c?D/3xx b/IJ KKBJJ   ;  LL: I   W * W W W W W.  LL93 ?I sF(D7AE:6E7E::E%EAE%E!E% E: E#!AE:6F(7 EF(EF(E:E%!E%#E:%E7+E. ,E73E:: F%F F%F( F%%F(r )r!zOptional[FRConfig]returnNone)r9bytesr?zlist[dict[str, Any]])rrrrr#r>rrrrr%s$+6rr)r __future__rr3loggingr dataclassesrrtypingrr getLoggerrr)r rrrrrGsU#  (   8 $      LLr