o i4 @sPdZddlZddlmZmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZmZmZdd lmZmZmZdd lmZdd lmZmZdd lmZzddlZdd lm Z m!Z!Wn e"yZ#ze $de#e $de%de#dZ#[#wwdedee&fddZ'eGdddeZ(GdddeZ)Gddde)Z*dS)zAWS Polly text-to-speech service implementation. This module provides integration with Amazon Polly for text-to-speech synthesis, supporting multiple languages, voices, and SSML features. N) dataclassfield)AsyncGeneratorListOptional)logger) BaseModel)create_stream_resampler) ErrorFrameFrameTTSAudioRawFrame) NOT_GIVEN TTSSettings _NotGiven) TTSService)Languageresolve_language) traced_tts) BotoCoreError ClientErrorz Exception: zHIn order to use AWS services, you need to `pip install pipecat-ai[aws]`.zMissing module: languagereturncCslitjdtjdtjdtjdtjdtjdtjdtjdtj dtj d tj d tj d tj d tjd tjdtjd tjditjdtjdtjdtjdtjdtjdtjdtjdtjdtjdtjdtjdtjdtjdtj dtj!dtj"dtj#dtj$dtj%d tj&d!tj'd"tj(d#tj)d$tj*d%tj+d&tj,d'tj-d'i }t.||d(d)S)*zConvert a Language enum to AWS Polly language code. Args: language: The Language enum value to convert. Returns: The corresponding AWS Polly language code, or None if not supported. arbzar-AEzca-ESzcmn-CNzyue-CNzcs-CZzda-DKznl-NLznl-BEen-USzen-AUzen-GBzen-INzen-NZzen-ZAzfi-FIzfr-FRzfr-BEzfr-CAzde-DEzde-ATzde-CHzhi-INzis-ISzit-ITzja-JPzko-KRznb-NOzpl-PLzpt-PTzpt-BRzro-ROzru-RUzes-ESzes-MXzes-USzsv-SEztr-TRzcy-GBF) use_base_code)/rARAR_AECAZHYUEYUE_CNCSDANLNL_BEENEN_AUEN_GBEN_INEN_NZEN_USEN_ZAFIFRFR_BEFR_CADEDE_ATDE_CHHIISITJAKONONBNB_NOPLPTPT_BRPT_PTRORUESES_MXES_USSVTRCYCY_GBr)r LANGUAGE_MAPrIL/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/aws/tts.pylanguage_to_aws_language(s     !"$&(*,./0245HrKc@seZdZUdZedddZeeBed<edddZ eeBed<edddZ eeBed <ed ddZ eeBed <ed ddZ e eeBed <dS)AWSPollyTTSSettingsaGSettings for AWSPollyTTSService. Parameters: engine: TTS engine to use ('standard', 'neural', etc.). pitch: Voice pitch adjustment (for standard engine only). rate: Speech rate adjustment. volume: Voice volume adjustment. lexicon_names: List of pronunciation lexicons to apply. cCtSNr rIrIrIrJzAWSPollyTTSSettings.)default_factoryenginecCrMrNrOrIrIrIrJrPrQpitchcCrMrNrOrIrIrIrJrPrQratecCrMrNrOrIrIrIrJrPrQvolumecCrMrNrOrIrIrIrJrPrQ lexicon_namesN)__name__ __module__ __qualname____doc__rrSstrr__annotations__rTrUrVrWrrIrIrIrJrL|s  "rLcseZdZUdZeZeed<GdddeZdddddddddde e de e d e e d e e d e e d e e d e ede effddZ de fddZdede e fddZde de fddZede de deedffddZZS)AWSPollyTTSServicezAWS Polly text-to-speech service. Provides text-to-speech synthesis using Amazon Polly with support for multiple languages, voices, SSML features, and voice customization options including prosody controls. _settingsc@sxeZdZUdZdZeeed<ej Z eeed<dZ eeed<dZ eeed<dZ eeed<dZeeeed<dS) zAWSPollyTTSService.InputParamsa9Input parameters for AWS Polly TTS configuration. .. deprecated:: 0.0.105 Use ``AWSPollyTTSService.Settings`` directly via the ``settings`` parameter instead. Parameters: engine: TTS engine to use ('standard', 'neural', etc.). language: Language for synthesis. Defaults to English. pitch: Voice pitch adjustment (for standard engine only). rate: Speech rate adjustment. volume: Voice volume adjustment. lexicon_names: List of pronunciation lexicons to apply. NrSrrTrUrVrW)rXrYrZr[rSrr\r]rr%rrTrUrVrWrrIrIrIrJ InputParamss r`N)api_keyaws_access_key_idaws_session_tokenregionvoice_id sample_rateparamssettingsrarbrcrdrerfrgrhc s|jddddddddd} |dur|dd|| _|durB|d|sB|j| _|jr/|jnd| _|j| _|j| _|j| _|j| _|durK| |t j d|dd| d | |p_t d |pet d |pkt d |prt d dd|_t|_t|_dS)a^Initializes the AWS Polly TTS service. Args: api_key: AWS secret access key. If None, uses AWS_SECRET_ACCESS_KEY environment variable. aws_access_key_id: AWS access key ID. If None, uses AWS_ACCESS_KEY_ID environment variable. aws_session_token: AWS session token for temporary credentials. region: AWS region for Polly service. Defaults to 'us-east-1'. voice_id: Voice ID to use for synthesis. Defaults to 'Joanna'. .. deprecated:: 0.0.105 Use ``settings=AWSPollyTTSService.Settings(voice=...)`` instead. sample_rate: Audio sample rate. If None, uses service default. params: Additional input parameters for voice customization. .. deprecated:: 0.0.105 Use ``settings=AWSPollyTTSService.Settings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to parent TTSService class. NJoannar)modelvoicerrSrTrUrVrWrerkrgT)rfpush_start_framepush_stop_framesrhAWS_ACCESS_KEY_IDAWS_SECRET_ACCESS_KEYAWS_SESSION_TOKEN AWS_REGIONz us-east-1)rbaws_secret_access_keyrc region_namerI)Settings"_warn_init_param_moved_to_settingsrkrSrrTrUrVrW apply_updatesuper__init__osgetenv _aws_paramsaioboto3Session _aws_sessionr _resampler) selfrarbrcrdrerfrgrhkwargsdefault_settings __class__rIrJrxsN$       zAWSPollyTTSService.__init__rcCsdS)zCheck if this service can generate processing metrics. Returns: True, as AWS Polly service supports metrics generation. TrI)rrIrIrJcan_generate_metricssz'AWSPollyTTSService.can_generate_metricsrcCst|S)zConvert a Language enum to AWS Polly language format. Args: language: The language to convert. Returns: The AWS Polly-specific language code, or None if not supported. )rK)rrrIrIrJlanguage_to_service_languages z/AWSPollyTTSService.language_to_service_languagetextcCsd}|jj}|d|d7}g}|jjdkr%|jjr%|d|jjd|jjr4|d|jjd|jjrC|d|jjd|rP|d d |d 7}||7}|rZ|d 7}|d 7}|d7}t |d||S)Nzzstandardzpitch=''zrate='zvolume='z z zzz SSML: ) r_rrSrTappendrUrVjoinrtrace)rrssmlr prosody_attrsrIrIrJ_construct_ssmls( z"AWSPollyTTSService._construct_ssml context_idc Cst|d|dz||}|dd|jj|jjd|jjd}dd|D}|jj di|j 4Id Hu}|j di|Id H}d |vrU|d }| Id H} n t |d d } |j| d |jId H} ||Id H|j} tdt| | D]#} | | | | } t| dkr|Id Ht| |jd|d} | Vq~Wd Id HWd S1Id HswYWd Sttfy}zdt|}t|dVWYd }~d Sd }~ww)aGenerate speech from text using AWS Polly. 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. z: Generating TTS []rpcm16000)TextTextType OutputFormatVoiceIdEngine SampleRate LexiconNamescSsi|] \}}|dur||qSrNrI).0kvrIrIrJ Xsz.AWSPollyTTSService.run_tts..pollyN AudioStreamz No audio stream in responsei>r)rzAWS Polly TTS error: )error)rrI)rdebugrr_rkrSrWitemsr~clientr{synthesize_speechreadrrresamplerfstart_tts_usage_metrics chunk_sizerangelenstop_ttfb_metricsr rrr\r )rrrrrgfiltered_paramsrresponsestream audio_data CHUNK_SIZEichunkframer error_messagerIrIrJrun_tts;sJ    2zAWSPollyTTSService.run_tts)rXrYrZr[rLrtr]rr`rr\intrxboolrrrrrrr r __classcell__rIrIrrJr^sF   W !(r^cs$eZdZdZeZfddZZS)PollyTTSServicezDeprecated alias for AWSPollyTTSService. .. deprecated:: 0.0.67 `PollyTTSService` is deprecated, use `AWSPollyTTSService` instead. c s^tjdi|ddl}||d|dtWddS1s(wYdS)zInitialize the deprecated PollyTTSService. Args: **kwargs: All arguments passed to AWSPollyTTSService. rNalwayszB'PollyTTSService' is deprecated, use 'AWSPollyTTSService' instead.rI)rwrxwarningscatch_warnings simplefilterwarnDeprecationWarning)rrrrrIrJrxs  "zPollyTTSService.__init__)rXrYrZr[rLrtrxrrIrIrrJrvsr)+r[ry dataclassesrrtypingrrrlogururpydanticrpipecat.audio.utilsr pipecat.frames.framesr r r pipecat.services.settingsr rrpipecat.services.tts_servicerpipecat.transcriptions.languagerr(pipecat.utils.tracing.service_decoratorsrr|botocore.exceptionsrrModuleNotFoundErrorer Exceptionr\rKrLr^rrIrIrIrJs6      Th