o y i\7@sNdZddlZddlZddlZddlZddlmZddlmZGdddZ dS)zR Sarvam AI Speech-to-Text Service Handles audio transcription using Sarvam AI API N)Optional)Logc@seZdZdZddZdededeefddZd edeefd d Zd e dedeefd dZ de dedee fddZ de dee fddZ dde dede fddZde dedeefddZdS)SarvamSTTServicez8Sarvam AI Speech-to-Text service for audio transcriptioncCsjtdd|_tdd|_tdd|_tdd|_d |_|jd |_t d |jd |jdS) NZSARVAM_API_KEYZ$sk_plejgbkc_tIeBnij5Cl5YQINTez4opajYZSARVAM_BASE_URLzhttps://api.sarvam.aiZ SARVAM_MODELz saarika:v2.5ZSARVAM_LANGUAGEzen-INz/speech-to-textu-🎤 Sarvam STT Service initialized - Model: z , Language: ) osgetenvapi_keyZbase_urlmodellanguagetimeoutstt_urlrinfo)selfrservices/sarvam_stt_service.py__init__szSarvamSTTService.__init__ audio_data session_idreturnc sznt|}td|dt|dt|dkr)tdt|dWdS|||IdH}|s?td|WdS|||IdH}|rm||}|rdt d|d |dd d |WStd |d WdSt y}zt d|d|WYd}~dSd}~ww)a Transcribe audio using Sarvam AI API Args: audio_data: Base64 encoded audio data session_id: Session identifier for logging Returns: Transcribed text or None if failed u🎤 Decoded audio for session :  bytes@u/🎤 Audio too short for Sarvam transcription: Nu2⚠️ Failed to process audio format for session u*🎤 Sarvam transcribed audio for session : 'd...'u1🎤 Filtered out false positive transcription: ''u.❌ Error in Sarvam transcription for session ) base64 b64decoderdebuglen_process_audio_formatwarning_call_sarvam_api_filter_transcriptionr Exceptionerror)rrr audio_bytesprocessed_audio transcriptionfiltered_transcriptionerrrtranscribe_audios0    z!SarvamSTTService.transcribe_audior(csb|sdS|}td|dgdddl}|dd|}|vs,|vr7td|ddSt|d krHtd |ddS|}t|d kret|dd kretd |ddS|}t|d krtfdd|Drtd|ddS| dd dd dd dddkrtd|ddStd|d|S)a Filter out common false positive transcriptions from Sarvam STT. Args: transcription: Raw transcription from Sarvam Returns: Filtered transcription or None if it's a false positive Nu 🎤 Raw Sarvam transcription: 'r)yesyeahZyepokayokmmhmmZuhZumZahZohnoZnopeZsurerightZalrightZhellohiZheyZthanksz thank youZbyeZgoodbyerz[.,!?]+u🎤 Filtered false positive: 'u$🎤 Filtered short transcription: 'u"🎤 Filtered single short word: 'c3s|]}|vVqdS)Nr).0ZwordZfalse_positivesrr zsz9SarvamSTTService._filter_transcription..u.🎤 Filtered repeated false positive words: '.,!?u/🎤 Filtered punctuation-only transcription: 'u🎤 Accepted transcription: ') striplowerrrresubrsplitallreplace)rr(ZcleanedrBZcleaned_no_punctZwordsZwords_no_punctrr:rr#Is4   ",z&SarvamSTTService._filter_transcription audio_chunksc szztd|dt|dd}|D]*}z t|}||7}Wqty>}ztd|d|WYd}~qd}~ww|sLtd|WdSt|d kr`td t|d WdS|||IdH}|snWdS|||IdH}|r| |} | rt d |d | ddd| WStd|dWdSty}zt d|d|WYd}~dSd}~ww)a Transcribe combined audio buffer using Sarvam AI API Args: audio_chunks: List of base64 encoded audio chunks session_id: Session identifier for logging Returns: Transcribed text or None if failed u2🎤 Transcribing Sarvam audio buffer for session  - z chunksu0⚠️ Failed to decode audio chunk for session rNu)⚠️ No valid audio chunks for session iu8🎤 Combined audio too short for Sarvam transcription: ru3🎤 Sarvam transcribed combined audio for session rrru8🎤 Filtered out false positive buffer transcription: 'ru5❌ Error in Sarvam buffer transcription for session ) rrrrrr$r!r r"r#r r%) rrGrZcombined_audiochunkZ decoded_chunkr*r'r(r)rrrtranscribe_audio_buffersF       z(SarvamSTTService.transcribe_audio_bufferr&c szC||}|std|WdS|j|dd}t|dkr/tdt|dWdStd|dt|d t|d|WStyb}ztd |d|WYd}~dSd}~ww) u Process audio format for Sarvam API (convert μ-law to WAV) Args: audio_bytes: Raw audio bytes session_id: Session identifier for logging Returns: Processed WAV audio bytes or None if failed u3⚠️ Failed to convert μ-law to PCM for session Nr) sample_rateu🎵 WAV file too short: ru(🎵 Processed audio format for session rz -> u.❌ Error processing audio format for session ) _mulaw_to_pcmrr!_create_wav_filerrr$r%)rr&rpcm_dataZwav_datar*rrrr s"  &z&SarvamSTTService._process_audio_format mulaw_datac CsRz ddl}||d}|WSty(}ztd|WYd}~dSd}~ww)u Convert μ-law encoded audio to PCM Args: mulaw_data: μ-law encoded audio bytes Returns: PCM audio bytes or None if failed rNu$❌ Error converting μ-law to PCM: )audioopZulaw2linr$rr%)rrPrRrOr*rrrrMs  zSarvamSTTService._mulaw_to_pcmrrOrLcCs~ddl}ddl}|}||d}|d|d||||Wdn1s1wY|d| S)z Create WAV file from PCM data Args: pcm_data: PCM audio data sample_rate: Sample rate in Hz Returns: WAV file bytes rNwbr7rQ) waveioBytesIOopenZ setnchannelsZ setsampwidthZ setframerateZ writeframesseekread)rrOrLrTrUZ wav_bufferZwav_filerrrrNs      z!SarvamSTTService._create_wav_filec sdzd|ji}t}|jd|ddd|d|j|d|jtd|d t|d t 4Id H}|j |j ||tj |j d d 4Id H}|jdkr|Id H}|dd}td|d|dt|dkr|tdt|d|dvrtd|d|Wd Id HWd Id HWS|Id H} td|d|jd| tdt|d  Wd Id HWd Id HWd S1Id HswYWd Id HWd S1Id HswYWd Stjytd|Yd Sty1} ztd|d| WYd } ~ d Sd } ~ ww) z Call Sarvam AI API for transcription Args: audio_data: WAV audio data session_id: Session identifier for logging Returns: Transcribed text or None if failed zapi-subscription-keyfilez audio.wavz audio/wav)filenameZ content_typer Z language_codeu$🎤 Calling Sarvam API for session z - Audio size: rN)total)headersdatar Z transcriptr5u%🎤 Sarvam API response for session rri>u#⚠️ Short audio sent to Sarvam: z" bytes (may cause false positives))r,r-r.r/r0r1u.⚠️ Potential false positive from Sarvam: 'z)' - Audio may be too short or low qualityu!❌ Sarvam API error for session rrHu(🎤 Audio data size that caused error: u#❌ Sarvam API timeout for session u)❌ Error calling Sarvam API for session )raiohttpZFormDataZ add_fieldr r rrrZ ClientSessionZpostr Z ClientTimeoutr statusZjsongetr!rAr@textr%asyncio TimeoutErrorr$) rrrr]r^ZsessionZresponseresultr(Z error_textr*rrrr"sX      2z!SarvamSTTService._call_sarvam_apiN)r)__name__ __module__ __qualname____doc__rstrrr+r#listrKbytesr rMintrNr"rrrrrs ,>7!r) rjrrr`rdtypingrZservices.log_utilsrrrrrrs