
    vj5                         d dl Z d dlmZmZmZmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZmZ d dlmZmZmZ dd	lmZ dd
lmZ dgZ ej        ej        e	j                   G d de                      ZdS )    N)AnyDictListUnion)process_data)Preprocessors)Model)0TtsDataPreprocessorAudioConfigNotExistsException(TtsDataPreprocessorDirNotExistsException)Fields
FrameworksTasks   )Preprocessor)PREPROCESSORSKanttsDataPreprocessor)	group_keymodule_namec                   4    e Zd Zd Z	 	 	 	 ddZ	 	 	 	 ddZdS )	r   c                     d S N )selfs    l/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/preprocessors/tts.py__init__zKanttsDataPreprocessor.__init__   s        F7PinYinFNc           	      <    |                      |||||||           d S r   )do_data_process)r   data_dir
output_diraudio_config_pathspeaker_nametarget_langskip_scriptse_models           r   __call__zKanttsDataPreprocessor.__call__   s;     	Xz3D);X	O 	O 	O 	O 	Or   c           	      D   t           j                            |          st          d          t           j                            |          st          d          t           j                            |          st	          d          t          |||||||           d S )Nz$Preprocessor: dataset dir not existsz#Preprocessor: output dir not existsz%Preprocessor: audio config not exists)ospathexistsr   r
   r   )r   datadir	outputdiraudio_configr$   
targetLangr&   r'   s           r   r    z&KanttsDataPreprocessor.do_data_process'   s     w~~g&& 	8:68 8 8w~~i(( 	7:57 7 7w~~l++ 	9B79 9 9Wi|h	8 	8 	8 	8 	8r   )r   r   FN)__name__
__module____qualname__r   r(   r    r   r   r   r   r      sk        
   #%"	O 	O 	O 	O &*#+$)!%8 8 8 8 8 8r   )r*   typingr   r   r   r   kantts.preprocess.data_processr   modelscope.metainfor   modelscope.models.baser	   %modelscope.utils.audio.tts_exceptionsr
   r   modelscope.utils.constantr   r   r   baser   builderr   __all__register_moduletext_to_speechkantts_data_preprocessorr   r   r   r   <module>r@      sB   
			 ) ) ) ) ) ) ) ) ) ) ) ) 7 7 7 7 7 7 - - - - - - ( ( ( ( ( (. . . . . . . . @ ? ? ? ? ? ? ? ? ?       " " " " " "#
$ "68 8 8"8 "8 "8 "8 "8\ "8 "88 8"8 "8 "8r   