/Ui ddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z dZ e ddf dd ZeGd d ZGd d Zy)) annotationsN) dataclass) UserError)npnpti]cN|jtjk(r:tj|dd}|dzj tj }n(|jtj k7r t dtj}tj|d5}|j||j||j||j|j|j!ddddd|dfS#1swYxYw) N?0Buffer must be a numpy array of int16 or float32wrz audio.wavz audio/wav)dtyperfloat32clipastypeint16rioBytesIOwaveopen setnchannels setsampwidth setframerate writeframestobytesseek)buffer frame_rate sample_widthchannels audio_filewav_files n/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/voice/input.py_buffer_to_audio_filer&s ||rzz!s+5.((2  !JKKJ :s #xh'l+j)V^^-.  [ 11s *A$DD$c^eZdZUdZded< eZded< dZded< dZded < d d Z dd Z y ) AudioInputz7Static audio to be used as input for the VoicePipeline.z"npt.NDArray[np.int16 | np.float32]rintr rr!rr"cnt|j|j|j|jS)z2Returns a tuple of (filename, bytes, content_type))r&rr r!r"selfs r% to_audio_filezAudioInput.to_audio_file:s($T[[$//4CTCTVZVcVcddc|jjtjk(rXtj|jdd|_|jdzj tj |_n2|jjtj k7r tdtj|jjjdS)z2Returns the audio data as a base64 encoded string.r r r rzutf-8) rrrrrrrrbase64 b64encoderdecoder+s r% to_base64zAudioInput.to_base64>s ;;   *''$++tS9DK;;.66rxx@DK [[  "(( *NO O 3 3 56==gFFr.N)returntuple[str, io.BytesIO, str])r4str) __name__ __module__ __qualname____doc____annotations__DEFAULT_SAMPLE_RATEr r!r"r-r3r.r%r(r((sEA ..*J)?L#<HcBe Gr.r(ceZdZdZdZddZy)StreamedAudioInputzAudio input represented as a stream of audio data. You can pass this to the `VoicePipeline` and then push audio data into the queue using the `add_audio` method. c6tj|_y)N)asyncioQueuequeuer+s r%__init__zStreamedAudioInput.__init__OsOV}} r.cVK|jj|d{y7w)zAdds more audio data to the stream. Args: audio: The audio data to add. Must be a numpy array of int16 or float32 or None. If None passed, it indicates the end of the stream. N)rCput)r,audios r% add_audiozStreamedAudioInput.add_audioRsjjnnU###s )')N)rGz)npt.NDArray[np.int16 | np.float32] | None)r7r8r9r:rDrHr=r.r%r?r?Js_$r.r?) rz/npt.NDArray[np.int16 | np.float32 | np.float64]r r)r!r)r"r)r4r5) __future__rrAr0rr dataclassesr exceptionsrimportsrr r<r&r(r?r=r.r%rMs" !" * 2 ;222 2 ! 22 GG GB$$r.