o i @sTdZddlmZmZddlmZddlmZmZddl m Z m Z Gddde Z dS) zGated frame aggregator for conditional frame accumulation. This module provides a gated aggregator that accumulates frames based on custom gate open/close functions, allowing for conditional frame buffering and release in frame processing pipelines. )ListTuple)logger)Frame SystemFrame)FrameDirectionFrameProcessorcsBeZdZdZejfdeffdd Zdedeffdd ZZ S)GatedAggregatorahAccumulate frames, with custom functions to start and stop accumulation. Yields gate-opening frame before any accumulated frames, then ensuing frames until and not including the gate-closed frame. The aggregator maintains an internal gate state that controls whether frames are passed through immediately or accumulated for later release. directioncs,t||_||_||_||_g|_dS)a{Initialize the gated aggregator. Args: gate_open_fn: Function that returns True when a frame should open the gate. gate_close_fn: Function that returns True when a frame should close the gate. start_open: Whether the gate should start in the open state. direction: The frame direction this aggregator operates on. N)super__init__ _gate_open_fn_gate_close_fn _gate_open _direction _accumulator)self gate_open_fn gate_close_fn start_openr  __class__X/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/processors/aggregators/gated.pyr s  zGatedAggregator.__init__framecst||IdHt|tr|||IdHdS||jkr+|||IdHdS|j}|jr9|| |_n|||_||jkrV|jrIdnd}t d|d||jrx|||IdH|j D] \}}|||IdHqeg|_ dS|j ||fdS)zProcess incoming frames with gated accumulation logic. Args: frame: The frame to process. direction: The direction of the frame flow. Nopenclosedz Gate is now z because of ) r process_frame isinstancer push_framerrrr rdebugrappend)rrr old_statestatefdrrrr5s*     zGatedAggregator.process_frame) __name__ __module__ __qualname____doc__r DOWNSTREAMr rr __classcell__rrrrr s r N) r)typingrrlogururpipecat.frames.framesrr"pipecat.processors.frame_processorrrr rrrrs