
    9i5                         S SK r S SKJrJrJrJr  S SKJr  S SKJ	r	  S SK
Jr  S SKJrJr  S SKJrJrJr  SS	KJr  SS
KJr  S/r\R0                  " \R2                  \	R4                  S9 " S S\5      5       rg)    N)AnyDictListUnion)process_data)Preprocessors)Model)0TtsDataPreprocessorAudioConfigNotExistsException(TtsDataPreprocessorDirNotExistsException)Fields
FrameworksTasks   )Preprocessor)PREPROCESSORSKanttsDataPreprocessor)	group_keymodule_namec                   >    \ rS rSrS r    SS jr    SS jrSrg)r      c                     g N )selfs    \/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/preprocessors/tts.py__init__KanttsDataPreprocessor.__init__   s        Nc           	      ,    U R                  XUXEXg5        g r   )do_data_process)r   data_dir
output_diraudio_config_pathspeaker_nametarget_langskip_scriptse_models           r   __call__KanttsDataPreprocessor.__call__   s     	X3D)	Or   c           	      :   [         R                  R                  U5      (       d  [        S5      e[         R                  R                  U5      (       d  [        S5      e[         R                  R                  U5      (       d  [	        S5      e[        XX4XVU5        g )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    &KanttsDataPreprocessor.do_data_process'   s~     ww~~g&&:68 8ww~~i((:57 7ww~~l++B79 9Wh	8r   r   )F7PinYinFN)__name__
__module____qualname____firstlineno__r   r(   r    __static_attributes__r   r   r   r   r      s.    
 #%"	O &*#+$)!%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>rF      sm    
 ) ) 7 - (. @ ?  "#
$ ""668"8\ "88"8r   