o j@sddlZddlmZddlZddlmZddlmZddlm Z e ddZ Gdd d eZ ed krxe d d Zeeeeeejeeeee d d dZededde d ddZededddSdS)N)List)IPA)BasePhonemizer) PunctuationguɡcseZdZdZedddfdeffdd ZeddZ dd ed ed efd dZ ddZ ddZ ed e fddZddZeddZZS)Gruutu]Gruut wrapper for G2P Args: language (str): Valid language code for the used backend. 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. use_espeak_phonemes (bool): If true, use espeak lexicons instead of default Gruut lexicons. Defaults to False. keep_stress (bool): If true, keep the stress characters after phonemization. Defaults to False. Example: >>> from TTS.tts.utils.text.phonemizers.gruut_wrapper import Gruut >>> phonemizer = Gruut('en-us') >>> phonemizer.phonemize("Be a voice, not an! echo?", separator="|") 'b|i| ə| v|ɔ|ɪ|s, n|ɑ|t| ə|n! ɛ|k|o|ʊ?' TFlanguagecs"tj|||d||_||_dS)N) punctuations keep_puncs)super__init__use_espeak_phonemes keep_stress)selfrr r r r __class___/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/text/phonemizers/gruut_wrapper.pyr )s zGruut.__init__cCsdS)Ngruutrrrrrname5sz Gruut.name|text separatorreturnc sg}tj||j|jdD]C}|D]>}|jr(|r |d|jq||jgq|jrNg}|jD]}|js:t |}| t }|rF| |q0|rN||qq fdd|D} d| } | S)uConvert input text to phonemes. Gruut phonemizes the given `str` by seperating each phoneme character with `separator`, even for characters that constitude a single sound. It doesn't affect 🐸TTS since it individually converts each character to token IDs. Examples:: "hello how are you today?" -> `h|ɛ|l|o|ʊ| h|a|ʊ| ɑ|ɹ| j|u| t|ə|d|e|ɪ` 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. )langespeakcsg|]}|qSr)join).0 word_phonemesrrr isz)Gruut.phonemize_gruut.. )r sentencesrr is_breakappendrphonemesrrwithout_stress translateGRUUT_TRANS_TABLEextendr) rrrtieph_listsentencewordr word_phonemeph_wordsphrr rphonemize_gruut9s.     zGruut.phonemize_gruutcCs|j||ddS)NF)r+)r2)rrrrrr _phonemizemszGruut._phonemizecCs t|S)z6Returns True if `language` is supported by the backend)ris_language_supported)rrrrris_supported_languageps zGruut.is_supported_languagecCs ttS)zmGet a dictionary of supported languages. Returns: List: List of language codes. )listrget_supported_languagesrrrrsupported_languagests zGruut.supported_languagescCstjS)zmGet the version of the used backend. Returns: str: Version of the used backend. )r __version__)rrrrversion}sz Gruut.versioncCstjdduS)z-Return true if ESpeak is available else falserN) importlibutil find_spec)clsrrr is_availableszGruut.is_available)rF)__name__ __module__ __qualname____doc__r default_puncsstrr staticmethodrr2r3r5rr8r: classmethodr? __classcell__rrrrrs& 4r__main__zen-us)rF)rr `zhello how are you today?Tzhello how, are you today?)r;typingrr gruut_ipar#TTS.tts.utils.text.phonemizers.baserTTS.tts.utils.text.punctuationrrE maketransr)rr@eprintr8r:rrr? phonemizerrrrs(     }