o
    
jr                     @   sl   d dl mZ d dlZd dlZddlmZ dedee fddZd	edefd
dZddededefddZ	dS )    )ListN   )PINYIN_DICTtextreturnc                 C   s(   t j| t jjddd}dd |D }|S )NFT)style	heteronymneutral_tone_with_fivec                 S   s   g | ]	}|D ]}|qqS  r
   ).0sublistitemr
   r
   a/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/text/chinese_mandarin/phonemizer.py
<listcomp>   s    z0_chinese_character_to_pinyin.<locals>.<listcomp>)pypinyinpinyinStyleTONE3)r   pinyinspinyins_flat_listr
   r
   r   _chinese_character_to_pinyin	   s   r   r   c                 C   s.   | d d }| d }t |dgd }|| S )N r   )r   get)r   segmenttonephonemer
   r
   r   _chinese_pinyin_to_phoneme   s   r   |	seperatorc                 C   sf   t j| dd}d|}t|}g }|D ]}|d dv r't|}|t|7 }q|t|7 }q||S )NF)HMM r   12345)jiebacutjoinr   r   list)r   r   tokenized_textpinyined_textresultstokenpinyin_phonemesr
   r
   r   chinese_text_to_phonemes   s   

r,   )r   )
typingr   r#   r   pinyinToPhonemesr   strr   r   r,   r
   r
   r
   r   <module>   s    