o 㥵i @sddlZddlmZddlmZddlmZddlmZhdZ ddhZ d ee Bd edBfd d Z d dZ ddZeddfd eee fdee deded eef ddZd ee Bd eeee e e ffddZdS)N)Path)Union)logger) natsorted> .aac.aif.m4a.mp3.ogg.wav.wma.aifc.aiff.flacz.mp4z.avipathreturncCsFt|}|dur dSt|dtjjd}t|dkrdS|dS)NFz*.ckpt)keyr)rexistssortedglobosrgetmtimelen)rckpt_dirckptsrJ/home/ubuntu/.local/lib/python3.10/site-packages/fish_speech/utils/file.pyget_latest_checkpoints  rcCsL|rt|s dSt|d }|}Wd|S1swY|S)Nrb)rropenread) file_pathwav_filewavrrraudio_to_bytes)s   r%cCsRt|}|r'|r'|jddd }|WdS1s"wY|S)Nrutf-8encoding)rris_filer r!)ref_textrfilerrr read_ref_text1s  r-FT extensions recursivesortcsLttr tstddfdd|D}|r$t|}|S)aoList files in a directory. Args: path (Path): Path to the directory. extensions (set, optional): Extensions to filter. Defaults to None. recursive (bool, optional): Whether to search recursively. Defaults to False. sort (bool, optional): Whether to sort the files. Defaults to True. Returns: list: List of files. z Directory z does not exist.cs&g|]}d|D]}|q qS)*)rglob).0extr,rrr Qs&zlist_files..) isinstancestrrrFileNotFoundErrorr)rr.r/r0filesrr5r list_files9s r;cCsht}g}d\}}ddgddgdgd}t|dddw}|D]j}|jd d d }t|d kr:td |q |\} } } } t| } |  } | dkrPd} | dvs[Jd| || }| |vrptd| |d7}q | std| |d7}q | | | || fq Wdn1swY|dkrtd||dkrtd||S)z+ Load a Bert-VITS2 style filelist. )rrzhenjp)r<r>r=r&r'r(|)maxsplitzInvalid line: jazInvalid language zDuplicated file: zFile not found: NrzTotal duplicated files: zTotal files not found: ) setr readlinesstripsplitrrwarningrlowerrappend)rr:resultscount_duplicatedcount_not_foundLANGUAGE_TO_LANGUAGESflinesplitsfilenamespeakerlanguagetextr, languagesrrr load_filelistYsH    rX)rpathlibrtypingrlogururnatsortrAUDIO_EXTENSIONSVIDEO_EXTENSIONSr8rr%r-rEboollistr;tuplerXrrrrs6       *