o #1 i$/ @sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl ZddlZddlmZmZmZmZe rFddlmZeeZeraddlZeZedd d eDZnedd d iZeroddlZd d ej DZ!GdddZ"de e#ej$fde e#ej$ffddZ%dej$dej$fddZ&dej$dej$dej$fddZ'ede(de(de#de#de(dej$f d d!Z)dS)"N)Enum)cache)Path) TYPE_CHECKINGUnion)assert_soundfile_installedassert_soxr_installedis_soundfile_installedis_soxr_installed)RawAudio AudioFormatcCsi|]}||qSr ).0 format_namer r `/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/mistral_common/audio.py srnonecCsg|]}|jqSr )valuelower)rvr r r 'src@seZdZdejdededdfddZdefdd Zd'd d Z e de fd d Z e d(dededdfddZe d(dededdfddZe d(dededdfddZe d(dededdfddZd)dededefddZe d*d"d#Zd$eddfd%d&ZdS)+Audio audio_array sampling_rateformatreturnNcCs||_||_||_|dS)a Initialize 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)selfrrrr r r__init__+s zAudio.__init__cCs*d|jdt|j|jdd|jjS)NzAudio - sampling_rate=z Hz, duration=z.2fz s, shape=)rlenrshaperr r r__repr__8s zAudio.__repr__cCsdt|jtjsJttj|jjdksJd|jjt|jtvs0Jd|jdtdS)Nzself.audio_array.ndim=z self.format= not in EXPECTED_FORMAT_VALUES=) isinstancernpndarraytypendimrrEXPECTED_FORMAT_VALUESr!r r rr?s$zAudio._check_validcCs|jjd|j}|S)z{Calculate the duration of the audio in seconds. Returns: The duration of the audio in seconds. r)rr r)rdurationr r rr+EszAudio.durationTurlstrictc Csxzt|}|tj|j|dWStjy'}ztd||d}~wty;}z td|d|d}~ww)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. r-z#Failed to download audio from URL: Nz*Failed to create Audio instance from URL: z .) requestsgetraise_for_statusr from_bytescontentRequestException ValueError Exception)r,r-responseer r rfrom_urlPs zAudio.from_url audio_base64c Cs`ttd|r|dd}zt|}Wnty(}ztd|d}~wwtj ||dS)aCreate 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.Nr.) rrematchsplitbase64 b64decoder6r5rr2)r:r- audio_bytesr8r r r from_base64ds  zAudio.from_base64filecCstt|tr|dr|dd}t|s!td|dt|d }|}Wdn1s5wYt j ||dS)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 existrbr.) rr%str startswithrexistsFileNotFoundErroropenreadrr2)rCr-frAr r r from_file{s     zAudio.from_filerAc Cst|)}t|}|jdd}|j}|j}Wdn1s"wYWdn1s1wYt|}|j }|j dkrT|rNt d|j |j dd}t |||dS)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. Zfloat32)ZdtypeNr#zaudio_array.ndim=Zaxis)rrr)ioBytesIOsfZ SoundFilerKZ sampleraterr rrr)r5meanr) rAr- audio_filerLrrZ audio_formatZ format_enumrr r rr2s     zAudio.from_bytesFprefixcCst|tvsJd|dtt$}tj||j|j|d| dt |  d}Wdn1s= 2zRequire min_frequency: z <= max_frequency: rg@NrNgzNAt 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.) r5rwr&Zlinspaceryrrmaxany) rrrrrZmel_minZmel_maxZ mel_freqsr{rzZ mel_filtersZenormr r rmel_filter_bank&s*  r)*r?rOloggingr<enumr functoolsrpathlibrtypingrrnumpyr&r/Zmistral_common.importsrrr r Z&mistral_common.protocol.instruct.chunkr getLoggerrbloggerZ soundfilerQZavailable_formatsr r` __members__valuesr*rrgr'rwryrrerr r r rsR     &>