i(zdZddlZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZeGddZy) zi Voice Activity Detection (VAD) Module Energy-based VAD using RMS with adaptive noise floor estimation. N) dataclassfield)Optional) SAMPLE_RATE_INBYTES_PER_SAMPLE VAD_FRAME_MSEND_SILENCE_MSMIN_UTTERANCE_MSMIN_RMS_THRESHOLDNOISE_MULTIPLIERNOISE_EMA_ALPHAFRAME_BYTES_8KMIN_PCM_BYTES_8KceZdZUdZeZeed<eZ eed<e Z eed<e Z eed<eZeed<eZeed<edd Zeed <edd Zeed <ed d Zeed <edd Zeed<edd Zeed<ed d Zeed<edefdZedefdZedefdZedefdZ edefdZ!dZ"dZ#dede$e%fdZ&dede'e%fdZ(de'e%fdZ)de'e%fd Z*defd!Z+defd"Z,y#)$ VADDetectora6 Voice Activity Detector using energy-based RMS thresholding. Uses exponential moving average (EMA) for adaptive noise floor estimation. Attributes: frame_ms: Duration of each analysis frame in milliseconds end_silence_ms: Silence duration (ms) to mark end of utterance min_utterance_ms: Minimum utterance length (ms) to accept min_rms_threshold: Absolute minimum RMS threshold noise_multiplier: Factor to multiply noise floor for threshold noise_ema_alpha: EMA smoothing factor for noise estimation frame_msend_silence_msmin_utterance_msmin_rms_thresholdnoise_multipliernoise_ema_alphaF)defaultrepr _vad_buffer_speech_buffer _speakingr _silence_ms _noise_rms_buffer_committedreturncNtt|jdz ztzS)zNumber of bytes per VAD frameg@@)intrrrselfs +/home/sas/my/fyp/ringai/ringai/audio/vad.py frame_byteszVADDetector.frame_bytes8s#>T]]V%;|jsd|_d|_ d|_ |xj|z c_ d|_ y|jr]|xj|jz c_ |xj|z c_ |j|jk\r|jSy)z Process a single audio frame for VAD. Returns utterance dict if speech segment completed, None otherwise. g?TrFN)audiooprmsrrrr!maxrr%rrr"rrr_commit_utterance)r'r<rA threshold is_speechs r(r9zVADDetector._process_framews kk%!12~~t+++t>&&,- O ..DOOdF[F[4[0\] 9$ >>!%#$ ).&   5 (  D ~~  DMM1 ##u,###t':'::1133rc`d|_tt|jtt zz dz}||j kr|jyt|j|jkr|jy|j|d}d|_ d|_d|_ |S)z3Commit current speech buffer as completed utteranceFiN)r6 duration_msTrr) rr%r8rrrrr3r+r"r)r'rGr=s r(rCzVADDetector._commit_utterances $$ %)9N)J Kt S  .. . JJL t"" #d&8&8 8 JJL))&  "&! rcT|jr |jry|jS)z Force commit current speech buffer (for wall-clock timeout). Returns utterance dict if buffer is valid, None otherwise. N)rr"rCr&s r( force_commitzVADDetector.force_commits' ""d&<&<%%''rcnt|jt|j|jzS)z"Get current adaptive RMS threshold)rBrr%r!rr&s r( get_thresholdzVADDetector.get_thresholds)4))3tAVAV/V+WXXrc|jS)zGet current estimated noise RMS)r!r&s r(get_noise_floorzVADDetector.get_noise_floors rN)-__name__ __module__ __qualname____doc__r rr%__annotations__r rr rr rr rfloatrrrrbytesrrboolrr!r"propertyr)r+r-r/r1r3r5listdictr>rr9rCrIrKrMrr(rrs !Hc (NC(,c,.s..e.,OU,s7K7!#E:NE:E6It6QU3K3c6J6#E>t> QSQQ s  T#u## C  ' $t*2'E'htn'R8D>6 (htn (YsYrr)rQr@ dataclassesrrtypingrconfigrrr r r r r rrrrrYrr(r]s@ (    rr rr