o
    i                     @   sR   d dl mZ d dlmZ zd dlmZ W n ey   dZY nw G dd dZdS )    )
Collection)jaconv)vietnamese_cleanersNc                   @   s6   e Zd ZdZd
dee fddZdedefdd	ZdS )TextCleanerzText cleaner.

    Examples:
        >>> cleaner = TextCleaner("tacotron")
        >>> cleaner("(Hello-World);   &  jr. & dr.")
        'HELLO WORLD, AND JUNIOR AND DOCTOR'

    Ncleaner_typesc                 C   s6   |d u r	g | _ d S t|tr|g| _ d S t|| _ d S N)r   
isinstancestrlist)selfr    r   L/home/ubuntu/.local/lib/python3.10/site-packages/funasr/tokenizer/cleaner.py__init__   s
   

zTextCleaner.__init__textreturnc                 C   sr   | j D ]3}|dkr
q|dkrt|}q|dkr&td u r tdt|}q|dkr0t|}qtd| |S )Ntacotronr   
vietnamesezPlease install undertheseakorean_cleanerzNot supported: type=)r   r   	normalizer   RuntimeErrorvietnamese_cleanerKoreanCleanernormalize_text)r   r   tr   r   r   __call__    s   
zTextCleaner.__call__r   )__name__
__module____qualname____doc__r   r	   r   r   r   r   r   r   r      s    		r   )typingr   r   r   r   ImportErrorr   r   r   r   r   <module>   s    