qi (dZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z ddl mZdejd<ddlmZdd lmZdd lmZmZmZmZdd lmZmZmZmZdd lmZdd l m!Z! ddl"Z#ddl$m%cm&cm'Z(eGddeZ-GddeZ.y#e)$r7Z*ejVde*ejVde,de*dZ*[*wwxYw)zNVIDIA Riva text-to-speech service implementation. This module provides integration with NVIDIA Riva's TTS services through gRPC API for high-quality speech synthesis. N) dataclassfield)AnyAsyncGenerator AsyncIterator GeneratorMappingOptional) traced_ttsfalseGRPC_ENABLE_FORK_SUPPORT)logger) BaseModel) ErrorFrameFrame StartFrameTTSAudioRawFrame) NOT_GIVEN TTSSettings _NotGiven_warn_deprecated_param) TTSService)Languagez Exception: zNIn order to use NVIDIA Riva TTS, you need to `pip install pipecat-ai[nvidia]`.zMissing module: c6eZdZUdZedZeezed<y)NvidiaTTSSettingszdSettings for NvidiaTTSService. Parameters: quality: Audio quality setting (0-100). ctSN)rM/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/nvidia/tts.pyzNvidiaTTSSettings.5sYr)default_factoryqualityN) __name__ __module__ __qualname____doc__rr#intr__annotations__rrr rr-s %5FGGS9_GrrceZdZUdZeZeed<GddeZdddddd ddd d d e d e de e de e de e e fde ede ede ffdZde fdZdedee efffd ZdZdZdeffd Zede de deedffd ZxZS)!NvidiaTTSServicezNVIDIA Riva text-to-speech service. Provides high-quality text-to-speech synthesis using NVIDIA Riva's cloud-based TTS models. Supports multiple voices, languages, and configurable quality settings. _settingscNeZdZUdZej Zeeed<dZ ee ed<y)NvidiaTTSService.InputParamsaQInput parameters for Riva TTS configuration. .. deprecated:: 0.0.105 Use ``NvidiaTTSSettings`` directly via the ``settings`` parameter instead. Parameters: language: Language code for synthesis. Defaults to US English. quality: Audio quality setting (0-100). Defaults to 20. languager#N) r$r%r&r'rEN_USr/r r)r#r(rrr InputParamsr.Cs* (0~~(8$5!###rr2zgrpc.nvcf.nvidia.com:443Nz$877104f7-e885-42b9-8de8-f6e4c6303969zmagpie-tts-multilingual) function_id model_nameT)servervoice_id sample_ratemodel_function_mapparamssettingsuse_sslapi_keyr5r6r7r8r9r:r;c t|jddtjd} |t dtd|| _|Lt dt|s:|j |j | _|j|j| _|| j|t |(d |d d | d | ||_ ||_ |jd |_ ||_d|_d|_y) a+Initialize the NVIDIA Riva TTS service. Args: api_key: NVIDIA API key for authentication. server: gRPC server endpoint. Defaults to NVIDIA's cloud endpoint. voice_id: Voice model identifier. Defaults to multilingual Aria voice. .. deprecated:: 0.0.105 Use ``settings=NvidiaTTSSettings(voice=...)`` instead. sample_rate: Audio sample rate. If None, uses service default. model_function_map: Dictionary containing function_id and model_name for the TTS model. params: Additional configuration parameters for TTS synthesis. .. deprecated:: 0.0.105 Use ``settings=NvidiaTTSSettings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. use_ssl: Whether to use SSL for the NVIDIA Riva server. Defaults to True. **kwargs: Additional arguments passed to parent TTSService. r4zMagpie-Multilingual.EN-US.Ariar0)modelvoicer/r#Nr6r?r9T)r7push_start_framepush_stop_framesr:r3r)rgetrr1rr?r/r# apply_updatesuper__init___server_api_key _function_id_use_ssl_service_config) selfr<r5r6r7r8r9r:r;kwargsdefault_settings __class__s r rEzNvidiaTTSService.__init__QsN-$((62^^     ":/@' J%-  "   "8-> ???.06$->>-/5~~$,    ) )( 3  #!!%      .22=A   rr>cKddl}|j5|jd|jdtddddy#1swYyxYww)aSet the TTS model. .. deprecated:: 0.0.104 Model cannot be changed after initialization for NVIDIA Riva TTS. Set model and function id in the constructor instead, e.g.:: NvidiaTTSService( api_key=..., model_function_map={"function_id": "", "model_name": ""}, ) Args: model: The model name to set. rNalwaysa'set_model' is deprecated. Model cannot be changed after initialization for NVIDIA Riva TTS. Set model and function id in the constructor instead, e.g.: NvidiaTTSService(api_key=..., model_function_map={'function_id': '', 'model_name': ''})) stacklevel)warningscatch_warnings simplefilterwarnDeprecationWarning)rLr>rTs r set_modelzNvidiaTTSService.set_modelsX   $ $ &   ! !( + MMK#     sA*A  A AAdeltareturnclKt||d{}|s|S|j||S7w)zgApply a settings delta. Settings are stored but not applied to the active connection. N)rD_update_settings _warn_unhandled_updated_settings)rLrZchangedrOs r r]z!NvidiaTTSService._update_settingss; 077N --g6 8s 424c|jyd|jgdd|jgg}tjj d|j |j|}tjj||_y)Nz function-id authorizationzBearer ) rJrHrGrivaclientAuthrIrFSpeechSynthesisService)rLmetadataauths r _initialize_clientz#NvidiaTTSService._initialize_clientss == $ D-- .  7 8 {{dmmT\\8L ::4@ rc|jsy|jjjtjj j j}|Sr)rJstubGetRivaSynthesisConfigrbrcproto riva_tts_pb2RivaSynthesisConfigRequest)rLconfigs r _create_synthesis_configz)NvidiaTTSService._create_synthesis_configsK}} ##:: KK   * * E E G  rframecKt||d{|j|j|_t j d|jjy7Vw)zStart the Cartesia TTS service. Args: frame: The start frame containing initialization parameters. Nz)Initialized NvidiaTTSService with model: ) rDstartrhrprKrdebugr,r>)rLrqrOs r rszNvidiaTTSService.starts[ gmE""" !446  @AUAU@VWX #sA/A-AA/text context_idcN Kdttjddfffd }d dttjf fd } jJdj Jdt jdd tj|d{}||23d{}jd{t|jjd | }|J7V7G716jd{7y#tj$r6}t j d t#d |Yd}~yd}~wt$$r8}t j d|t#d |Yd}~yd}~wwxYww)aGenerate speech from text using NVIDIA Riva TTS. Args: text: The text to synthesize into speech. context_id: The context ID for tracking audio frames. Yields: Frame: Audio frames containing the synthesized speech data. r[Nc jjjjjjj djj i}|S)N)sample_rate_hzzero_shot_audio_prompt_filezero_shot_qualitycustom_dictionary)rJsynthesize_onliner,r?r/r7r#) responsesrLrus r read_audio_responsesz6NvidiaTTSService.run_tts..read_audio_responsess] 77$$''#//,0"&.."8"8"$8I rc8 t|S#t$rYywxYwr)next StopIteration)its r async_nextz,NvidiaTTSService.run_tts..async_nexts" Bx   s c`K tj|d{}|y|'7 wr)asyncio to_thread)iteratoritemrs r async_iteratorz0NvidiaTTSService.run_tts..async_iterators4$..z8DD< Ds .,.zTTS service not initializedz#Synthesis configuration not createdz: Generating TTS [])audior7 num_channelsrvz# timeout waiting for audio responsez error: )errorz exception: )rrttsSynthesizeSpeechResponserrJrKrrtrrstop_ttfb_metricsrrr7start_tts_usage_metrics TimeoutErrorrr Exception) rLrurvrrr~resprqers `` @r run_ttszNvidiaTTSService.run_ttss i0M0MtUY0Y&Z    mD2C:3C>6D C< .D:C><D>DDDF%F"-,EF% F"*.FF%F""F%)r$r%r&r'rSettingsr)rr2strr r(r boolrErYdictrr]rhrprrsr rrr __classcell__)rOs@r r+r+8s1!H   $i $$1"&%)A31 )-04NN N 3- N c] N$CH-N%N,-NN`S8 ,= $sCx.  A Y Y<9#<93<9>%QU+;V<9<9rr+)/r'ros dataclassesrrtypingrrrrr r (pipecat.utils.tracing.service_decoratorsr environlogururpydanticrpipecat.frames.framesrrrrpipecat.services.settingsrrrrpipecat.services.tts_servicerpipecat.transcriptions.languager riva.clientrbriva.client.proto.riva_tts_pb2rcrlrmrModuleNotFoundErrorrrrrr+rrr rs  (SS?*1 %& `_34,11 H H Ho9zo9#,FLL;qc"#FLLab &qc* ++,s)BC2C  C