o h@stdZddlZddlmZmZmZddlZddlm Z ddl m Z ddl m Z mZmZddlmZGd d d e ZdS) z Processor class for Pop2Piano.N)ListOptionalUnion) BatchFeature)ProcessorMixin) BatchEncodingPaddingStrategyTruncationStrategy) TensorTypecs eZdZdZddgZdZdZfddZ       d#d ee j e e e e j fd ee e e fde deedee efdeeeefdeeeefdee dee dedeeeffddZ d$dededefddZeddZfdd Zed!d"ZZS)%Pop2PianoProcessora Constructs an Pop2Piano processor which wraps a Pop2Piano Feature Extractor and Pop2Piano Tokenizer into a single processor. [`Pop2PianoProcessor`] offers all the functionalities of [`Pop2PianoFeatureExtractor`] and [`Pop2PianoTokenizer`]. See the docstring of [`~Pop2PianoProcessor.__call__`] and [`~Pop2PianoProcessor.decode`] for more information. Args: feature_extractor (`Pop2PianoFeatureExtractor`): An instance of [`Pop2PianoFeatureExtractor`]. The feature extractor is a required input. tokenizer (`Pop2PianoTokenizer`): An instance of ['Pop2PianoTokenizer`]. The tokenizer is a required input. feature_extractor tokenizerPop2PianoFeatureExtractorPop2PianoTokenizercst||dSN)super__init__)selfr r __class__h/var/www/html/ai/venv/lib/python3.10/site-packages/transformers/models/pop2piano/processing_pop2piano.pyr/szPop2PianoProcessor.__init__NTFaudio sampling_ratesteps_per_beatresamplenotespadding truncation max_lengthpad_to_multiple_ofverbosereturnc  Ks|dur|dur|durtd|dur%|dur%|jd||||d| } |dur8|jd||||| | d| } |dur>| S|dusF|durH| S| d| d<| S)a" This method uses [`Pop2PianoFeatureExtractor.__call__`] method to prepare log-mel-spectrograms for the model, and [`Pop2PianoTokenizer.__call__`] to prepare token_ids from notes. Please refer to the docstring of the above two methods for more information. Nz{You have to specify at least audios and sampling_rate in order to use feature extractor or notes to use the tokenizer part.)rrrr)rrr r!r"r# token_idsr) ValueErrorr r)rrrrrrrr r!r"r#kwargsinputsencoded_token_idsrrr__call__2s<  zPop2PianoProcessor.__call__feature_extractor_output return_midicCs|jj|||dS)z This method uses [`Pop2PianoTokenizer.batch_decode`] method to convert model generated token_ids to midi_notes. Please refer to the docstring of the above two methods for more information. )r%r+r,)r batch_decode)rr%r+r,rrrr-ls zPop2PianoProcessor.batch_decodecCs"|jj}|jj}tt||Sr)rmodel_input_namesr listdictfromkeys)rtokenizer_input_namesfeature_extractor_input_namesrrrr.|sz$Pop2PianoProcessor.model_input_namesc s>tj|rtd|dtj|ddtj|fi|S)NzProvided path (z#) should be a directory, not a fileT)exist_ok)ospathisfiler&makedirsrsave_pretrained)rsave_directoryr'rrrr9s z"Pop2PianoProcessor.save_pretrainedcKs|j|fi|}||Sr)_get_arguments_from_pretrained)clspretrained_model_name_or_pathr'argsrrrfrom_pretrainedsz"Pop2PianoProcessor.from_pretrained) NNrTNFNNNT)T)__name__ __module__ __qualname____doc__ attributesfeature_extractor_classtokenizer_classrrnpndarrayrfloatintrboolr strr r rrr*r-propertyr.r9 classmethodr? __classcell__rrrrr sh        >   r )rCr5typingrrrnumpyrGfeature_extraction_utilsrprocessing_utilsrtokenization_utilsrr r utilsr r rrrrs