$h%PddlZddlZddlZddlZddlmZddlmZGddZy)N)datetime)timezonecNeZdZdZdZdZdZedede fdZ edZ y ) ProctoringcDd|_d|_d|_tdy)zE Initialize without loading heavy models immediately NFzProctoring service initialized) faceDetectorfa_models_loadedprint)selfs 0/var/www/html/engine/services/proctor/proctor.py__init__zProctoring.__init__s$!# ./c@|jsq ddlm}ddl}|d|_|jj j |j|jjdd|_ d|_yy#t$r}td ||d}~wwxYw) z4 Lazy load models when first needed r)YOLONz'services/proctor/models/yolov8n-face.ptFcpu) flip_inputdeviceTzError loading models: ) r ultralyticsrface_alignmentrmodelfuse FaceAlignment LandmarksTypeTWO_Dr Exceptionr )r rres r _load_modelszProctoring._load_modelss"" ,%$()R$S!!!'',,.(66"0066$ 7 '+## .qc23 sA/A?? BBBc |js|j|jdd\}}|jj |ddd}d}d}d}d}|j t |j dk(rddddS|j D]]} tt| jdj\} } } } | dk\s9| dk\s?| |ksE| |ksK| | kDsQ| | kDsW|dz }d }_|j D]} |rntt| jdj\} } } } | dk\r| dk\r| |kr| |kr | | kDr| | kDsX|| | | | f}|jdk(rs |jj|}|t |dk(r|d}|t!j"| | gz }t!j"|d |d |d |d|d|dgt j$}t!j"gdgdgdgdgdgdgt j$}|}t!j"|d|dz gd||dz ggdgt j$}t!j&d}t)j*||||\}}}|st)j,|\}}t!j.|ddz|ddzz}|dkr:t!j0|d |d}t!j0|d |}d}nRt!j0|d |d!}t!j0|d |}t!j0|d|d}t3t!j4|dt3t!j4|dt3t!j4|dd"}d }|||dS#t$r}td |Yd}~d}~wwxYw)#z Analyze the frame to detect faces and compute head orientation. Returns: dict with keys: face_detected, people_count, head_pose (yaw, pitch, roll) Use case: Core vision engine for your proctoring system. NiF)sourceimgszverboser) face_detected people_count head_poseTzLandmark detection failed: $-06)dtype)r/r/)r/gOg))gfffffE皙Y@@:)gfffffE@r0r1)fffff)r'r )r'r')r r)r r')r r )yawpitchroll)r rshaperpredictboxeslenmapintxyxytolistsizer get_landmarks_from_imagerr nparrayfloat64zeroscv2solvePnP Rodriguessqrtarctan2rounddegrees)r framehwpreds faceDetected peopleCountheadPosefirstFaceAnalyzedboxx1y1x2y2faceCrop landmarksrlm imagePoints modelPoints focalLength cameraMatrix distCoeffssuccessrot_Rsyxyzs r analyzeFramezProctoring.analyzeFrame's ""    {{2A1!!))c5)QRST  ! ;; #ekk"2a"7!& !! ;; $C chhqk&8&8&:;NBBQw27rQw27rBw2PR7q #  $;;L )C  chhqk&8&8&:;NBB!GaB!GaBGPRUWPWRUBrE\*H}}!  GG<g}|dkDr|jdn|dkr|jd|dkDr|jdn|dkr|jd|dkDr|jdn|d kr|jd |rd j|Sd S) Nz looking rightiz looking leftz looking downz looking upzhead tilted rightizhead tilted leftz, zfacing forward)appendjoin)r5r6r7 orientations r r~zProctoring.describeHeadPoses 8[''8 3Y **>: 2:{)).9 S[+,,\: "9k(()<= CZ++,>?)4tyy%J:JJrN) __name__ __module__ __qualname__rrrjr staticmethoddictstrrr~rr rrsR0,w rHT1t111  K Krr) rFnumpyrBrrwrpytzrrrrr rs" ~K~Kr