qi$dZddlZddlZddlmZddlmZddlZ ddl m Z ddl m Z mZmZmZddlmZddlmZdd lmZmZ ddlZ ddlZGddeZGddeZGddeZGddeZ y#e$r7Ze j6d ee j6d ed edZ[wwxYw#e$r7Ze j6d ee j6d ed edZ[wwxYw)zTkinter-based local transport implementation for Pipecat. This module provides a local transport using Tkinter for video display and PyAudio for audio I/O, suitable for desktop applications and testing. N)ThreadPoolExecutor)Optional)logger)InputAudioRawFrameOutputAudioRawFrameOutputImageRawFrame StartFrame)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: zOtkinter missing. Try `apt install python3-tk` or `brew install python-tk@3.10`.c:eZdZUdZdZeeed<dZeeed<y)TkTransportParamsaConfiguration parameters for Tkinter transport. Parameters: audio_input_device_index: PyAudio device index for audio input. If None, uses default. audio_output_device_index: PyAudio device index for audio output. If None, uses default. Naudio_input_device_indexaudio_output_device_index) __name__ __module__ __qualname____doc__rrint__annotations__rM/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/transports/local/tk.pyrr0s'/3hsm2/3x}3rrcleZdZUdZeed<dejdeffd Zde ffd Z fdZ d Z xZ S) TkInputTransportzTkinter-based audio input transport. Captures audio from the system's audio input device using PyAudio and converts it to InputAudioRawFrame objects for pipeline processing. _paramspy_audioparamscNt||||_d|_d|_y)zInitialize the Tkinter input transport. Args: py_audio: PyAudio instance for audio device management. params: Transport configuration parameters. Nr)super__init__ _py_audio _in_stream _sample_rate)selfrr __class__s rr"zTkInputTransport.__init__Es(  !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_callbackinputinput_device_index)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 rr3zTkInputTransport.startQs gmE""" ??  LL==[A[A[**S01A5 ..-->>77:\\33""( 33#||DD.  $$&&&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=zTkInputTransport.cleanuplsOgo ?? 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#TkInputTransport._audio_in_callbackts\"))77  (()>)>u)EtGZGZG\]g(())r)rrrrrrrJPyAudior"r r3r=r8 __classcell__r's@rrr<s>   :K ..6# *rrceZdZUdZeed<dejdejdeffd Z de ffd Z fd Z ded efd Zded efd Zdefd ZxZS)TkOutputTransportzTkinter-based audio and video output transport. Plays audio through PyAudio and displays video frames in a Tkinter window, providing a complete multimedia output solution for desktop applications. rtk_rootrrct||||_d|_d|_t d|_tjddz}ddd dd j|jtjjz}tjdd|d }tj|| |_|j j#y)zInitialize the Tkinter output transport. Args: tk_root: The root Tkinter window for video display. py_audio: PyAudio instance for audio device management. params: Transport configuration parameters. Nr) max_workers)rYzP5 rY  255 PPMwidthheightdatar,image)r!r"r# _out_streamr%r _executornponesencodeastypeuint8tobytestk PhotoImageLabel _image_labelpack)r&rUrrarrayrbphotor's rr"zTkOutputTransport.__init__s  !,:(3.TF!D6'..05<<3I3Q3Q3SS DDOHHWE:  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.outputoutput_device_index) r!r3reraudio_out_sample_rater%r#r5r6audio_out_channelsrr9r:)r&r(r's rr3zTkOutputTransport.starts gmE"""     LL>>]%B]B]>>..>>77:\\44"" $ F F /  %%'&&u---! # .s"C9C5CC9/C70C97C9cKt|d{|jr<|jj|jj d|_yy7Mw)z)Stop and cleanup the audio output stream.N)r!r=rer>r?r@s rr=zTkOutputTransport.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)rerIrun_in_executorrfwriterCr&r(s rwrite_audio_framez#TkOutputTransport.write_audio_framesZ   %%'77 0 0 6 6     sAA$A"A$c`K|jj|j|yw)zWrite a video frame to the Tkinter display. Args: frame: The video frame to display in the Tkinter window. Returns: True if the video frame was written successfully, False otherwise. T)rI call_soon_write_frame_to_tkr~s rwrite_video_framez#TkOutputTransport.write_video_frames* ''(?(?Gs,.c|jd}|jd}d|d|dj|jz}tj|||d}|j j |||j _y ) z,Write frame data to the Tkinter image label.rrWzP6 r\r]r^r_rcN)sizerirdrmrnrpconfig)r&r(r`rarbrss rrz$TkOutputTransport._write_frame_to_tks} 1 AUG1VHE*113ekkA E&tER   u -#(r)rrrrrrrmTkrJrPr"r r3r=rboolrrrrrQrRs@rrTrTsy !!!J[!0..0$-@T -@ T  ((; (rrTcXeZdZdZdej deffd ZdefdZ de fdZ xZ S)TkLocalTransportzComplete Tkinter-based local transport with audio and video capabilities. Provides a unified interface for local multimedia I/O using Tkinter for video display and PyAudio for audio, suitable for desktop applications and testing. rUrct|||_||_t j |_d|_d|_y)zInitialize the Tkinter local transport. Args: tk_root: The root Tkinter window for video display. params: Transport configuration parameters. N) r!r"_tk_rootrrJrP_pyaudio_input_output)r&rUrr's rr"zTkLocalTransport.__init__s:   ) 26 48 rrzc||js%t|j|j|_|jS)zGet the input frame processor for this transport. Returns: The Tkinter input transport processor. )rrrrr&s rr1zTkLocalTransport.inputs+ {{*4==$,,GDK{{rc|js0t|j|j|j|_|jS)zGet the output frame processor for this transport. Returns: The Tkinter output transport processor. )rrTrrrrs rruzTkLocalTransport.outputs1 ||,T]]DMM4<<XDL||r) rrrrrmrrr"rr1rTrurQrRs@rrrs: 9 9/@ 9&')rr)!rrFtkinterrmconcurrent.futuresrtypingrnumpyrglogururpipecat.frames.framesrrrr pipecat.transports.base_inputr pipecat.transports.base_outputr !pipecat.transports.base_transportr r rJModuleNotFoundErroreerror ExceptionrrrTrrrrrs 1 =>L,, 4 4B*)B*Jg(+g(T,},U,FLL;qc"#FLL I &qc* ++ ,,FLL;qc"#FLLbc &qc* ++,s0A8B78B4=2B//B47C3<2C..C3