o
    
iS                     @   s@   d dl Z dd ZdededefddZdededefd	d
ZdS )    Nc                    sF   |r fddt t|| t|| D S  fddt|| D S )Nc                    s"   g | ]\}}|r|  | qS  strip).0edstrip_charsr   S/home/ubuntu/.local/lib/python3.10/site-packages/misaki/vi_cleaner/passage_utils.py
<listcomp>   s   " z'split_text_passages.<locals>.<listcomp>c                    s   g | ]}|  qS r   r   )r   r   r   r   r
   r      s    )zipresplitfindall)textregexinclude_split_charsr	   r   r   r
   split_text_passages   s   (r   passages	maxLengthreturnc                 C   s   t | dkr| S t | d |kr| d gt| dd  |d S t | d | d  |kr@t| d d | d  g| dd   |dS | d gt| dd  |d S )N   r   )r   
   )lencombine_passages)r   r   r   r   r
   r   
   s   *r   c                 C   sL   g }| D ]}t ||kr|t|ddd |  S ||g |  S d S )Nz[.!?]T )r   appendr   )r   r   sub_passagespassager   r   r
   split_long_passages   s   r    )r   r   listintr   r    r   r   r   r
   <module>   s    