o i @s dZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZdd lmZdd lmZmZzddlZWn eybZzed eed ed edZ[wwGdddeZGdddeZGdddeZGdddeZdS)zLocal audio transport implementation for Pipecat. This module provides a local audio transport that uses PyAudio for real-time audio input and output through the system's default audio devices. N)ThreadPoolExecutor)Optional)logger)InputAudioRawFrameOutputAudioRawFrame StartFrame)FrameProcessor)BaseInputTransport)BaseOutputTransport) BaseTransportTransportParamsz Exception: z~In order to use local audio, you need to `pip install pipecat-ai[local]`. On MacOS, you also need to `brew install portaudio`.zMissing module: c@s2eZdZUdZdZeeed<dZeeed<dS)LocalAudioTransportParamszConfiguration parameters for local audio transport. Parameters: input_device_index: PyAudio device index for audio input. If None, uses default. output_device_index: PyAudio device index for audio output. If None, uses default. Ninput_device_indexoutput_device_index) __name__ __module__ __qualname____doc__rrint__annotations__rrrR/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/transports/local/audio.pyr #s r cs\eZdZUdZeed<dejdeffdd Zde ffdd Z fd d Z d d Z Z S)LocalAudioInputTransportzLocal audio input transport using PyAudio. Captures audio from the system's audio input device and converts it to InputAudioRawFrame objects for processing in the pipeline. _paramspy_audioparamscs"t|||_d|_d|_dS)zInitialize the local audio input transport. Args: py_audio: PyAudio instance for audio device management. params: Transport configuration parameters. Nr)super__init__ _py_audio _in_stream _sample_rateselfrr __class__rrr8s  z!LocalAudioInputTransport.__init__framec st|IdH|jrdS|jjp|j|_t|jdd}|jj|j d|jj |j||j d|jj d|_|j ||IdHdS)z~Start the audio input stream. Args: frame: The start frame containing initialization parameters. NdT)formatchannelsrateframes_per_bufferstream_callbackinputr)rstartrraudio_in_sample_rater rropenget_format_from_widthaudio_in_channels_audio_in_callbackr start_streamset_transport_ready)r"r% num_framesr#rrr.Es"  zLocalAudioInputTransport.startc:tIdH|jr|j|jd|_dSdS)z(Stop and cleanup the audio input stream.N)rcleanupr stop_streamcloser"r#rrr8`   z LocalAudioInputTransport.cleanupcCs4t||j|jjd}t|||dtj fS)z+Callback function for PyAudio input stream.)audio sample_rate num_channelsN) rr rr2asynciorun_coroutine_threadsafepush_audio_frameget_event_looppyaudio paContinue)r"in_data frame_count time_infostatusr%rrrr3hs z+LocalAudioInputTransport._audio_in_callback)rrrrr rrDPyAudiorrr.r8r3 __classcell__rrr#rr/s  rcsfeZdZUdZeed<dejdeffdd Zde ffdd Z fd d Z de d e fd dZZS)LocalAudioOutputTransportzLocal audio output transport using PyAudio. Plays audio frames through the system's audio output device by converting OutputAudioRawFrame objects to playable audio data. rrrcs.t|||_d|_d|_tdd|_dS)zInitialize the local audio output transport. Args: py_audio: PyAudio instance for audio device management. params: Transport configuration parameters. Nr) max_workers)rrr _out_streamr r _executorr!r#rrr~s z"LocalAudioOutputTransport.__init__r%csvt|IdH|jrdS|jjp|j|_|jj|jd|jj |jd|jj d|_|j | |IdHdS)zStart the audio output stream. Args: frame: The start frame containing initialization parameters. Nr'T)r(r)r*outputr) rr.rOraudio_out_sample_rater rr0r1audio_out_channelsrr4r5r"r%r#rrr.s  zLocalAudioOutputTransport.startcr7)z)Stop and cleanup the audio output stream.N)rr8rOr9r:r;r#rrr8r<z!LocalAudioOutputTransport.cleanupreturncs0|jr||j|jj|jIdHdSdS)zWrite an audio frame to the output stream. Args: frame: The audio frame to write to the output device. Returns: True if the audio frame was written successfully, False otherwise. NTF)rOrCrun_in_executorrPwriter=rTrrrwrite_audio_frames  z+LocalAudioOutputTransport.write_audio_frame)rrrrr rrDrJrrr.r8rboolrXrKrrr#rrLus  rLcsBeZdZdZdeffdd ZdefddZdefdd ZZ S) LocalAudioTransportzComplete local audio transport with input and output capabilities. Provides a unified interface for local audio I/O using PyAudio, supporting both audio capture and playback through the system's audio devices. rcs*t||_t|_d|_d|_dS)zuInitialize the local audio transport. Args: params: Transport configuration parameters. N)rrrrDrJ_pyaudio_input_output)r"rr#rrrs   zLocalAudioTransport.__init__rUcC|js t|j|j|_|jS)z}Get the input frame processor for this transport. Returns: The audio input transport processor. )r\rr[rr;rrrr-zLocalAudioTransport.inputcCr^)zGet the output frame processor for this transport. Returns: The audio output transport processor. )r]rLr[rr;rrrrQr_zLocalAudioTransport.output) rrrrr rrr-rQrKrrr#rrZs  rZ)rr@concurrent.futuresrtypingrlogururpipecat.frames.framesrrr"pipecat.processors.frame_processorrpipecat.transports.base_inputr pipecat.transports.base_outputr !pipecat.transports.base_transportr r rDModuleNotFoundErroreerror Exceptionr rrLrZrrrrs0        FK