Gj] ZdZddlZddlZddlZddlZddlZddlZddlm 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/ej`ejbZ2de3de4deee3e3fe3ffdZ5dee6e3fde6fdZ7de6fdZ8de3de4de4fdZ9de3de4fdZ:de3de4de4fdZ;dee6e3fde4dee3ddffdZd$e6d%e6d&e6de6fd'Z?Gd(d)Z@y)*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|ts tdi}|d|jdD]}|jdd\}}|||<|||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 R/root/workspace/ytshorts/venv/lib/python3.12/site-packages/edge_tts/communicate.pyget_headers_and_datar02sw dE ",--G^m$**73ZZa( U  D*,- --stringcJt|tr|jd}t|ts t dt |}t |D]>\}}t|}d|cxkrdksnd|cxkrdksnd|cxkrdks7n:d ||<@d j|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 "566F|Eu% TI NNd 0b 0bD6FB6FE#J 775>r1c>tjjS)zZ Returns a UUID without dashes. Returns: str: A UUID without dashes. )uuiduuid4hexr1r/ connect_idrLfs ::<  r1textlimitc\|jdd|}|dkr|jdd|}|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_limitrTps4"zz%E*H!|::dAu- Or1 text_segmentct|}|dkDr |d|jd|S|S#t$r|dz}YnwxYw|dkDr3)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 3O O "  MH  Q,s *;;rSc|dkDrBd|d|vr;|jdd|}|jd||dk7r |S|}|dkDrd|d|vr;|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_entityrasm" Q,44 ?2++dAx8 99T?H 5 ;  O# Q,44 ?2 Or1 byte_lengthc#Kt|tr|jd}t|ts t d|dkr t dt ||kDrlt||}|dkr t|}t||}|dkr t d|d|j}|r|||dkDr|ndd}t ||kDrl|j}|r|yyw)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_lengthris6${{7# dE "344a=>> d)k !;D+N a<248H6dHE a<C  Yh%%' K1 H!565 d)k !:jjlOs CC! C!tc escaped_textc t|tr|jd}d|jd|jd|j d|j d|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/mkssmlrqsd,&#**73  z"88*HRWWIZ {" .   r1cRtjdtjS)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  ==Ddkkm 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{0s) zl# k" &  r1cJeZdZdZefdddddddddd ed ed ed ed ededdeejdeedee dee fdZ de de fdZd dZdee dffdZdee dffdZ d!deee fdeeee fddfdZdee ddffdZ d!deee fdeeee fddfdZy)" Communicatez' Communicate with the service. z+0%z+0HzSentenceBoundaryN <)rorprnboundary connectorproxyconnect_timeoutreceive_timeoutrMrmrorprnr WordBoundaryr~rrrrct||||||_t|ts t dt t t|d|_|t|ts t d||_ t| ts t dt| ts t dtjdd| | |_ |%t|tjs t d||_dd d d d d d |_y) 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__Gs$E4I$$./ // 1$7 8    Zs%;/0 0$) /3/9: :/3/9: :&44(%    Iw?T?T)UEF F:C #$$%!&!"&' (  r1r!r#ctj|dD]X}|d}|dvr;|dd|jdz}|dd}|||t|ddd d cS|d vrLt 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 IH (I@@V$X.k(sLtA|j"r|j"d$|s tCd%dddd{dddd{y7777776?7&#1d{7swY6xYw7-#1d{7swYyxYww)'Nc Kjjdk(}|rdnd}|sdnd}jdtd|d|dd{y7w) z)Sends the command request to the service.rtruefalsez X-Timestamp:z Content-Type:application/json; charset=utf-8 Path:speech.config {"context":{"synthesis":{"audio":{"metadataoptions":{"sentenceBoundaryEnabled":"z","wordBoundaryEnabled":"z9"},"outputFormat":"audio-24khz-48kbitrate-mono-mp3"}}}} N)rrsend_strrv) word_boundarywdsqr websockets r/send_command_requestz2Communicate.__stream..send_command_requestsn OO44FM(gB,'B$$~/01./1T1J2$O    sA AAAc Kjtttt j j dd{y7w)z&Sends the SSML request to the service.rN)rr{rLrvrqrr)rrsr/send_ssml_requestz/Communicate.__stream..send_ssml_requestsM$$&L"$ >2   sAAAAFT)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*GL';$l&7&7 &D($J&>>'48D00*.*?*?*E--,H5 8SS #9:,002%AA-.EFF]]g&7&7&>&>>8==)A-0` %(NN8==!3De$LM$s8=='990W (< }($J "~~g.(:0Q$.>>/4#HL#+@@0[ $+t9>$1Z 4yA~0X *.&JJ23s4y@3#*D99]]g&7&7&=&==()1 N'4N N&&N 'N&+ N;6N$7N;?N&NNN  N N&N! NN! N&$N;&N8,N/ -N84N;cK|jdr tdd|jd<|jD]=|jd<d|jd< |j23d{}|y7 6E#tj $r_}|j dk7rtj|d|jd<|j23d{7}|6Yd}~d}~wwxYww) 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.stream6s ::) *@A A*. &'+/** "DJJ~ &./DJJ* + "%)]]_""'!M ""_.. "88s?00323 ./%)]]_""'!M&5  "ssAC2A=(A;,A9-A;0A=7C29A;;A=<C2=C/AC*C$C C$C*%C2*C//C2 audio_fnamemetadata_fnamecK|t|ddn t}|5t|d5}|j23d{}|ddk(r|j|d&t |t s7|dd vs?t j|||jd g7b6 dddn #1swYnxYwdddy#1swYyxYww) zE Save the audio and metadata to the specified files. Nwr4)encodingwbrrr!r )openrrwriter'rrdump)rrrmetadatarrs r/savezCommunicate.saveWs) w 7   )tK. )%!% ) )g6?g-KK0-8WV_Q>IIgx0NN4( ) ) ) ) ) ) )sbC CB1B&B$ B&.B14B1<(B1$B&&B1( C1B: 6C= CC Cc#Kdtddffd }t}tjj5}|j || |j }|n| dddy#1swYyxYww)z-Synchronous interface for async stream methodqueuer#Ncdfd }tj}tj||j||j y)NcKj23d{}j|76jdywN)rput)itemrrs r/ get_itemszECommunicate.stream_sync..fetch_async_items..get_itemsss:"&++-$$$IIdO$- $sA1/1A1Ar)asyncionew_event_loopset_event_looprun_until_completeclose)rrlooprs` r/fetch_async_itemsz2Communicate.stream_sync..fetch_async_itemsrs? ))+D  " "4 (  # #IK 0 JJLr1)r concurrentfuturesThreadPoolExecutorsubmitr)rrrexecutorrs` r/ stream_synczCommunicate.stream_syncos U t w    2 2 4  OO-u 5yy{<     s7A<,A0' A<0A95A<ctjj5}|jtj |j ||}|jdddy#1swYyxYw)z,Synchronous interface for async save method.N)rrrrrrunrresult)rrrrfutures r/ save_synczCommunicate.save_syncsZ    2 2 4 __ TYY{NCF MMO    s AA))A2rr)__name__ __module__ __qualname____doc__rr=rr rrrrr(rrrrrrr rr rr rKr1r/r}r}As#9 @R59#)+)+9 9 9  9  9 9 <=9 G1129 }9 "#9 "#9 vCUCxC&,(Kx~ >KZ" $ '"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/rsI "   .%$ / %3 % %]W]]_ =. . #. 4u u $%.05e+<8C50e8Ucc