o i'@spddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z e GdddZ dS)N)OptionalListTuple) load_dotenvc@seZdZddZdedefddZdeefddZd edeefd d Zd ede fd dZ dded eedeefddZ dS) TTSServicecCstjtjtjt|_||_tdd}tdd}| ||_ | ||_ d|_ t d|js;t dnt d|j|j rMtj|j sWt d |j d nt d |j |jrdd nd|_ |j rpt d dSt ddS)zk Initializes the Piper TTS service with automatic path correction and absolute resolution. PIPER_MODEL_PATHz&assets/models/en_US-lessac-medium.onnxPIPER_CONFIG_PATHz+assets/models/en_US-lessac-medium.onnx.jsonFu! 🔍 Piper TTS Diagnostic Check:u8 - Binary: ❌ NOT FOUND (Check PIPER_EXECUTABLE_PATH)u - Binary: ✅ FOUND at u% - Model: ❌ NOT FOUND (Checked: )u - Model: ✅ FOUND at Tu$🚀 Piper TTS Service Fully Active u9⚠️ Piper TTS Service Disabled (Missing requirements) N)ospathdirnameabspath__file__base_dir_resolve_piper_executable piper_pathgetenv_make_absolute model_path config_path availableprintexists)self model_env config_envrC/var/www/eduai.edurigo.com/roleplay/testing/services/tts_service.py__init__s$        zTTSService.__init__r returncCs(|sdStj|r |Stj|j|S)z@Converts relative paths to absolute paths based on project root.)r r isabsjoinr)rr rrrr2szTTSService._make_absolutecsddg}td}|r9tj|r%||vr$tj|r$t|tjr$|Snt|r9t fdd|Ds9SdD]}t|rSt fdd|DsSSq;dD]tjrmttjrm|vrmSqVd S) z Strategically searches for the piper binary in common production paths. STRICTLY blocks /usr/bin/piper to avoid the Ubuntu GTK conflict. z/usr/bin/piperz /bin/piperPIPER_EXECUTABLE_PATHc3|]}|vVqdSNr.0b)resolvedrr Iz7TTSService._resolve_piper_executable..)z piper-ttspiperc3r$r%rr&)r rrr*Or+)z/usr/local/bin/piper-ttsz/usr/local/bin/piperz/opt/piper/piperz/opt/piper/piper/piperN) r rr r!raccessX_OKshutilwhichany)r BLACKLISTenv_pathnamer)r r)rr8s(  "  "z$TTSService._resolve_piper_executablercCs|d}tj|r |SdS)z7Returns the .json config path for a model if it exists.z.jsonN)r r r)rrrrrr_get_config_pathZszTTSService._get_config_pathvoicec stj|jdd}tj|sdS||dd|ddg}td|}t|dkrB| d|dd | d|dd t t |}|D]}tj||d }tj|rf|| |fSqK|g|}hd fd d |D}|D]/}tj|d|d} t| } | r| d} td|dtj| | | | fSq{|jrtj|jrtd|dt|dtj|j|j|jfSttj|d} | r| d} tdtj| | | | fSdS)z Ultra-robust resolution for Piper voice models. Tries exact match, naming variants, and globbing to find the best possible file. assetsmodels)NN_-z[-_]Nz.onnx> low-polyengbinuslowhighr6mediumcs(g|]}|vrt|dkr|qS))lowerlen)r't stopwordsrr s(z3TTSService._resolve_voice_model..*z*.onnxru🎙️ Fuzzy match found for 'z': u⚠️ Piper voice 'z' not found (tried z, variants + glob). Falling back to default: u!⚠️ Using emergency fallback: )r r r"rrreplaceresplitrGappendlistdictfromkeysr5globrbasenamerr)rr6 models_dirvariantspartsvr search_termstermpatternmatchesmatchavailable_modelsfallbackrrIr_resolve_voice_model_sJ        ( zTTSService._resolve_voice_modelNtextcCs|js tddS|r|stddS|r+||\}}|s*td|ddSn|j}|j}|dddd}d}z}z tjd d d  }|j }Wdn1sXwY|j d |d |g}|rtt j |rt|d|gtd|pydd|dddtj|tjtjtjddd} | j|dd\} } | jdkrtd| jd| WW|rt j |rzt |WdStyYdSwdSdSt j |r(t j |dkr(t|d=} t| d} tdt| d| WdWW|rt j |rzt |WStyYSwSS1s#wYtdWW|rLt j |rNzt |WdStyKYdSwdSdStjytd YW|r|t j |r~zt |WdSty{YdSwdSdSty}z1td!t|WYd}~W|rt j |rzt |WdStyYdSwdSdSd}~ww|rt j |rzt |WwtyYwwww)"a Generates high-quality audio base64 string using the specified Piper voice model. Thread-safe and optimized using temporary file isolation. Args: text: The text to synthesize. voice: Optional Piper voice model name (e.g. "en_US-hfc_female-medium"). If provided, uses that model; otherwise uses the default model. uE⚠️ TTS skipped: Service not available (Check binary/model paths).Nu(⚠️ TTS skipped: Empty text provided.u!⚠️ TTS skipped: Voice model 'z' not found in assets/models/."r 'z.wavF)suffixdeletez--modelz --output_filez--configu🎙️ Generating voice (defaultz ) for text: 2z...Tzutf-8)stdinstdoutstderrrbencoding)inputtimeoutru ❌ Piper Execution Error (Code z): rbu"✅ Voice generated successfully (z chars)u"❌ Piper generated an empty file.u(❌ Piper process timed out (30s limit).u'❌ Exception in Piper TTS processing: )rrstriprarrrMtempfileNamedTemporaryFiler4rr r rextend subprocessPopenPIPE communicate returncoderemove Exceptiongetsizeopenbase64 b64encodereaddecoderGTimeoutExpiredstr)rrbr6 active_model active_config clean_text temp_wav_pathtemp_wavcommandprocessrjrk audio_fileb64errrgenerate_voice_base64s  "     z TTSService.generate_voice_base64r%) __name__ __module__ __qualname__rrrrrr5tuplerarrrrrr s$"$s