߲j4%DUdZddlmZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZmZmZmZej$dZdZd Zd Zeej0d z Zd ZiZd ed<iZd ed<eedZded<ddZddZ ddZ!d dZ"d!dZ#d"dZ$dZ%ddZ&d"dZ'ddZ(y)#aLocal on-device TTS engines for ``tools.tts_tool``: NeuTTS, Piper, KittenTTS. All three synthesize WAV natively; :func:`_finalize_wav_output` converts/renames to the requested container. Piper and KittenTTS keep loaded models in small LRU caches registered in ``_LOCAL_TTS_MODEL_CACHES`` so warm/release can pre-load or drop them. ``_import_piper`` / ``_import_kittentts`` are resolved through the origin module at call time (test monkeypatches). ) annotationsN)Path)AnyCallableDictTuple)_finalize_wav_output_origin_section_wav_sidecar_pathztools.tts_toolz!KittenML/kitten-tts-nano-0.8-int8Jasperzen_US-lessac-mediumneutts_samplesDict[str, Any]_piper_voice_cache_kittentts_model_cache)piper kittenttszDict[str, Dict[str, Any]]_LOCAL_TTS_MODEL_CACHESc||vr|j|||<||S|}|||<t|tkDr7|jtt |dt|tkDr7|S)zGet ``key`` from ``cache`` or load it, LRU-bounded at ``_TTS_MODEL_CACHE_MAX`` (a hit refreshes recency via pop + reinsert; eviction only releases the slot, not live references).N)poplen_TTS_MODEL_CACHE_MAXnextiter)cachekeyloadvalues 9/home/ubuntu/.hermes/hermes-agent/tools/tts_tool_local.py_tts_cache_get_or_loadr!(sn e|YYs^c Sz FEE#J e*+ + $tE{#T* e*+ + Lc Vtj|dddd|tjS)NTzutf-8replace)capture_outputtextencodingerrorstimeoutstdin) subprocessrunDEVNULL)cmdr)s r _run_helperr/5s* >> DtgiY`hrhzhz r"c|jdxsi}t|}tjt t t jdz d|d|d|jddxst tdz d |jd dxst td z d |jd dd|jddg}t|d}|jdk7rq|jjjDcgc]}|jdr|}}tdt!dj#|xsdt%||Scc}w)Nneuttszneutts_synth.pyz--textz--outz --ref-audio ref_audiozjo.wavz --ref-textref_textzjo.txtz--modelmodelzneuphonic/neutts-air-q4-ggufz--devicedevicecpuxrzOK:zNeuTTS synthesis failed: z unknown error)getr sys executablestrr__file__parent_NEUTTS_SAMPLESr/ returncodestderrstrip splitlines startswith RuntimeErrorchrjoinr ) r& output_path tts_config neutts_configwav_pathr.resultl error_liness r _generate_neuttsrP<sBNN8,2M -H DN114EEF$}((b9\SS[A[=\m'' B7Z3QY?Y;Z=$$W.LMM%%h6 8Cc "F A"(--"5"5"7"B"B"D`QALLY^L_q` `6s2w||K7P7cTc6deff + 66as E"&E"c\ddlm}t|dd}|jdd|S)zS``/cache/piper-voices/`` so voice downloads follow profile boundaries.r)get_hermes_dirzcache/piper-voicespiper_voices_cacheTparentsexist_ok)hermes_constantsrRrmkdir)rRroots r _get_piper_voices_dirrZOs-/ 35IJ KDJJtdJ+ Kr"c |xst}t|j}|jj dk(r|j r t |S||dz }|j r!||dz j r t |Stjd|| ttjdd|dt |gd}|jd k7r6|j xsd j#xsd }td|d|d d|j std|dt |S#tj$r}td|d |d }~wwxYw)zResolve *voice* (an .onnx path or a name like ``en_US-lessac-medium``, downloaded into *download_dir* on first use) to a concrete .onnx file; RuntimeError when it can't be.z.onnxz .onnx.jsonz0[Piper] Downloading voice '%s' to %s (first use)z-mzpiper.download_voicesz--download-diri,z/Piper voice download timed out after 300s for ''Nrr3zno stderr outputz!Piper voice download failed for 'z': iz#Piper voice download completed but uh is missing — check voice name (see: https://github.com/OHF-Voice/piper1-gpl/blob/main/docs/VOICES.md))DEFAULT_PIPER_VOICEr expandusersuffixlowerexistsr=loggerinfor/r;r<r+TimeoutExpiredrFrArBrC)voice download_dir candidatecachedrMexcrBs r _resolve_piper_voice_pathrjWs|  ((EU &&(I7*y/?/?/A9~ ugUO +F }}LeWJ+??GGI6{ KKBE<X` ^^T#:ECSUXYeUf gil  A--%2,,.D2D>ugSPTQTWXX ==?1&:( )* * v;  $ $`LUGSTUV\__`s-)EE-E((E-ctjt|d}|jdxst}t |jdxs t j}|jddt|jddt||fd}d }tt|||fS) zResolve + load (or fetch from cache) the selected Piper voice -> ``(voice, piper_config)``. Shared by synthesis and ``warm_tts_provider`` so a warm-up fills exactly the slot synthesis hits.rre voices_dirTrTuse_cudaFctjdj}tjd|S)Nz[Piper] Loading voice: %s)rmz[Piper] Voice loaded)rbrcr)v PiperVoice model_pathrms r _load_piper_voicez7_load_piper_voice_for_config.._load_piper_voice~s6 /< OOJO : *+r"z::cuda=) r _import_piperr r:r]rrZr^rXboolrjr!r) rJ piper_config voice_namerfrr cache_keyrprqrms @@@r _load_piper_voice_for_configrxss((*JJ0L!!'*A.AJ ((6Q:O:QR]]_Ltd3L$$Z78H*:|DJ,ghZ0I !"4iAR SUa aar" length_scale noise_scale noise_w_scalevolumenormalize_audio speaker_idc  ddl}t|\} jdd}t|tur|nd}d}t fdt Dr ddlm}|t jddt jddt jd d t jd dt jd d |}t|} |j| d5} ||j!|| |n|j!|| dddt#| |S#t$rtjdY{wxYw#1swY9xYw)Nrrc3&K|]}|v yw)N).0krus r z&_generate_piper_tts..s <1  ._load_kittentts_models. 3Z@ j ! ;<r")r _import_kittenttsr r:DEFAULT_KITTENTTS_MODELr!r)rJ kt_configrrrs @@r _load_kittentts_model_for_configrsN ++-I[1Iw(?@J ""8*F[ \^g ggr"c  t|\}}|j||jdt|jdd|jdd}ddl}t |}|j ||dt||S) Nrespeedr clean_textT)rerrri])rgenerater:DEFAULT_KITTENTTS_VOICE soundfiler writer )r&rIrJr5raudiosfrLs r _generate_kittenttsrs~7 CE9 NN IMM'+BCmmGS)immLRV6W  YE -HHHXue$ + 66r")rrrr=rzCallable[[], Any]returnr)r.listr)rrzsubprocess.CompletedProcess)r&r=rIr=rJrrr=)rr)rer=rfrrr=)rJrrzTuple[Any, Dict[str, Any]]))__doc__ __future__rloggingr+r;pathlibrtypingrrrrtools.tts_tool_deliveryr r r r getLoggerrbrrr]r>r?r@rr__annotations__rrr!r/rPrZrjrxrrrrrr"r rs# --^^   + ,="+x.''*::  &(N')++ .D6F2F 7&8b,t7> h7r"