o 4ѹg @s6dZddlmZddlZddlmZGdddZdS)zGSubMaker module is used to generate subtitles from WordBoundary events.)ListN)TTSChunkc@sZeZdZdZdddZdeddfddZd eddfd d Zde fd d Z de fddZ dS)SubMakerzL SubMaker is used to generate subtitles from WordBoundary messages. returnNcCs g|_dSN)cuesselfr U/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/edge_tts/submaker.py__init__s zSubMaker.__init__msgc Csh|ddkr td|jtjt|jdtj|dddtj|d|ddd|d d d S) z Feed a WordBoundary message to the SubMaker object. Args: msg (dict): The WordBoundary message. Returns: None typeZ WordBoundaryz-Invalid message type, expected 'WordBoundary'roffset ) microsecondsdurationtextindexstartendcontentN) ValueErrorrappendsrtSubtitlelen timedelta)r rr r r feeds  z SubMaker.feedwordscCs|dkrtdt|jdkrdSg}|jd}|jddD]%}t|j|kr=tj|j|j|j |jd|jd}q| ||}q| |||_dS)z Merge cues to reduce the number of cues. Args: words (int): The number of words to merge. Returns: None rz.Invalid number of words to merge, expected > 0Nr r) rrrrsplitrrrrrr)r r!Znew_cuesZ current_cueZcuer r r merge_cues(s$     zSubMaker.merge_cuescCs t|jS)z Get the SRT formatted subtitles from the SubMaker object. Returns: str: The SRT formatted subtitles. N)rZcomposerr r r r get_srtHs zSubMaker.get_srtcCs|Sr)r%r r r r __str__QszSubMaker.__str__)rN) __name__ __module__ __qualname____doc__r rr intr$strr%r&r r r r r s   r)r*typingrrrrr r r r s