o i] @s(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/ej0e1dZ2de3de4deee3e3fe3ffddZ5dee6e3fde6fddZ7de6fddZ8de3de4de4fddZ9de3de4fdd Z:de3d!e4de4fd"d#Z;dee6e3fd$e4dee3ddffd%d&Zd-e6d.e6d/e6de6fd0d1Z?Gd2d3d3Z@dS)4zCommunicate 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_lengthreturncCsZt|ts tdi}|d|dD]}|dd\}}|||<q|||ddfS)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!headerslinekeyvaluer-N/home/kim/smarthome/.venv/lib/python3.10/site-packages/edge_tts/communicate.pyget_headers_and_data2s  r/stringcCst|tr |d}t|tstdt|}t|D].\}}t|}d|kr-dksEnd|kr8dksEnd|krCdkrInqd ||<qd |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)r0charsidxcharcoder-r-r.remove_incompatible_charactersJs  D rCcCs tjS)zZ Returns a UUID without dashes. Returns: str: A UUID without dashes. )uuiduuid4hexr-r-r-r. connect_idfs rGtextlimitcCs(|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)rHrIsplit_atr-r-r.(_find_last_newline_or_space_within_limitpsrN text_segmentcCsPt|}|dkr&z |d|d|WSty!|d8}Ynw|dks|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). rNr1r)lenr9UnicodeDecodeError)rOrMr-r-r._find_safe_utf8_split_points   rRrMcCs^|dkr-d|d|vr-|dd|}|d||dkr |S|}|dkr-d|d|vs |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)rHrMZampersand_indexr-r-r."_adjust_split_point_for_xml_entitys rX byte_lengthccst|tr |d}t|tstd|dkrtdt||kr[t||}|dkr/t|}t ||}|dkrz)r%r&r9voicepitchratevolume)r_r`r-r-r.mkssmls recCstdtS)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))timestrftimegmtimer-r-r-r.date_to_string s ri request_id timestampssmlcCsd|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 r-)rjrkrlr-r-r.ssml_headers_plus_data0s  rmc@s,eZdZdZefdddddddddd ed ed ed ed ededdeej deedee dee fddZ de de fddZd(ddZdee dffddZdee dffddZ d)d eee fd!eeee fddfd"d#Zdee ddffd$d%Z d)d eee fd!eeee fddfd&d'ZdS)* Communicatez' Communicate with the service. z+0%z+0HzSentenceBoundaryN <)rcrdrbboundary connectorproxyconnect_timeoutreceive_timeoutrHrarcrdrbrr WordBoundaryrorsrtrurvc Cst||||||_t|tstdttt|d|_|dur)t|ts)td||_ t| t s5tdt| t s>tdt j dd| | d|_ |durVt|t jsVtd||_dd d d d d d |_dS) Nztext must be strizproxy must be strzconnect_timeout must be intzreceive_timeout must be int)totalconnectZ sock_connectZ sock_readz'connector must be aiohttp.BaseConnectorrF) partial_textoffset_compensationlast_duration_offsetstream_was_calledchunk_audio_bytescumulative_audio_bytes)r tts_configr%r:r'r^r rCtextsrtintaiohttpZ ClientTimeoutsession_timeout BaseConnectorrsstate) selfrHrarcrdrbrrrsrtrurvr-r-r.__init__Gs<     zCommunicate.__init__r r"cCst|dD]7}|d}|dvr3|dd|jd}|dd}|||t|ddd d S|d vr8qtd |td )NZMetadataTyperwZDataZOffsetr}ZDurationrHText)typeoffsetdurationrH)Z SessionEndzUnknown metadata type: zNo WordBoundary metadata found)jsonloadsrrrr)rr Zmeta_objZ meta_typecurrent_offsetZcurrent_durationr-r-r.Z__parse_metadatas  zCommunicate.__parse_metadatacCsB|jd|jd7<|jddtt|jd<d|jd<dS)aUpdate inter-chunk offset_compensation from cumulative CBR audio bytes. The output format is audio-24khz-48kbitrate-mono-mp3 (48 kbps CBR). For any CBR stream the byte-to-tick conversion is exact integer arithmetic: ticks = total_bytes * 8 * 10_000_000 // 48_000. This replaces the previous metadata-based accumulation which drifted on long texts due to variable AI silence and Microsoft's integer overflow in reported offsets. rrr2r}rN)rrrrr-r-r.Z__compensate_offsets zCommunicate.__compensate_offsetc sd)fdd }d)fdd }d}tjjdjd4IdH4}|jtd td td t d j t t t d 4IdH|IdH|IdH2z3dHW}|jtjjkr|jd}t||d\}}|dd} | dkr|} | V| d| djd<qR| dkrn| dvrtdqR|jtjjkrt|jdkrtdt|jddd} | t|jkrtdt|j| \}}|ddkrtd|dd} | dvrtd | durt|d!krqRtd"t|d!krtd#d}jd$t|7<d%|d&VqR|jtjjkr,t |jr)|jd'qR6|s5t!d(WdIdHn 1IdHsFwYWdIdHdS1IdHs]wYdS)*Nr"c sPjjdk}|r dnd}|sdnd}dtd|d|dIdHdS) z)Sends the command request to the service.rxtruefalsez 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)rrrsend_strri)Z word_boundarywdsqrZ websocketr-r.send_command_requests    z2Communicate.__stream..send_command_requestc s0ttttjjdIdHdS)z&Sends the SSML request to the service.r|N)rrmrGrirerrr-rr-r.send_ssml_requestsz/Communicate.__stream..send_ssml_requestFT)rs trust_envtimeoutz&ConnectionId=z &Sec-MS-GEC=z&Sec-MS-GEC-Version=)compressrtr)sslr1s sPathsaudio.metadatarrr~sturn.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.fetch_async_items..get_itemsr)asyncioZnew_event_loopZset_event_loopZrun_until_completeclose)rrloopr)rr.fetch_async_itemsrs    z2Communicate.stream_sync..fetch_async_items)r concurrentfuturesThreadPoolExecutorsubmitr)rrrexecutorrr-rr. stream_syncos  "zCommunicate.stream_synccCsNtj}|tj|||}|WddS1s wYdS)z,Synchronous interface for async save method.N)rrrrrrunrresult)rrrrfuturer-r-r. save_syncs  "zCommunicate.save_syncrr)__name__ __module__ __qualname____doc__rr:rr rrrrr&rrrrrrr rrrrr-r-r-r.rnAsr    ;   $   rn)Arrconcurrent.futuresrrrrfrD contextlibriorrrtypingrrrrr r r r Zxml.sax.saxutilsr rrcertifiZtyping_extensionsr constantsrrrrrrZ data_classesrZdrmr exceptionsrrrrrrcreate_default_contextwhererr&rr/r:rCrGrNrRrXr^rerirmrnr-r-r-r.sX   (          E