o j @sbddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddZ e dZd d Zd d Ze d rAd ZeZne drKdZeZndZdZddededeefddZGddde ZedkreddZeeeeeejeeeeedddZededdedddZededddSdS)N)DictList)Version)BasePhonemizer) PunctuationcCsddlm}||duS)Nr)which)shutilr)namerr `/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/text/phonemizers/espeak_wrapper.pyis_tool s  r z.text-to-speech:\s(?P\d+\.\d+(\.\d+)?)cCstd}t|}|dS)Nzespeak --versionversion) subprocess getoutputespeak_version_patternsearchgroup)outputmatchr r r get_espeak_versions   rcCstd}|dS)Nzespeak-ng --version)rrsplit)rr r r get_espeakng_versions  r espeak-ngespeakF espeak_libargsreturncCs|dddg}||tdt|tj|tjtjdU}t|j j d}|sE|j |j r4|j |j r<|j |WdSg}|D]}||qI|j |j r^|j |j rf|j |Wd|S1suwY|S)z$Run espeak with the given arguments.z-qz-b1zespeakng: executing %s)stdoutstderrN)extendloggingdebugreprrPopenPIPESTDOUTiterrreadlinecloser stdinappendwait)rrsynccmdpresres2liner r r _espeak_exe/sF           r5cseZdZdZeZeZde dfde ffdd Z e ddZ e d d Ze jd dZ dd dZeddZd de de d e fddZd!ddZed efddZd e fddZeddZZS)"ESpeaku\ESpeak wrapper calling `espeak` or `espeak-ng` from the command-line the perform G2P Args: language (str): Valid language code for the used backend. backend (str): Name of the backend library to use. `espeak` or `espeak-ng`. If None, set automatically prefering `espeak-ng` over `espeak`. Defaults to None. punctuations (str): Characters to be treated as punctuation. Defaults to Punctuation.default_puncs(). keep_puncs (bool): If True, keep the punctuations after phonemization. Defaults to True. Example: >>> from TTS.tts.utils.text.phonemizers import ESpeak >>> phonemizer = ESpeak("tr") >>> phonemizer.phonemize("Bu Türkçe, bir örnektir.", separator="|") 'b|ʊ t|ˈø|r|k|tʃ|ɛ, b|ɪ|r œ|r|n|ˈɛ|c|t|ɪ|r.' NTlanguagecsZ|jdur td|j|_|dkrd}|dkrd}tj|||d|dur+||_dSdS)NzI [!] No espeak backend found. Install espeak-ng or espeak to your system.enen-uszzh-cncmn) punctuations keep_puncs) _ESPEAK_LIB Exceptionbackendsuper__init__)selfr7r?r;r< __class__r r rAps  zESpeak.__init__cC|jSN)r=rBr r r r?zESpeak.backendcCrErF) _ESPEAK_VERrGr r r backend_versionrHzESpeak.backend_versioncCs:|dvr td|||_|dkrt|_dSt|_dS)N)rrzUnknown backend: %sr)r>r=rrrI)rBr?r r r r?s  rcCs<tdr d|_t|_dStdrd|_t|_dStd)Nrrz?Cannot set backend automatically. espeak-ng or espeak not found)r r=rrIrr>rGr r r auto_set_espeak_libs  zESpeak.auto_set_espeak_libcCsdS)Nrr r r r r r sz ESpeak.name|Ftext separatorcCsd|jg}|r|jdkr|dn%|dn|jdkr3t|jtdkr-|dn |dn|d|rA|d|||d}t|j|dd D]}td t || d  }t d d|}|| 7}qP|d |S)umConvert input text to phonemes. Args: text (str): Text to be converted to phonemes. tie (bool, optional) : When True use a '͡' character between consecutive characters of a single phoneme. Else separate phoneme with '_'. This option requires espeak>=1.49. Default to False. z-vrz--ipa=1z--ipa=3z1.48.15z--tie=%sTr/line: %sutf8z\(.+?\)_) _languager?r-rrJr5r=r#r$r%decodestripresubreplace)rBrMrNtierphonemesr4 ph_decodedr r r phonemize_espeaks(           zESpeak.phonemize_espeakcCs|j||ddS)NF)rZ)r])rBrMrNr r r _phonemizeszESpeak._phonemizecCstduriSdg}i}d}tt|ddD])}|d}|dkr1|}|d}|d}|||<td t||d7}q|S) zsGet a dictionary of supported languages. Returns: Dict: Dictionary of language codes. Nz--voicesrTrPrRrrQ)_DEF_ESPEAK_LIBr5rUrVrr#r$r%)rlangscountr4cols lang_code lang_namer r r supported_languagess zESpeak.supported_languagescCsLdg}t|j|ddD]}|dd}tdt||SdS)zmGet the version of the used backend. Returns: str: Version of the used backend. z --versionTrPrRrQN)r5r?rUrVrr#r$r%)rBrr4r r r r r s zESpeak.versioncCstdptdS)z-Return true if ESpeak is available else falserr)r )clsr r r is_availableszESpeak.is_available)rN)rLFrF)__name__ __module__ __qualname____doc__r`r=_DEF_ESPEAK_VERrIr default_puncsstrrApropertyr?rJsetterrK staticmethodr r]r^rrfr classmethodri __classcell__r r rCr r6Ss*      4 r6__main__r9)r7)r7r<`zhello how are you today?T)F)r#rWrtypingrrpackaging.versionr#TTS.tts.utils.text.phonemizers.baserTTS.tts.utils.text.punctuationrr compilerrrr`rnrpr5r6rjeprintrfr r7r ri phonemizer r r r sD    $*