o cBj@sdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl mZddlmZeeZe Gd d d Zd$ddZd%ddZd&ddZd'ddZd(ddZd)d"d#ZdS)*zFDownload and normalize recording files for Sarvam STT (WAV, OGG, MP3).) annotationsN) dataclass)ListOptional)urlparse)wav_duration_secondsc@s&eZdZUded<ded<ded<dS) PreparedAudiostrpathfloatdurationz List[str] cleanup_pathsN)__name__ __module__ __qualname____annotations__rrH/home/aiteam/pcaa-dev/call-proccessing/stt_pipeline/stt/audio_prepare.pyr s  r urlr returncCs6tt|pdj}dD] }||r|Sq dS)N).wav.ogg.mp3z.mpegz.m4az.flacr)rr r lowerendswith)rr extrrr_suffix_from_urls  rdatabytescCst|dkr|dddkrdSt|dkr |dddkr dSt|dkr0|dddkr0d St|d krF|dd krF|d d @d krFd St|S)NrsRIFFrsOggSrsID3rr)lenr)rrrrr_detect_format"s(r'r suffixr c Cs|dkrt|Sz'ddlm}||}|dur.t|dddur.t|jjp%d}|dkr.|WSWntyH}z t d||WYd}~nd}~ww|dkrQt|St d|)Nrr)Fileinfoz"mutagen duration failed for %s: %sz'Could not determine audio duration for ) rmutagenr)getattrr r*length Exceptionloggerdebug RuntimeError)r r( MutagenFilemetar-excrrr_duration_for_path.s" r5src_pathcallidcCstd}|s tdtjddd\}}t||dddd d |d d d ddd|g}z tj|dddddWn)tj y^}ztj |rIt ||j pMd}td|pU||d}~wwtd|tj ||S)Nffmpegu5ffmpeg not installed — cannot convert non-WAV audior/tmpr(dirz-yz -hide_bannerz -loglevelerrorz-iz-acodec pcm_s16lez-ar16000z-ac1Tx)checkcapture_outputtexttimeoutrzffmpeg conversion failed: u$[%s] Converted %s → WAV via ffmpeg)shutilwhichr1tempfilemkstemposclose subprocessrunCalledProcessErrorr existsunlinkstderrstripr/r*basename)r6r7r8out_fdout_pathcmdr4rPrrr_ffmpeg_convert_to_wav@s>    rVcontentc Cst||}g}tj|dd\}}t|||t|d }||Wdn1s/wYz(|dkrFt||}t |||dWSt||}t ||} || t | ||dWSt y|D]} z| rutj | rut| WqetyYqeww)z Save downloaded bytes with the correct extension and return a WAV path for Sarvam. WAV inputs are used as-is. OGG/MP3/etc. are converted with ffmpeg when available. r9r:wbNr)r r r)r'rGrHrIrJappendopenwriter5r rVr.r rNrOOSError) rWrr7r(cleanupsrc_fdr6fhr wav_pathr rrr prepare_audioas6            rapreparedNonec Cs@|jD]}z|rtj|rt|WqtyYqwdS)N)rrIr rNrOr\)rbr rrrcleanup_prepareds   rd)rr rr )rr rr rr )r r r(r rr )r6r r7r rr )rWr rr r7r rr )rbr rrc)__doc__ __future__rloggingrIrErKrG dataclassesrtypingrr urllib.parser audio_chunkerr getLoggerrr/r rr'r5rVrardrrrrs(         !"