o 0j@sddlZddlmZmZmZmZmZddlZddl m Z m Z ddl m Z ddlmZddlmZddlmZd d lmZejGd d d eZdS) N)AnyDictListOptionalUnion) HPIConfigPaddlePredictorOption)TextToPinyinResult)Fastspeech2Result) PwganResult) benchmark) BasePipelinecseZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffddZ de eeeejeejfd efddZde eeefd efddZde eeefd efddZZS)TextToSpeechPipelinez Text to Speech Pipeline PipelineZtext_to_speechNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc sltjd||||||d||dd} || |_|dd} || |_|dd} || |_dS)aInitializes the text-to-speech pipeline. Args: config (Dict): Configuration dictionary containing model and other parameters. device (Optional[str], optional): The device to use for prediction. Defaults to `None`. engine (Optional[str], optional): Inference engine. Defaults to `None`. engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. Defaults to `None`. pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options. Defaults to `None`. use_hpip (bool, optional): Whether to use HPIP. Defaults to `False`. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): HPIP configuration. Defaults to `None`. rZ SubModulesZ TextToPinyinZTextToSpeechAcousticZTextToSpeechVocoderN)super__init__Z create_modeltext_to_pinyin_modeltext_to_speech_acoustic_modeltext_to_speech_vocoder_model) selfrrrrrrrkwargsZtext_to_pinyin_model_configZ$text_to_speech_acoustic_model_configZ#text_to_speech_vocoder_model_config __class__rt/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/text_to_speech/pipeline.pyr"s0  zTextToSpeechPipeline.__init__inputc ksg}t|trW|drStj|std|z"t|ddd}dd|D}Wdn1s6wYWnt yR}z t d |d |d}~ww|g}nmt|t r|D]Z}t|tr|drtj|swtd |z%t|ddd}| d d|DWdn1swYWq^t y}z t d |d |d}~ww| |q^n t d t|d|std|D]}||ddg}||dg} || EdHqdS)anPredicts speech recognition results for the given input. Args: input (Union[str, list[str], np.ndarray, list[np.ndarray]]): The input audio or path. **kwargs: Additional keyword arguments that can be passed to the function. Returns: PwganResult: The predicted pwgan results, support str and json output. z.txtz(The specified text file does not exist: rzutf-8)encodingcSg|]}|qSrstrip.0linerrr$ iz0TextToSpeechPipeline.predict..Nz)An error occurred while reading the file z: z4The specified text file in the list does not exist: cSr(rr)r+rrr$r.{r/zUnsupported input type: z$. Expected str, list, or np.ndarray.zs