qiCdZddlZddlmZddlmZddlmZmZddl Z ddl m Z ddl mZddlmZdd lmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZ ddlZddl m!Z!eejLjOdZ(dZ)dZ*de+defdZ,dedefdZ-dede+fdZ.eGddeZ/Gd d!eZ0y#e"$r7Z#e jHde#e jHde%de#dZ#[#wwxYw)"z4Kokoro TTS service implementation using kokoro-onnx.N) dataclass)Path)AsyncGeneratorOptional)logger) BaseModel)create_stream_resampler) ErrorFrameFrameTTSAudioRawFrame) TTSSettings_warn_deprecated_param) TTSService)Languageresolve_language) traced_tts)Kokoroz Exception: zEIn order to use Kokoro, you need to `pip install pipecat-ai[kokoro]`.zMissing module: z~/.cache/kokoro-onnxz_https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/kokoro-v1.0.onnxz^https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/voices-v1.0.binurldestctjd|d|d|jjddt j |dd}|j t|d5}|jd D]}|j| d d d tjd |y #1swY"xYw) z1Download a file from a URL to a destination path.z Downloading z to z...T)parentsexist_oki,)streamtimeoutwbi ) chunk_sizeNz Downloaded ) rdebugparentmkdirrequestsgetraise_for_statusopen iter_contentwrite)rrrespfchunks M/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/kokoro/tts.py_download_filer*,s LL<uDc23KKdT2 <<D# 6D dD Q&&$&7 E GGEN  LL;tf%&s .)B99C model_path voices_pathc|jstt||jstt|yy)z)Download model files if they don't exist.N)existsr*KOKORO_MODEL_URLKOKORO_VOICES_URL)r+r,s r)_ensure_model_filesr18s5    '4    (+6 languagereturncbtjdtjdtjdtjdtj dtj dtjdtjdtjdtjd i }t||d S) zConvert a Language enum to kokoro-onnx language code. Args: language: The Language enum value to convert. Returns: The corresponding kokoro-onnx locale code. zen-uszen-gbesfrhiitjaptzhT) use_base_code) rENEN_USEN_GBESFRHIITJAPTZHr)r3 LANGUAGE_MAPs r)language_to_kokoro_languagerI@sv  W T T T T T T T L Hl$ GGr2ceZdZdZy)KokoroTTSSettingszSettings for KokoroTTSService.N)__name__ __module__ __qualname____doc__r2r)rKrKZs(r2rKc eZdZUdZeZeed<GddeZddddddde e de e d e e d e ed e ef fd Z d e fdZ ded e fdZede de d eedffdZxZS)KokoroTTSServicezKokoro TTS service implementation. Provides local text-to-speech synthesis using kokoro-onnx. Automatically downloads model files on first use. _settingsc4eZdZUdZej Zeed<y)KokoroTTSService.InputParamszInput parameters for Kokoro TTS configuration. .. deprecated:: 0.0.105 Use ``KokoroTTSSettings`` directly via the ``settings`` parameter instead. Parameters: language: Language to use for synthesis. r3N)rLrMrNrOrr>r3__annotations__rPr2r) InputParamsrUks &[[((r2rWN)voice_idr+r,paramssettingsrXr+r,rYrZc tddttj}|t dtd||_|,t dt|st|j |_||j|t |$d dd|d||r t|ntdz }|r t|ntd z } t|| tt|t| |_t!|_y) a1Initialize the Kokoro TTS service. Args: voice_id: Voice identifier to use for synthesis. .. deprecated:: 0.0.105 Use ``settings=KokoroTTSSettings(voice=...)`` instead. model_path: Path to the kokoro ONNX model file. Defaults to auto-downloaded file. voices_path: Path to the voices binary file. Defaults to auto-downloaded file. params: Configuration parameters for synthesis. .. deprecated:: 0.0.105 Use ``settings=KokoroTTSSettings(...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. **kwargs: Additional arguments passed to parent `TTSService`. N)modelvoicer3rXr]rYT)push_start_framepush_stop_framesrZzkokoro-v1.0.onnxzvoices-v1.0.binrP)rKrIrr>rr]r3 apply_updatesuper__init__rKOKORO_CACHE_DIRr1rstr_kokoror _resampler) selfrXr+r,rYrZkwargsdefault_settings model_filevoices __class__s r)rbzKokoroTTSService.__init__ws>-0=    ":/@' J%-  "   "8-> ?,G,X )    ) )( 3  !!%   *4T*%9IL^9^ &1k"7GJ[7[J/c*os6{; 13r2r4cy)z;Indicate that this service supports TTFB and usage metrics.TrP)rgs r)can_generate_metricsz%KokoroTTSService.can_generate_metricssr2r3ct|S)zConvert a Language enum to kokoro-onnx language format. Args: language: The language to convert. Returns: The kokoro-onnx language code. )rI)rgr3s r)language_to_service_languagez-KokoroTTSService.language_to_service_languages+844r2text context_idcKtj|d|d |j|d{|jj ||j j |j jd}|23d{\}}|jd{|dzjtjj}|jj|||jd{}t!||jd|7777'6n(#t"$r}t%d | Yd}~nd}~wwxYw|jd{7y#|jd{7wxYww) zSynthesize speech from text using kokoro-onnx. Uses the async streaming API to generate audio frames. Args: text: The text to synthesize. context_id: Unique identifier for this TTS context. z: Generating TTS []Ng?)r]langspeedi)audio sample_rate num_channelsrrzUnknown error occurred: )error)rrstart_tts_usage_metricsre create_streamrSr]r3stop_ttfb_metricsastypenpint16tobytesrfresampleryr Exceptionr ) rgrqrrrsamplesry audio_int16 audio_dataes r)run_ttszKokoroTTSService.run_ttssm  v/vQ78 +..t4 4 4\\//DNN00t~~7N7NVY0F/5  *g{,,...&66rxx@HHJ #'??#;#;d.>.>$ '$ $ 0 0!")  5 . /5 C%=aS#AB B B C((* * *$((* * *sF D+D!A D+D)D#D) D+ D%!AD+?D'"D+#D)%D+'D+)D+*E-+ E4E E- EE-F &E)'F -FFFF )rLrMrNrOrKSettingsrVrrWrrdrbboolrnrrprrr r __classcell__)rls@r)rRrRas !H   )i )#'$(%)(,04B43-B4SM B4 c] B4 % B4,-B4Hd 5X 5# 5$+#$+3$+>%QU+;V$+$+r2rR)1rOos dataclassesrpathlibrtypingrrnumpyrlogururpydanticrpipecat.audio.utilsr pipecat.frames.framesr r r pipecat.services.settingsr rpipecat.services.tts_servicerpipecat.transcriptions.languagerr(pipecat.utils.tracing.service_decoratorsrr kokoro_onnxrModuleNotFoundErrorrr{rpath expanduserrcr/r0rdr*r1rIrKrRrPr2r)rs ; !+7 J3F?," **+ABCtd ' '4 '7D7t7H(HsH4     N+zN+C,FLL;qc"#FLLXY &qc* ++,s B;;C72C22C7