Z|h<ddlZddlZddlmZddlZGddZy)N)YOLOcBeZdZdZdZededefdZedZ y) Proctoringctd|_|jjjt j tj jdd|_y)z Load face detection (YOLOv8) and landmark detection (face_alignment) models. Use case: Detect faces and estimate head pose in a given frame. z'services/proctor/models/yolov8n-face.ptFcpu) flip_inputdeviceN) r faceDetectormodelfuseface_alignment FaceAlignment LandmarksTypeTWO_Dfa)selfs :E:\xampp\htdocs\new-grp\engine\services\proctor\proctor.py__init__zProctoring.__init__sQ !!JK $$& ..  ( ( . . c |jdd\}}|jj|ddd}d}d}d}|jD]}t t |j dj\} } } } d}|dz }|| | | | f} |jj| }|d|d}|tj| | gz }tj|d |d |d |d |d |dgtj}tjgdgdgdgdgdgdgtj}|}tj|d|dz gd||dz ggdgtj}tjd}tj||||\}}}|r*tj |\}}tj"|ddz|ddzz}|dkr:tj$|d |d}tj$|d |}d}nRtj$|d|d}tj$|d |}tj$|d|d}t'tj(|dt'tj(|dt'tj(|dd }n|||d!S)"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)sourceimgszverboserT$-06)dtype)r#r#)r#gOg))gfffffE皙Y@@:)gfffffE@r$r%)fffff)rr)rr)rr)rr)rr)yawpitchroll) face_detected people_count head_pose)shaper predictboxesmapintxyxytolistrget_landmarks_from_imagenparrayfloat64zeroscv2solvePnP Rodriguessqrtarctan2rounddegrees)rframehwpreds faceDetected peopleCountheadPoseboxx1y1x2y2faceCrop landmarkslm imagePoints modelPoints focalLength cameraMatrix distCoeffssuccessrot_Rsyxyzs r analyzeFramezProctoring.analyzeFrames{{2A1!!))c5)QRST  ;;C chhqk&8&8&:;NBBL 1 KRUBrE\*H88BI 1B "((B8$ $B((212222 $ZZ!K((#$#%$ $ZZ!KK88aQ'KQ'%ZZ !L &)J!ll; \S]^OGS!}}S)1WWQtWaZ!D'1*459 AdG8QtW5A AdG8R0AA 1T7AdG4A AdG8R0A 1T7AdG4A!A2"2::a=!4!"**Q-3 D*'!  remotionsreturnc|d|dz|dzdz }|d|dzdz }|d}|d}|d}|||||d }t||j S) Nfearangrydisgusthappyneutralrsad)nervousrelaxedrfrhrb)key)maxget)r_rirjrfrhrb composites r fuseEmotionszProctoring.fuseEmotionshsF#hw&77(9:MMQRRG$x '::a?!uo  9)--00rc>g}|dkDr|jdn|dkr|jd|dkDr|jdn|dkr|jd|d kDr|jd n|d kr|jd |rd j|SdS)Nz looking rightiz looking leftz looking downrz looking upzhead tilted rightizhead tilted leftz, zfacing forward)appendjoin)r)r*r+ orientations rdescribeHeadPosezProctoring.describeHeadPoseys 8[''8 3Y **>: 3; **>: RZ++L9 "9k(()<= CZ++,>?)4tyy%J:JJrN) __name__ __module__ __qualname__rr^ staticmethoddictstrrorwrrrrsF  S h1t111 KKrr)r;numpyr7 ultralyticsrr rr~rrrs |K|Kr