o ٷi* @sddlmZddlZddlmZddlmZmZzddl Z Wn e y)dZ Ynwzddl Z Wn e y;dZ Ynwee Z GdddZdS))BytesION) init_logger) AudioResponse CreateAudioc@s:eZdZdZdedefddZdejde de fd d Z d S) AudioMixinz+Mixin class to add audio-related utilities. audio_objreturnc CsV|j}|j}|j}|j}|j}|j}|dkrtd|tdur't d|j dkr5td|j d| |||\}}dd ifd d d d ifddifddifddifddd difd}||vrlt d|dd}||\} } } t} tj| ||fd| i| | } Wdn1swY|rddl}|| d} t| | dS)z6Convert audio tensor to bytes in the specified format.audiozUnsupported stream format: NzYsoundfile is required for audio generation. Please install it with: pip install soundfilez$Unsupported audio tensor dimension: z/. Only mono (1D) and stereo (2D) are supported.WAVz audio/wavRAWz audio/pcmsubtypePCM_16FLACz audio/flacMP3z audio/mpegAACz audio/aacOGGz audio/oggOPUS)wavpcmflacmp3aacopuszUnsupported response format 'z', defaulting to 'wav'.rformatrzutf-8) audio_data media_type) audio_tensor sample_rateresponse_formatlower stream_format base64_encodespeed ValueError soundfile ImportErrorndim_apply_speed_adjustmentloggerwarningrwritegetvaluebase64 b64encodedecoder)selfrrrrr!r"r#supported_formatssoundfile_formatrkwargsbufferrr-r5b/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/entrypoints/openai/audio_utils_mixin.py create_audiosH        zAudioMixin.create_audiorr#rc Cs|dkr||fStdurtdzt|jtjs|tj}tjj ||d}||fWSt yD}z t d|t d|d}~ww)zBApply speed adjustment to the audio tensor while preserving pitch.g?NzUlibrosa is required for speed adjustment. Please install it with: pip install librosa)yratez+An error occurred during speed adjustment: z!Failed to apply speed adjustment.)librosar&np issubdtypedtypefloatingastypefloat32effects time_stretch Exceptionr)errorr$)r0rr#rstretched_audioer5r5r6r(Ls   z"AudioMixin._apply_speed_adjustmentN) __name__ __module__ __qualname____doc__rrr7r;ndarrayfloatintr(r5r5r5r6rs4r)iornumpyr; vllm.loggerr+vllm_omni.entrypoints.openai.protocol.audiorrr%r&r:rGr)rr5r5r5r6s