%ui|dZddlmZddlZddlZddlmZmZddlm Z eje Z eGddZ Gdd Zy) aTTS client for CosyVoice3 Singlish TTS server. Sends text to the TTS HTTP endpoint and returns the streaming WAV response as raw bytes. The orchestrator can then forward the audio to the dashboard for operator-side playback and/or to the robot's speaker on the Ghost PC. ) annotationsN) dataclassfield)Optionalc>eZdZUdZedZded<dZded<y ) TTSConfigzConfiguration for the TTS client. Attributes: endpoint: Full URL for the TTS REST endpoint. timeout_s: Request timeout in seconds (covers full streaming response). c.tjddS)N TTS_ENDPOINTz!http://kluster.klass.dev:8200/tts)osgetenvG/home/nelsen/Projects/kognitive/orchestrator/src/services/tts_client.pyzTTSConfig.s ?! r)default_factorystrendpointint timeout_sN)__name__ __module__ __qualname____doc__rr__annotations__rr rrrrs+ Hc Isrrc,eZdZdZdddZdd dZd dZy) TTSClientaAsync HTTP client for the CosyVoice3 TTS server. Usage:: client = TTSClient() wav_bytes = await client.speak("Hello, are you okay?") if wav_bytes: # wav_bytes contains the raw WAV audio data ... await client.close() The TTS server (POST /tts) accepts:: {"text": "...", "sample_rate": 16000} and responds with a streaming ``audio/wav`` response. The client collects the full stream and returns it as ``bytes`` (or ``None`` on failure). Nc8|xs t|_d|_yN)rconfig_session)selfr s r__init__zTTSClient.__init__8s+  *. rcK|stjdy ddl}|j |j |_ |j|jj}|j j|jj||d|4d{}|jdk(r?t}|jj23d{}|j!||j)d{}tj d |j|dd dddd{y#t$rtj dYywxYw776tj#d |dd t%|t'|cdddd{7S77p#1d{7swYyxYw#t*$r } tj d | Yd} ~ yd} ~ wwxYww) a Send text to TTS server and return the WAV audio bytes. Args: text: The message to synthesise. sample_rate: Audio sample rate in Hz (default 16000). Returns: Raw WAV bytes on success, None on any error. u3TTSClient.speak called with empty text — skippingNruYaiohttp is not installed — cannot call TTS server. Install it with: pip install aiohttp)total)text sample_rate)jsontimeoutzTTS spoke: %r (%d bytes)PzTTS server returned %d: %szTTS request failed: %s)loggerwarningaiohttp ImportErrorerrorr! ClientSession ClientTimeoutr rpostrstatus bytearraycontentiter_anyextendinfolenbytesr& Exception) r"r&r'r.r)respchunkschunk error_bodyexcs rspeakzTTSClient.speak<s NNP Q   == #113DM ++$++2G2G+HG}})) $$";?*  ;;#%&[F'+||'<'<'>--e e,$(99;.  0$++z$3?O#     LL7    -'>KK :D"Is6{S =   /    &  LL13 7 sG0E!G0AG!E&"G%4F/E*E(E*!&F/F+)F/1 G<F-=GG0E# G0"E##G0&G(E**.F/ G$F'%G*G0+F/-G/G5F8 6G=GG0G G- G(#G0(G--G0c|K|j*|jjd{d|_yy7 w)z%Close the underlying aiohttp session.N)r!close)r"s rrDzTTSClient.closeos5 == $--%%' ' ' DM % 's *<:<r)r zOptional[TTSConfig]returnNone)i>)r&rr'rrEzOptional[bytes])rErF)rrrrr#rBrDr rrrr#s(/1f!rr)r __future__rloggingr dataclassesrrtypingr getLoggerrr,rrr rrrLsR# (   8 $      P!P!r