:i/ &SSKrSSKrSSKrSSKrSSKJr SSKJr SSKJ r SSK J r SSK r SSKrSSKJrJrJrJr \ (aSSKJr \R,"\5r\"5(a0SSKr\R6"5r\"S\Vs0sHoU_M sn5rO \"SS S 05r\"5(aSSKr\R<R?5Vs/sHoR@RC5PM snr""S S 5r#S \$\ RJ-S \$\ RJ-4Sjr&S\ RJS \ RJ4Sjr'S\ RJS\ RJS \ RJ4Sjr(\S\)S\)S\$S\$S\)S \ RJ4 Sj5r*gs snfs snf)N)Enum)cache)Path) TYPE_CHECKING)assert_soundfile_installedassert_soxr_installedis_soundfile_installedis_soxr_installed)RawAudio AudioFormatnonecB\rSrSrS\R S\S\SS4SjrS\4Sjr SS jr \ S\ 4S j5r \SS \S \SS4S jj5r\SS\S \SS4Sjj5r\SS\S \SS4Sjj5r\SS\S \SS4Sjj5rSS\S\S\4Sjjr\SSj5rS\SS4SjrSrg)Audio* audio_array sampling_rateformatreturnNcHXlX lX0lUR5 g)zInitialize an Audio instance with audio data, sampling rate, and format. Args: audio_array: The audio data as a numpy array. sampling_rate: The sampling rate of the audio in Hz. format: The format of the audio file. N)rrr _check_valid)selfrrrs N/home/james-whalen/.local/lib/python3.13/site-packages/mistral_common/audio.py__init__Audio.__init__+s!'*  cSURS[UR5UR- SSURR3$)NzAudio - sampling_rate=z Hz, duration=z.2fz s, shape=)rlenrshapers r__repr__Audio.__repr__8sU$T%7%7$89D,,-0B0BB3GH%%++, . rcz[UR[R5(d[ [R55eURR S:XdSURR <35e[ 5 UR[;dSUR<S[<35eg)Nzself.audio_array.ndim=z self.format= not in EXPECTED_FORMAT_VALUES=) isinstancernpndarraytypendimrrEXPECTED_FORMAT_VALUESrs rrAudio._check_valid?s$**BJJ77Ibjj9II7$$)F.Dd.>.>.C.C-E+FF)"${{44hFfOeNg6hh4rcRURRSUR- nU$)zcCalculate the duration of the audio in seconds. Returns: The duration of the audio in seconds. r)rrr)rdurations rr-Audio.durationEs***003d6H6HHrurlstrictc[R"U5nUR5 [R UR US9$![R an[SU35UeSnAf[an[SUS35UeSnAff=f)zCreate an Audio instance from a URL. Args: url: The URL of the audio file. strict: Whether to strictly enforce mono audio. Returns: An instance of the Audio class. r0z#Failed to download audio from URL: Nz*Failed to create Audio instance from URL: z .) requestsgetraise_for_statusr from_bytescontentRequestException ValueError Exception)r/r0responsees rfrom_urlAudio.from_urlPs Z||C(H  % % '##H$4$4V#D D(( QB3%HIq P ZI#bQRXY Y Zs$AAB A)) B 6BB  audio_base64c[5 [R"SU5(aURS5Sn[R "U5n[RX!S9$![ an[S5UeSnAff=f)zCreate an Audio instance from a base64 encoded string. Args: audio_base64: The base64 encoded audio data. strict: Whether to strictly enforce mono audio. Defaults to True. Returns: An instance of the Audio class. z^data:audio/\w+;base64,,r#zHbase64 decoding failed. Please check the input string is a valid base64.Nr2) rrematchsplitbase64 b64decoder:r9rr6)r?r0 audio_bytesr<s r from_base64Audio.from_base64ds #$ 88. = ='--c215L p **<8K ;; pghno o psA%% B/ A;;Bfilecf[5 [U[5(aURS5(aUSSn[ U5R 5(d[ SU<S35e[US5nUR5nSSS5 [RWUS9$!,(df  N"=f)zCreate an Audio instance from an audio file. Args: file: Path to the audio file. strict: Whether to strictly enforce mono audio. Defaults to True. Returns: An instance of the Audio class. zfile://Nzfile=z does not existrbr2) rr%str startswithrexistsFileNotFoundErroropenreadrr6)rJr0frGs r from_fileAudio.from_file{s #$ dC T__Y%?%?8DDz  ""#utg_$=> > $ &&(K F;; s 5B"" B0rGc[R"U5n[R"U5nUR SS9nUR nUR nSSS5 SSS5 [W5nURR5nWRS:wa/U(a[SUR<35eURSS9n[UWUS9$!,(df  N=f!,(df  N=f)zCreate an Audio instance from bytes. Args: audio_bytes: The audio data as bytes. strict: Whether to strictly enforce mono audio. Defaults to True. Returns: An instance of the Audio class. float32)dtypeNr#zaudio_array.ndim=axis)rrr)ioBytesIOsf SoundFilerS sampleraterr valuelowerr)r9meanr) rGr0 audio_filerTrr audio_format format_enumrs rr6Audio.from_bytessZZ $ j)Qff9f5 !  xx *%",/ ""((*   q  $5K$4$4#6!788)..A.6 MRXYY*)% $s"C&(CC& C# C&& C4prefixc[5 U[;dSU<S[<35e[R"5n[R "X0R URUR5S9 URS5 [R"UR55RS5nSSS5 U(aSUR5SW3nW$!,(df  N-=f) zConvert the audio data to a base64 encoded string. Args: format: The format to encode the audio in. prefix: Whether to add a data prefix to the base64 encoded string. Returns: The base64 encoded audio data. zformat=r$)rrzutf-8Nz data:audio/z;base64,)rr*r\r]r^writerrupperseekrE b64encoderSdecoderb)rrrhrd base64_strs r to_base64Audio.to_base64s #$//^GF9<\E[D]1^^/ ZZ\Z HHZ!1!143E3Eflln ] OOA ))*//*;<CCGLJ &v||~&6hzlKJ \s A=C!! C/c<[UR[5(a[R UR5$[UR[ 5(a[R UR5$[S[UR535e)zCreate an Audio instance from a RawAudio object. Args: audio: The RawAudio object containing audio data. Returns: An instance of the Audio class. zUnsupported audio data type: ) r%databytesrr6rNrHr9r()audios rfrom_raw_audioAudio.from_raw_audiosm ejj% ( (##EJJ/ /  C ( ($$UZZ0 0/K$44Kq!"u AB/I jj!bjj6XYK rnum_frequency_bins num_mel_bins min_frequency max_frequencyrcUS:a[SUS35eX#:a[SUSU35e[U5n[U5n[R"XVUS-5n[ U5n[R"SUS-U5n [ X5n SUSUS-USU- - n U [R "U S5-n U RSS 9S :HR5(a[S US US 35eU $)a"Create a Mel filter bank matrix for converting frequency bins to the Mel scale. This function generates a filter bank matrix that can be used to transform a spectrum represented in frequency bins to the Mel scale. The Mel scale is a perceptual scale of pitches judged by listeners to be equal in distance from one another. Args: num_frequency_bins: The number of frequency bins in the input spectrum. num_mel_bins: The number of desired Mel bins in the output. min_frequency: The minimum frequency (in Hz) to consider. max_frequency: The maximum frequency (in Hz) to consider. sampling_rate: The sampling rate of the audio signal. Returns: A filter bank matrix of shape (num_mel_bins, num_frequency_bins) that can be used to project frequency bin energies onto Mel bins. rzRequire num_frequency_bins: z >= 2zRequire min_frequency: z <= max_frequency: rg@NrZgzNAt least one mel filter has all zero values. The value for `num_mel_filters` (z?) may be set too high. Or, the value for `num_frequency_bins` (z) may be set too low.) r9rr&linspacerrrmaxany) rrrrrmel_minmel_max mel_freqsrr mel_filtersenorms rmel_filter_bankr&s*2A78J7K5QRR$2=/ATUbTcdee=)G=)G GlQ.>?I *L A}13EFI0IK <L1$45 ]l8SS TE2>>%++KQ3&++-- 00<~>###8  :   r)+rEr\loggingrBenumr functoolsrpathlibrtypingrnumpyr&r3mistral_common.importsrrr r &mistral_common.protocol.instruct.chunkr getLoggerrlogger soundfiler^available_formatsr r| __members__valuesrarbr*rrr'rrrrr) format_namevs00rrs   ?   8 $,,.}Sd&eSdKK'?Sd&efK}vv&67K3>3J3J3Q3Q3ST3Sa''--/3STz/z/zurzz)ebjj.@0rzzbjj&bjj WYWaWa(5555 5  5 ZZ 55O'fUs  F #F