qi\dZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZdd lmZdd lmZmZ ddlZGddeZGddeZGddeZGddeZy#e$r7Zej0d eej0d ed edZ[wwxYw)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:eZdZUdZdZeeed<dZeeed<y)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__rP/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/transports/local/audio.pyrr#s')- ,)-#-rrcleZdZUdZeed<dejdeffd Zde ffd Z fdZ d Z xZ 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_audioparamscNt||||_d|_d|_y)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__s rr"z!LocalAudioInputTransport.__init__8s(  !rframec FKt||d{|jry|jjxs |j|_t |j dz dz}|jj|jjd|jj|j ||jd|jj|_|jj|j|d{y77w)z~Start the audio input stream. Args: frame: The start frame containing initialization parameters. NdT)formatchannelsrateframes_per_bufferstream_callbackinputr)r!startr$raudio_in_sample_rater%rr#openget_format_from_widthaudio_in_channels_audio_in_callbackr start_streamset_transport_ready)r'r) num_framesr(s rr3zLocalAudioInputTransport.startEs gmE""" ??  LL==[A[A[**S01A5 ..-->>77:\\33""( 33#||>>.  $$&&&u---' #& .s"D!DC?D!DD!D!cKt|d{|jr<|jj|jj d|_yy7Mw)z(Stop and cleanup the audio input stream.N)r!cleanupr$ stop_streamcloser'r(s rr=z LocalAudioInputTransport.cleanup`sOgo ?? OO ' ' ) OO ! ! #"DO  A%A#AA%ct||j|jj}t j |j ||jdtjfS)z+Callback function for PyAudio input stream.)audio sample_rate num_channelsN) rr%rr7asynciorun_coroutine_threadsafepush_audio_frameget_event_looppyaudio paContinue)r'in_data frame_count time_infostatusr)s rr8z+LocalAudioInputTransport._audio_in_callbackhs\"))77  (()>)>u)EtGZGZG\]g(())r)rrrrrrrJPyAudior"rr3r=r8 __classcell__r(s@rrr/s> '&  :S ..6# *rrcveZdZUdZeed<dejdeffd Zde ffd Z fdZ de d e fd ZxZS) LocalAudioOutputTransportzLocal audio output transport using PyAudio. Plays audio frames through the system's audio output device by converting OutputAudioRawFrame objects to playable audio data. rrrcpt||||_d|_d|_t d|_y)zInitialize the local audio output transport. Args: py_audio: PyAudio instance for audio device management. params: Transport configuration parameters. Nr) max_workers)r!r"r# _out_streamr%r _executorr&s rr"z"LocalAudioOutputTransport.__init__~s7  !,:rr)cKt||d{|jry|jjxs |j|_|j j|j jd|jj|j d|jj|_|jj|j|d{y77w)zStart the audio output stream. Args: frame: The start frame containing initialization parameters. Nr,T)r-r.r/outputr) r!r3rXraudio_out_sample_rater%r#r5r6audio_out_channelsrr9r:)r'r)r(s rr3zLocalAudioOutputTransport.starts gmE"""     LL>>]%B]B]>>..>>77:\\44"" $ @ @ /  %%'&&u---! # .s"C9C5CC9/C70C97C9cKt|d{|jr<|jj|jj d|_yy7Mw)z)Stop and cleanup the audio output stream.N)r!r=rXr>r?r@s rr=z!LocalAudioOutputTransport.cleanupsVgo       ( ( *    " " $#D   rAreturncK|jrR|jj|j|jj|j d{yy7w)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)rXrIrun_in_executorrYwriterC)r'r)s rwrite_audio_framez+LocalAudioOutputTransport.write_audio_framesZ   %%'77 0 0 6 6     sAA$A"A$)rrrrrrrJrPr"rr3r=rboolrcrQrRs@rrTrTusM '&;;:S;"..0$-@TrrTc@eZdZdZdeffd ZdefdZdefdZxZ 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. rc~t|||_tj|_d|_d|_y)zuInitialize the local audio transport. Args: params: Transport configuration parameters. N)r!r"rrJrP_pyaudio_input_output)r'rr(s rr"zLocalAudioTransport.__init__s3  ) :> <@ rr_c||js%t|j|j|_|jS)z}Get the input frame processor for this transport. Returns: The audio input transport processor. )rirrhrr's rr2zLocalAudioTransport.inputs+ {{24==$,,ODK{{rc||js%t|j|j|_|jS)zGet the output frame processor for this transport. Returns: The audio output transport processor. )rjrTrhrrls rr[zLocalAudioTransport.outputs+ ||4T]]DLLQDL||r) rrrrrr"r r2r[rQrRs@rrfrfs/ A8 A"~rrf)rrFconcurrent.futuresrtypingrlogururpipecat.frames.framesrrr"pipecat.processors.frame_processorr pipecat.transports.base_inputr pipecat.transports.base_outputr !pipecat.transports.base_transportr r rJModuleNotFoundErroreerror ExceptionrrrTrfrrrrzs 1UU=<>L, . .C*1C*LH 3HV*-*K,FLL;qc"#FLL I &qc* ++ ,sA//B+42B&&B+