o y i&@spdZddlZddlZddlZddlmZmZmZddlZddl Z ddl m Z ddl m Z GdddZeZdS)zc Sarvam TTS Service for Text-to-Speech Conversion Handles text-to-speech using Sarvam AI's TTS API N)OptionalDictAny)Config)Logc@seZdZdZddefddZdedeefdd Zdedeefd d Z de fd d Z de ee ffddZdede fddZdS)SarvamTTSServicez Service for converting text to speech using Sarvam AI's TTS API. Provides cost-effective alternative to OpenAI TTS. Nvoice_idcCsdtj|_tj|_tj|_|stjr|ptjnd|_ tj |_ tj |_ td|jd|j dS)NZmanishau-🎤 Sarvam TTS Service initialized - Model: z , Voice: )rZSARVAM_TTS_API_KEYapi_keyZSARVAM_TTS_BASE_URLbase_urlZSARVAM_TTS_MODELmodelZSARVAM_TTS_VOICElowervoiceZSARVAM_TTS_LANGUAGElanguageZSARVAM_TTS_SAMPLE_RATE sample_raterinfo)selfrrservices/sarvam_tts_service.py__init__szSarvamTTSService.__init__textreturnc s|r|stddSz-td|ddd||IdH}|r4tdt|d|WStdWdStyV}ztd |WYd}~dSd}~ww) z Convert text to speech audio. Args: text: Text to convert to speech Returns: Audio bytes in WAV format or None if failed u"⚠️ Empty text provided for TTSNu 🎤 Converting text to speech: 2z...u ✅ TTS conversion successful -  bytesu❌ TTS conversion failedu(❌ Error in text-to-speech conversion: ) striprZwarningdebug_call_sarvam_tts_apirlenerror Exception)rr audio_dataerrrtext_to_speech"s"  zSarvamTTSService.text_to_speechc s>z|jddd}|d|jddddd }tjd d }tj|d 4Id H}|j|jd||d4Id H}|jdkr|Id H}d|vr{t |ddkr{|dd}t |} t dt | d| Wd Id HWd Id HWSt d| Wd Id HWd Id HWd S|Id H} t d|jd|  Wd Id HWd Id HWd S1Id HswYWd Id HWd S1Id HswYWd Stjy} zt d| WYd } ~ d Sd } ~ wty} zt d| WYd } ~ d Sd } ~ ww)z Make API call to Sarvam TTS service. Args: text: Text to convert Returns: Audio data bytes or None zapplication/jsonz audio/wav)zapi-subscription-keyz Content-TypeZAcceptzen-INg?g?g?i>)rZtarget_language_codeZspeakerZpitchZpaceZloudnessZspeech_sample_rate)total)timeoutNz/text-to-speech)headersjsonZaudiosru🎤 Received z bytes from Sarvam TTSu*❌ No audio data in Sarvam TTS response: u❌ Sarvam TTS API error z: u&❌ Network error calling Sarvam TTS: u)❌ Unexpected error calling Sarvam TTS: )r r aiohttpZ ClientTimeoutZ ClientSessionZpostr statusr&rbase64Z b64decoderrrrZ ClientErrorr) rrr%Z request_datar$ZsessionZresponseZ response_jsonZ audio_base64Z raw_audioZ error_textr rrrrAsf      2z%SarvamTTSService._call_sarvam_tts_apic s~z"d}||IdH}|rt|dkrtdWdStdWdSty>}ztd|WYd}~dSd}~ww) z Test connection to Sarvam TTS API. Returns: True if connection successful, False otherwise zHello, this is a test.Nru)✅ Sarvam TTS connection test successfulTu%❌ Sarvam TTS connection test failedFu)❌ Error testing Sarvam TTS connection: )r!rrrrr)rZ test_textrr rrrtest_connection~s  z SarvamTTSService.test_connectioncCsd|j|j|j|j|jdS)z Get information about the TTS service configuration. Returns: Dictionary with service information Z sarvam_tts)Z service_typer r rrr )r r rrr )rrrrget_service_infosz!SarvamTTSService.get_service_inforc Csz/|r t|dkr WdS|dddkrWdS|dddkr"WdStd t|d Wd StyJ}ztd |WYd}~dSd}~ww) z Validate that the audio data is in correct format. Args: audio_data: Audio bytes to validate Returns: True if valid, False otherwise ,FNsRIFF sWAVEu&🎤 Audio format validation passed - rTu#❌ Error validating audio format: )rrrrr)rrr rrrvalidate_audio_formats z&SarvamTTSService.validate_audio_format)N)__name__ __module__ __qualname____doc__strrrbytesr!rboolr+rrr,r1rrrrrs =r)r5Zasyncior(r*typingrrrioZwaveZconfigrZservices.log_utilsrrZsarvam_tts_servicerrrrs   2