o i' @s8dZddlZddlmZmZddlmZmZddlm Z ddl Z ddl m Z ddl mZddlmZmZmZdd lmZmZmZdd lmZdd lmZdd lmZzdd lmZWn eyzZ ze !de e !de"de dZ [ wweGdddeZ#GdddeZ$de%de%de&de%fddZ'dS)z%Speechmatics TTS service integration.N) dataclassfield)AsyncGeneratorOptional) urlencode)logger) BaseModel) ErrorFrameFrameTTSAudioRawFrame) NOT_GIVEN TTSSettings _NotGiven) TTSService)exponential_backoff_time) traced_tts) __version__z Exception: zQIn order to use Speechmatics, you need to `pip install pipecat-ai[speechmatics]`.zMissing module: c@s,eZdZUdZedddZeeBed<dS)SpeechmaticsTTSSettingsz|Settings for SpeechmaticsTTSService. Parameters: max_retries: Maximum number of retries for HTTP requests. cCstS)N)r rrU/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/speechmatics/tts.py.sz SpeechmaticsTTSSettings.)default_factory max_retriesN) __name__ __module__ __qualname____doc__rrintr__annotations__rrrrr&s rcseZdZUdZeZeed<dZGdddeZ ddedddd e d e d e e d e j d e ede e de effddZdefddZede de deedffddZZS)SpeechmaticsTTSServicezSpeechmatics TTS service implementation. This service provides text-to-speech synthesis using the Speechmatics HTTP API. It converts text to speech and returns raw PCM audio data for real-time playback. _settingsi>c@seZdZUdZdZeed<dS)z"SpeechmaticsTTSService.InputParamsaOptional input parameters for Speechmatics TTS configuration. .. deprecated:: 0.0.105 Use ``settings=SpeechmaticsTTSService.Settings(...)`` instead. Parameters: max_retries: Maximum number of retries for TTS requests. Defaults to 5. rN)rrrrrrrrrrr InputParams=s  r"z$https://preview.tts.speechmatics.comN)base_urlvoice_id sample_rateparamssettingsapi_keyr#r$aiohttp_sessionr%r&r'c  s|r||jkrtd|jd|d|jddddd} |dur*|dd || _|dur9|d |s9|j| _|durB| |tj d|d d | d |||_ ||_ ||_ |j s`t d dS)auInitialize the Speechmatics TTS service. Args: api_key: Speechmatics API key for authentication. base_url: Base URL for Speechmatics TTS API. voice_id: Voice model to use for synthesis. .. deprecated:: 0.0.105 Use ``settings=SpeechmaticsTTSService.Settings(voice=...)`` instead. aiohttp_session: Shared aiohttp session for HTTP requests. sample_rate: Audio sample rate in Hz. params: Input parameters for the service. .. deprecated:: 0.0.105 Use ``settings=SpeechmaticsTTSService.Settings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to TTSService. zSpeechmatics TTS only supports z Hz sample rate. Current rate of zHz may cause issues.Nsarahr!)modelvoicelanguagerr$r,r&T)r%push_start_framepush_stop_framesr'zMissing Speechmatics API keyr)SPEECHMATICS_SAMPLE_RATErwarningSettings"_warn_init_param_moved_to_settingsr,r apply_updatesuper__init___api_key _base_url_session ValueError) selfr(r#r$r)r%r&r'kwargsdefault_settings __class__rrr6IsF!     zSpeechmaticsTTSService.__init__returncCsdS)zCheck if this service can generate processing metrics. Returns: True, as Speechmatics service supports metrics generation. Tr)r;rrrcan_generate_metricssz+SpeechmaticsTTSService.can_generate_metricstext context_idc Cst|d|dd|jdd}d|i}t|j|jj|j}z:zd} |jj |||d 4Id H} |j d kr z6t |d d dd}|d7}||jj krUt td|d|dddVt|Id HWWd Id Hq)t tfytd|ddVYWd Id HWW|Id Hd Sw|j dkrtd|j ddV Wd Id HWW|Id Hd S||Id Hd} d} |j2z?3d HW} | sq| r|Id Hd} | | 7} t| dkrt| d} | d} | d | }| | d } t||jd|dVq6 Wd Id Hn 1Id Hs'wYq*WntyJ}ztd|dVWYd }~nd }~wwW|Id Hd SW|Id Hd S|Id Hw)aGenerate speech from text using Speechmatics' HTTP API. 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 []zBearer zapplication/json) Authorizationz Content-TyperBrT)jsonheadersNig?g @g?)attemptmin_waitmax_wait multiplierz#Service unavailable [503] (attempt z , retry in z.2fzs))errorz$Service unavailable [503] (attempts )zService unavailable [F)audior% num_channelsrCzError generating TTS: )rdebugr7_get_endpoint_urlr8r r,r%r9poststatusrrr:r asynciosleepArithmeticErrorstop_ttfb_metricsstart_tts_usage_metricscontentiter_anylenr Exception)r;rBrCrGpayloadurlrHresponse backoff_time first_chunkbufferchunkcomplete_samplescomplete_bytes audio_dataerrrrun_ttss    Q Q   ,NQ"zSpeechmaticsTTSService.run_tts)rrrrrr2rr0rr"strraiohttp ClientSessionrr6boolrArrr rl __classcell__rrr>rr1s:  O(rr#r,r%r@cCs<i}d||d<dt|d<t|}|d|d|S)aFormat the TTS endpoint URL with voice, output format, and version params. Args: base_url: The base URL for the TTS endpoint. voice: The voice model to use. sample_rate: The audio sample rate. Returns: str: The formatted TTS endpoint URL. pcm_ output_formatzpipecat/zsm-appz /generate/?)rr)r#r,r% query_paramsqueryrrrrUs  rU)(rrX dataclassesrrtypingrr urllib.parserrnlogururpydanticrpipecat.frames.framesr r r pipecat.services.settingsr r rpipecat.services.tts_servicerpipecat.utils.networkr(pipecat.utils.tracing.service_decoratorsrspeechmatics.rtrModuleNotFoundErrorrkrMr`rrrmrrUrrrrs8       f