qi[5dZddlZddlmZmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZmZmZdd lmZmZmZmZdd lmZdd lmZmZdd lmZ ddlZdd l m!Z!m"Z"dedee'fdZ(eGddeZ)GddeZ*Gdde*Z+y#e#$r7Z$e jJde$e jJde&de$dZ$[$wwxYw)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_warn_deprecated_param) TTSService)Languageresolve_language) traced_tts) BotoCoreError ClientErrorz Exception: zHIn order to use AWS services, you need to `pip install pipecat-ai[aws]`.zMissing module: languagereturncitjdtjdtjdtjdtj dtj dtjdtjdtjdtjd tjd tjd tjd tjd tjdtj d tj"ditj$dtj&dtj(dtj*dtj,dtj.dtj0dtj2dtj4dtj6dtj8dtj:dtj<dtj>dtj@dtjBdtjDdtjFdtjHdtjJd tjLd!tjNd"tjPd#tjRd$tjTd%tjVd&tjXd'tjZd'i }t]||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_MAPs J/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/aws/tts.pylanguage_to_aws_languagerM(sF UF F  W F  XF  hF F  WF  WF  WF !F$  W%F& 'F( )F* +F, -F. /F0 1F4  W5F8  W9F: ;F< =F@  WAFB CFD EFH  WIFL  WMFP  WQFT  WUFX  WYF\  W]F^  W_F` aFd  WeFh  WiFj  W W W W W WKFLP Hl% HHceZdZUdZedZeezed<edZ eezed<edZ eezed<ed Z eezed <ed Z e eezed <y )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. ctSNrrNrLzAWSPollyTTSSettings.IrN)default_factoryenginectSrRrSrTrNrLrUzAWSPollyTTSSettings.s9rNpitchctSrRrSrTrNrLrUzAWSPollyTTSSettings.s)rNratectSrRrSrTrNrLrUzAWSPollyTTSSettings.rVrNvolumectSrRrSrTrNrLrUzAWSPollyTTSSettings.srN lexicon_namesN)__name__ __module__ __qualname____doc__rrXstrr__annotations__rZr\r^r`rrTrNrLrPrP|sr$4EFFC)OF"3DEE3?E!2CDD# /D#4EFFC)OF+0AR+SM49y(SrNrPceZdZUdZeZeed<GddeZdddddddddde e de e d e e d e e d e e d e e d e ede effdZ de fdZdede e fdZde de fdZede de deedffdZxZS)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. _settingsceZdZUdZdZeeed<ejZ eeed<dZ eeed<dZ eeed<dZ eeed<dZeeeed<y) AWSPollyTTSService.InputParamsa1Input parameters for AWS Polly TTS configuration. .. deprecated:: 0.0.105 Use ``AWSPollyTTSSettings`` 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. NrXrrZr\r^r`)rarbrcrdrXrrerfrr(rrZr\r^r`rrTrNrL InputParamsrksg !% $'/{{(8$2#x}#"hsm" $ $-1 xS *1rNrlN)api_keyaws_access_key_idaws_session_tokenregionvoice_id sample_rateparamssettingsrmrnrorprqrrrsrtc tdddddddd} |tdtd|| _|tdt|s|j| _|jr|j |jnd| _|j | _|j| _|j| _|j| _ || j|t |0d|dd| d | |xstjd |xstjd |xstjd |xstjd dd|_t!j"|_t'|_y)aNInitializes 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=AWSPollyTTSSettings(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=AWSPollyTTSSettings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to parent TTSService class. NJoannar)modelvoicerrXrZr\r^r`rqrxrsT)rrpush_start_framepush_stop_framesrtAWS_ACCESS_KEY_IDAWS_SECRET_ACCESS_KEYAWS_SESSION_TOKEN AWS_REGIONz us-east-1)rnaws_secret_access_keyro region_namerT)rPrrxrXrlanguage_to_service_languagerZr\r^r` apply_updatesuper__init__osgetenv _aws_paramsaioboto3Session _aws_sessionr _resampler) selfrmrnrorprqrrrsrtkwargsdefault_settings __class__s rLrzAWSPollyTTSService.__init__szH/     ":/BG L%-  "   "8-@ A*0-- '55fooF !) *0 &(.  %*0-- '171E1E .    ) )( 3  #!!%    "3!Tbii@S6T%,%R :Q0R!2!Tbii@S6T!IRYY|[%I  %,,.13rNrcy)zCheck if this service can generate processing metrics. Returns: True, as AWS Polly service supports metrics generation. TrT)rs rLcan_generate_metricsz'AWSPollyTTSService.can_generate_metrics s rNrct|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. )rM)rrs rLrz/AWSPollyTTSService.language_to_service_languages(11rNtextcd}|jj}|d|dz }g}|jjdk(r?|jjr)|j d|jjd|jj r)|j d|jj d|jj r)|j d|jj d|r|d d j|d z }||z }|r|d z }|d z }|dz }tj|d||S)Nzzstandardzpitch=''zrate='zvolume='z z zzz SSML: ) rirrXrZappendr\r^joinrtrace)rrssmlr prosody_attrss rL_construct_ssmlz"AWSPollyTTSService._construct_ssmls4>>** "8*B// >> J .~~##$$wt~~/C/C.DA%FG >>    6$..*=*=)>a!@ A >>  8DNN,A,A+B!!D E  i 78: :D    L D   vWTF+, rN context_idcKtj|d|d |j|}|dd|jj|jj d|jj d}|jDcic] \}}| || }}}|jjdi|j4d{}|jdi|d{} d| vr| d} | jd{} ntj|d d} |jj| d |j d{} |j#|d{|j$} t'd t)| | D]O} | | | | z}t)|d kDs|j+d{t-||j d | }|Qdddd{ycc}}w7.77777G7#1d{7swYyxYw#t.t0f$r'}dt3|}t5|Yd}~yd}~wwxYww)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 LexiconNamesN AudioStreamz No audio stream in responsei>r)rzAWS Polly TTS error: )error)pollyrT)rdebugrrirxrXr`itemsrclientrsynthesize_speechreadrrresamplerrstart_tts_usage_metrics chunk_sizerangelenstop_ttfb_metricsr rrrer )rrrrrskvfiltered_paramsrresponsestream audio_data CHUNK_SIZEichunkframer error_messages rLrun_ttszAWSPollyTTSService.run_tts?s9  v/vQ78+ 2''-D" %>>//..//% $ < < F17 P1!-q!tPOP/t((//L4;K;KL $ $PU!8!8!8!K?!KK H,%m4F'-{{}!4JLLD6)E!FG!%J#'??#;#;JtO_O_#`` 224888!__ q#j/:>$A&q1z>:E5zA~"44666 08H8H!Xb c# $! $ $ $Q $K"5 a87' $ $ $ $.{+ 23CJ<@M=1 1 1 2sIA*H G.G.-HG4H H G7! HG:A H G< H$G>%?H%H8H9$H H(H)H-I.H7H:H<H>HHHH H HHIHI(I I II)rarbrcrdrPSettingsrfr rlrreintrboolrrrrrrr r __classcell__rs@rLrhrhs'#H""2i22"&+/+/ $"&%)(,26Y4#Y4$C= Y4 $C= Y4  Y43-Y4c]Y4%Y4./Y4vd 2X 2(3- 2CCB72#72372>%QU+;V7272rNrhc&eZdZdZeZfdZxZS)PollyTTSServicezDeprecated alias for AWSPollyTTSService. .. deprecated:: 0.0.67 `PollyTTSService` is deprecated, use `AWSPollyTTSService` instead. c t|di|ddl}|j5|j d|j dt dddy#1swYyxYw)zInitialize the deprecated PollyTTSService. Args: **kwargs: All arguments passed to AWSPollyTTSService. rNalwayszB'PollyTTSService' is deprecated, use 'AWSPollyTTSService' instead.rT)rrwarningscatch_warnings simplefilterwarnDeprecationWarning)rrrrs rLrzPollyTTSService.__init__sY "6"  $ $ &   ! !( + MMT"    s (AA)rarbrcrdrPrrrrs@rLrrzs#HrNr),rdr dataclassesrrtypingrrrlogururpydanticr pipecat.audio.utilsr pipecat.frames.framesr r r pipecat.services.settingsrrrrpipecat.services.tts_servicerpipecat.transcriptions.languagerr(pipecat.utils.tracing.service_decoratorsrrbotocore.exceptionsrrModuleNotFoundErrorer ExceptionrerMrPrhrrTrNrLrs  (117 `_3F?,>QIxQIHSMQIh T+T T$h2h2V(q ,FLL;qc"#FLL[\ &qc* ++,s BC2C  C