i] ~dZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZddlmZmZddlZddlZddlmZd d lmZm Z m!Z!m"Z"m#Z#m$Z$d d l%m&Z&d d l'm(Z(d d l)m*Z*m+Z+m,Z,m-Z-d dlm.Z.m/Z/ej0ej1Z2de3de4deee3e3fe3ffdZ5dee6e3fde6fdZ7de6fdZ8de3de4de4fdZ9de3de4fdZ:de3de4de4fdZ;dee6e3fde4dee3ddffdZd$e6d%e6d&e6de6fd'Z?Gd(d)Z@dS)*zCommunicate with the service. Only the Communicate class should be used by end-users. The other classes and functions are for internal use only.N) nullcontext) TextIOWrapper)Queue)AsyncGeneratorContextManagerDict GeneratorListOptionalTupleUnion)escapeunescape)Literal) DEFAULT_VOICEMP3_BITRATE_BPSSEC_MS_GEC_VERSIONTICKS_PER_SECOND WSS_HEADERSWSS_URL) TTSConfig)DRM)NoAudioReceivedUnexpectedResponseUnknownResponseWebSocketError)CommunicateStateTTSChunk)cafiledata header_lengthreturnct|tstdi}|d|dD] }|dd\}}|||<!|||dzdfS)z Returns the headers and data from the given data. Args: data (bytes): The data to be parsed. header_length (int): The length of the header. Returns: tuple: The headers and data to be used in the request. zdata must be bytesNs :r) isinstancebytes TypeErrorsplit)r!r"headerslinekeyvalues [/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/edge_tts/communicate.pyget_headers_and_datar02s dE " ".,---G^m^$**733ZZa(( U D*,,- --stringct|tr|d}t|tst dt |}t |D]G\}}t|}d|cxkrdks!nd|cxkrdksnd|cxkrdkrnBd ||<Hd |S) aS The service does not support a couple character ranges. Most important being the vertical tab character which is commonly present in OCR-ed PDFs. Not doing this will result in an error from the service. Args: string (str or bytes): The string to be cleaned. Returns: str: The cleaned string. utf-8zstring must be str or bytesr  ) r'r(decodestrr)list enumerateordjoin)r2charsidxcharcodes r/remove_incompatible_charactersrFJs&%  (w'' fc " "75666F||Eu%% TII NNNNNNNNd 0 0 0 0b 0 0 0 0bD6F6F6F6FB6F6F6F6F6FE#J 775>>r1c2tjjS)zZ Returns a UUID without dashes. Returns: str: A UUID without dashes. )uuiduuid4hexr1r/ connect_idrLfs :<< r1textlimitcn|dd|}|dkr|dd|}|S)a Finds the index of the rightmost preferred split character (newline or space) within the initial `limit` bytes of the text. This helps find a natural word or sentence boundary for splitting, prioritizing newlines over spaces. Args: text (bytes): The byte string to search within. limit (int): The maximum index (exclusive) to search up to. Returns: int: The index of the last found newline or space within the limit, or -1 if neither is found in that range.  r )rfind)rMrNsplit_ats r/(_find_last_newline_or_space_within_limitrTps<"zz%E**H!||::dAu-- Or1 text_segmentct|}|dkr; |d|d|S#t$r|dz}YnwxYw|dk;|S)a Finds the rightmost possible byte index such that the segment `text_segment[:index]` is a valid UTF-8 sequence. This prevents splitting in the middle of a multi-byte UTF-8 character. Args: text_segment (bytes): The byte segment being considered for splitting. Returns: int: The index of the safe split point. Returns 0 if no valid split point is found (e.g., if the first byte is part of a multi-byte sequence longer than the limit allows). rNr4r)lenr<UnicodeDecodeError)rUrSs r/_find_safe_utf8_split_pointrYs{<  H Q,,  ( # * *7 3 3 3O!    MHHH  Q,, Os6AArSc|dkrSd|d|vrG|dd|}|d||dkrn|}|dkr d|d|vG|S)a Adjusts a proposed split point backward to prevent splitting inside an XML entity. For example, if `text` is `b"this & that"` and `split_at` falls between `&` and `;`, this function moves `split_at` to the index before `&`. Args: text (bytes): The text segment being considered. split_at (int): The proposed split point index, determined by whitespace or UTF-8 safety. Returns: int: The adjusted split point index. It will be moved to the '&' if an unterminated entity is detected right before the original `split_at`. Otherwise, the original `split_at` is returned. r&N;)rindexfind)rMrSampersand_indexs r/"_adjust_split_point_for_xml_entityras}" Q,,44  ?22++dAx88 99T?H 5 5 ; ; # Q,,44  ?22 Or1 byte_lengthc#PKt|tr|d}t|tst d|dkrt dt ||krt||}|dkrt|}t||}|dkrt d|d| }|r|V||dkr|ndd}t ||k| }|r|VdSdS)a Splits text into chunks, each not exceeding a maximum byte length. This function prioritizes splitting at natural boundaries (newlines, spaces) while ensuring that: 1. No chunk exceeds `byte_length` bytes. 2. Chunks do not end with an incomplete UTF-8 multi-byte character. 3. Chunks do not split XML entities (like `&`) in the middle. Args: text (str or bytes): The input text. If str, it's encoded to UTF-8. byte_length (int): The maximum allowed byte length for any yielded chunk. Must be positive. Yields: bytes: Text chunks (UTF-8 encoded, stripped of leading/trailing whitespace) that conform to the byte length and integrity constraints. Raises: TypeError: If `text` is not str or bytes. ValueError: If `byte_length` is not positive, or if a split point cannot be determined (e.g., due to extremely small byte_length relative to character/entity sizes). r4ztext must be str or bytesrz"byte_length must be greater than 0zTMaximum byte length is too small or invalid text structure near '&' or invalid UTF-8Nr) r'r=encoder(r) ValueErrorrWrTrYrastrip)rMrbrSchunkremaining_chunks r/split_text_by_byte_lengthris`6$${{7## dE " "53444a=>>> d))k ! !;D+NN a<<2488H6dHEE a<<C  YhY%%''  KKK1 HH!5565 d))k ! !:jjllOr1tc escaped_textc t|tr|d}d|jd|jd|jd|jd|d S)z Creates a SSML string from the given parameters. Args: tc (TTSConfig): The TTS configuration. escaped_text (str or bytes): The escaped text. If bytes, it must be UTF-8 encoded. Returns: str: The SSML string. r4z_z)r'r(r<voicepitchratevolume)rjrks r/mkssmlrqs,&&4#**733    8  -/W  @B       r1cNtjdtjS)zg Return Javascript-style date string. Returns: str: Javascript-style date string. z:%a %b %d %Y %H:%M:%S GMT+0000 (Coordinated Universal Time))timestrftimegmtimerKr1r/date_to_stringrv s# =Ddkmm  r1 request_id timestampssmlcd|d|d|S)z Returns the headers and data to be used in the request. Returns: str: The headers and data to be used in the request. z X-RequestId:z1 Content-Type:application/ssml+xml X-Timestamp:zZ Path:ssml rK)rwrxrys r/ssml_headers_plus_datar{0s4 z         r1ceZdZdZefdddddddddd ed ed ed ed ededdeej deedee dee fdZ de de fdZd dZdee dffdZdee dffdZ d!deee fdeeee fddfdZdee ddffdZ d!deee fdeeee fddfdZdS)" Communicatez' Communicate with the service. z+0%z+0HzSentenceBoundaryN <)rorprnboundary connectorproxyconnect_timeoutreceive_timeoutrMrmrorprnr WordBoundaryr~rrrrct||||||_t|tst dt t t|d|_|$t|tst d||_ t| tst dt| tst dtj dd| | |_ |)t|tjst d||_dd d d d d d |_dS) Nztext must be strizproxy must be strzconnect_timeout must be intzreceive_timeout must be int)totalconnect sock_connect sock_readz'connector must be aiohttp.BaseConnectorr1rF) partial_textoffset_compensationlast_duration_offsetstream_was_calledchunk_audio_bytescumulative_audio_bytes)r tts_configr'r=r)rirrFtextsrintaiohttp ClientTimeoutsession_timeout BaseConnectorrstate) selfrMrmrorprnrrrrrs r/__init__zCommunicate.__init__GsU$E4II$$$ 0.// // 1$77 8 8     Zs%;%; /00 0$) /3// ;9:: :/3// ;9:: :&4(%      Iw?T)U)U EFF F:C #$$%!&!"&' ( (  r1r!r#cBtj|dD]v}|d}|dvrR|dd|jdz}|dd}|||t|ddd d cS|d vret d |t d )NMetadataTyperDataOffsetrDurationrMText)typeoffsetdurationrM) SessionEndzUnknown metadata type: zNo WordBoundary metadata found)jsonloadsrrrr)rr!meta_obj meta_typecurrent_offsetcurrent_durations r/__parse_metadatazCommunicate.__parse_metadatas 4((4 I IH (I@@@V$X..send_command_requests O4FM(5gB,9'B$$~///1 LN            r1c Ktttt jjdd{VdS)z&Sends the SSML request to the service.rN)rr{rLrvrqrr)rrsr/send_ssml_requestz/Communicate.__stream..send_ssml_requests$$&LL"$$ >2           r1FT)r trust_envtimeoutz&ConnectionId=z &Sec-MS-GEC=z&Sec-MS-GEC-Version=)compressrr+sslr4s sPathsaudio.metadatarrrsturn.end)sresponses turn.startzUnknown path receivedr&zBWe received a binary message, but it is missing the header length.bigz9The header length is greater than the length of the data.saudioz3Received binary message, but the path is not audio.s Content-Type)s audio/mpegNz=Received binary message, but with an unexpected Content-Type.rz*>w*G*GL';$l&7&7 &D&D(($J&>>'488D000*.*?*?*E*E-----,H5 8SS #9::,,00222%AAA-.EFFFB]g&7&>>>8=))A--0` %(NN8=!3De$L$LM$s8='9'9990W (< }(($J "~~g..(::0Q$.>>/4#H#HL#+@@@0[ $+t99>>$1Z 4yyA~~0X *.&J2333s4yy@333#*D9999999]g&7&===()1K O>]#,f& %\Kg g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g g sCA4N)2#NM JN3N) N N)N N)) N36N3cK|jdrtdd|jd<|jD]|jd<d|jd< |23d{V}|WV 68#tj$rV}|jdkrtj|d|jd<|23d{V}|WV 6Yd}~d}~wwxYwdS) au Streams audio and metadata from the service. Raises: NoAudioReceived: If no audio is received from the service. UnexpectedResponse: If the response from the service is unexpected. UnknownResponse: If the response from the service is unknown. WebSocketError: If there is an error with the websocket. rzstream can only be called once.TrrrNi) r RuntimeErrorr_Communicate__streamrClientResponseErrorstatusrhandle_client_response_error)rmessagees r/streamzCommunicate.stream6sI :) * B@AA A*. &'+/* " "DJ~ &./DJ* + "%)]]__"""""""'!MMMMM&5_. " " "8s??033323 ./%)]]__"""""""'!MMMMM&5_____  " " "s0A(A&A((C 7>C5C;CC  audio_fnamemetadata_fnamecK|t|ddn t}|5t|d5}|23d{V}|ddkr||d/t |t r4|dd vr*t j|||d y6 dddn #1swxYwYddddS#1swxYwYdS) zE Save the audio and metadata to the specified files. Nwr4)encodingwbrrr!r )openrrwriter'rrdump)rrrmetadatarrs r/savezCommunicate.saveWs) w 7 7 7 7   ) )tK.. )%!% ) ) ) ) ) ) )g6?g--KK0000-88)WV_Q>>Igx000NN4((("/ ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )sAC,C CA4C C,C C,C C,,C03C0c#Kdtddffd }t}tj5}||| |}|n|V ddddS#1swxYwYdS)z-Synchronous interface for async stream methodqueuer#Ncdfd }tj}tj||||dS)Nr#cK23d{V}|6ddSN)rput)itemrrs r/ get_itemszECommunicate.stream_sync..fetch_async_items..get_itemsss`"&++--$$$$$$$$IIdOOOO#0 $s5r)asyncionew_event_loopset_event_looprun_until_completeclose)rrlooprs` r/fetch_async_itemsz2Communicate.stream_sync..fetch_async_itemsrss       )++D  "4 ( ( (  # #IIKK 0 0 0 JJLLLLLr1)r concurrentfuturesThreadPoolExecutorsubmitr)rrrexecutorrs` r/ stream_synczCommunicate.stream_syncos U t      ww   2 2 4 4  OO-u 5 5 5 yy{{<                     s4BBBctj5}|tj|||}|ddddS#1swxYwYdS)z,Synchronous interface for async save method.N)rrrrrrunrresult)rrrrfutures r/ save_synczCommunicate.save_syncs   2 2 4 4 __ TYY{NCCF MMOOO                   sA A55A9<A9rr)__name__ __module__ __qualname____doc__rr=rr rrrrr(rrrrrrr rr rr rKr1r/r}r}As2#9 @R59#)+)+9 9 9 9 9  9  9 9 <=9 G129 }9 "#9 "#9 9 9 9 vCUCxCCCC&,,,,(Kx~ >KKKKZ" $ '""""H7;))3:&)!sEz!23)  ))))0Yxt';<87;  3:& !sEz!23        r1r})Ar rconcurrent.futuresrrrrsrH contextlibriorrrtypingrrrr r r r r xml.sax.saxutilsrrrcertifityping_extensionsr constantsrrrrrr data_classesrdrmr exceptionsrrrrrrcreate_default_contextwhererr(rr0r=rFrLrTrYrarirqrvr{r}rKr1r/rsII """"""                    .-------%%%%%%$##### /....... %3 %]W]__ = = =. . #. 4u u $%....05e+<8C50e8Ucc