o i2 @sXdZddlZddlmZmZmZmZddlmZm Z ddl m Z m Z Gddde Z dS)z:Idle frame processor for timeout-based callback execution.N) AwaitableCallableListOptional)Frame StartFrame)FrameDirectionFrameProcessorc sxeZdZdZdddedgedfdedeee ffddZ d e d e ffd d Z d dZddZddZZS)IdleFrameProcessoraMonitors frame activity and triggers callbacks on timeout. This processor waits to receive any frame or specific frame types within a given timeout period. If the timeout is reached before receiving the expected frames, the provided callback will be executed. N)typescallbacktimeoutr c s2tjdi|||_||_|pg|_d|_dS)aInitialize the idle frame processor. Args: callback: Async callback function to execute on timeout. Receives this processor instance as an argument. timeout: Timeout duration in seconds before triggering the callback. types: Optional list of frame types to monitor. If None, monitors all frames. **kwargs: Additional arguments passed to parent class. N)super__init__ _callback_timeout_types _idle_task)selfr r r kwargs __class__r[/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/processors/idle_frame_processor.pyrs   zIdleFrameProcessor.__init__frame directioncsrt||IdHt|tr||||IdH|js'|jdS|jD] }t||r6|jq*dS)zProcess incoming frames and manage idle timeout monitoring. Args: frame: The frame to process. direction: The direction of frame flow in the pipeline. N) r process_frame isinstancer_create_idle_task push_framer _idle_eventset)rrrtrrrr1s    z IdleFrameProcessor.process_framecs"|jr||jIdHdSdS)z,Clean up resources and cancel pending tasks.N)r cancel_taskrrrrcleanupHszIdleFrameProcessor.cleanupcCs(|jst|_|||_dSdS)z*Create and start the idle monitoring task.N)rasyncioEventr create_task_idle_task_handlerr$rrrrMs z$IdleFrameProcessor._create_idle_taskcsh z*ztj|j|jdIdHWntjy%||IdHYnwW|jn|jwq)z6Handle idle timeout monitoring and callback execution.T)r N)r&wait_forr waitr TimeoutErrorrclearr$rrrr)Ss z%IdleFrameProcessor._idle_task_handler)__name__ __module__ __qualname____doc__rrfloatrrtyperrrrr%rr) __classcell__rrrrr s  r )r1r&typingrrrrpipecat.frames.framesrr"pipecat.processors.frame_processorrr r rrrrs