U ,:%e @sddlZddlZddlmZddlmZmZddlZddlm Z ddl m Z ddl m Z ddlmZdd d d d iZGd dde ZdS)N)Path)TupleUnion)Tensor)Dataset)download_url_to_file) _extract_tarrelease1Zwavsz:https://data.keithito.com/data/speech/LJSpeech-1.1.tar.bz2Z@be1a30453f28eb8dd26af4101ae40cbf2c50413b1bb21936cbcdc6fae3de8aa5)folder_in_archiveurlchecksumc@seZdZdZeddedddfeeefeeedddd Z eeeeddd d Z e e e e eefd d dZe dddZdS)LJSPEECHa,*LJSpeech-1.1* :cite:`ljspeech17` dataset. Args: root (str or Path): Path to the directory where the dataset is found or downloaded. url (str, optional): The URL to download the dataset from. (default: ``"https://data.keithito.com/data/speech/LJSpeech-1.1.tar.bz2"``) folder_in_archive (str, optional): The top-level directory of the dataset. (default: ``"wavs"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``). r r r FN)rootr r downloadreturncCs|||||dSN)_parse_filesystem)selfrr r rr[/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torchaudio/datasets/ljspeech.py__init__#szLJSPEECH.__init__c Cst|}tj|}||}t|dd}||}|||_||d|_|rtj|jstj|st dd}t |||dt |n tj |jst d|jdt|jd d d "}tj|d tjd } t| |_W5QRXdS)Nz.tar.bz2rz metadata.csvr r )Z hash_prefixz The path zT doesn't exist. Please check the ``root`` path or set `download=True` to download itr)newline|) delimiterquoting)rospathbasenamesplit_pathZ_metadata_pathisdirisfile_RELEASE_CONFIGSrrexists RuntimeErroropencsvreader QUOTE_NONElist_flist) rrr r rrarchiver metadataflistrrrr-s(      zLJSPEECH._parse_filesystem)nrc Cs<|j|}|\}}}|j|d}t|\}}||||fS)arLoad the n-th sample from the dataset. Args: n (int): The index of the sample to be loaded Returns: Tuple of the following items; Tensor: Waveform int: Sample rate str: Transcript str: Normalized Transcript z.wav)r,r! torchaudioload) rr0lineZfileidZ transcriptZnormalized_transcriptZ fileid_audioZwaveformZ sample_raterrr __getitem__Js  zLJSPEECH.__getitem__)rcCs t|jSr)lenr,)rrrr__len__jszLJSPEECH.__len__)__name__ __module__ __qualname____doc__r$rstrrboolrrintrrr4r6rrrrr s     r )r(rpathlibrtypingrrr1ZtorchrZtorch.utils.datarZtorchaudio._internalrZtorchaudio.datasets.utilsrr$r rrrrs