o j1@sZddlZddlZddlmZddlZdZedddgZGdddeZGd d d Z dS) N)Enumu;:,.!?¡¿—…"«»“” _punc_indexpuncpositionc@seZdZdZdZdZdZdS) PuncPositionz#Enum for the punctuations positionsrN)__name__ __module__ __qualname____doc__BEGINENDMIDDLErrQ/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/text/punctuation.pyr s rc@sxeZdZdZefdefddZeddZe ddZ e j d dZ d d Z d d Z ddZeddZeddZdS) PunctuationaFHandle punctuations in text. Just strip punctuations from text or strip and restore them later. Args: puncs (str): The punctuations to be processed. Defaults to `_DEF_PUNCS`. Example: >>> punc = Punctuation() >>> punc.strip("This is. example !") 'This is example' >>> text_striped, punc_map = punc.strip_to_restore("This is. example !") >>> ' '.join(text_striped) 'This is example' >>> text_restored = punc.restore(text_striped, punc_map) >>> text_restored[0] 'This is. example !' puncscCs ||_dSN)r)selfrrrr__init__*s zPunctuation.__init__cCstS)z#Return default set of punctuations.) _DEF_PUNCSrrrr default_puncs-szPunctuation.default_puncscCs|jSr)_puncs)rrrrr2szPunctuation.puncscCsNt|tjs tddttt||_t dt |jd|_ dS)Nz%[!] Punctuations must be of type str.z(\s*[z]+\s*)+) isinstancesix string_types ValueErrorjoinlistdictfromkeysrrecompileescapepuncs_regular_exp)rvaluerrrr6s  cCst|jd|S)zRemove all the punctuations by replacing with `space`. Args: text (str): The text to be processed. Example:: "This is. example !" -> "This is example "  )r#subr&rstriplstrip)rtextrrrstrip=s zPunctuation.stripcCs||\}}||fS)zRemove punctuations from text to restore them later. Args: text (str): The text to be processed. Examples :: "This is. example !" -> [["This is", "example"], [".", "!"]] )_strip_to_restore)rr,rrrrstrip_to_restoreIs zPunctuation.strip_to_restorec Cs>tt|j|}|s|ggfSt|dkr'|d|kr'gt|tjgfSg}|D]0}tj }||dkrA| |rAtj}n||dkrQ| |rQtj }| t||q+g}t|D]8\}}||j} | d|j| dd} } | }| dkrqb| | |t|dkrt| dkr| | qb||fS)z+Auxiliary method for Punctuation.preserve()rrNr)r r#finditerr&lengroup _PUNC_IDXrr r startswithendswithrappend enumeratesplitrr) rr,matchesrmatchr splitted_textidxrr9prefixsuffixrrrr.Ws2    zPunctuation._strip_to_restorecCs |||S)aRestore punctuation in a text. Args: text (str): The text to be processed. puncs (List[str]): The list of punctuations map to be used for restoring. Examples :: ['This is', 'example'], ['.', '!'] -> "This is. example!" )_restore)clsr,rrrrrestorews zPunctuation.restorecCs|s|S|sddd|DgS|d}|jtjkr1||j|dg|dd|ddS|jtjkrM|d|jg||dd|ddSt|dkrc||d|jg|ddS||d|j|dg|dd|ddS)z*Auxiliary method for Punctuation.restore()rcss|]}|jVqdSr)r).0mrrr sz'Punctuation._restore..rrNr)rrrr r@rrr2)rAr,rcurrentrrrr@s , ,  4zPunctuation._restoreN)r r r r rstrr staticmethodrpropertyrsetterr-r/r. classmethodrBr@rrrrrs      r) collectionsr#enumrrr namedtupler4rrrrrrs