o iF@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZGd d d eZdS) z8Vonage Audio Connector WebSocket serializer for Pipecat.N)Optional)logger) KeypadEntry)create_stream_resampler) AudioRawFrameFrameInputAudioRawFrameInputDTMFFrameInterruptionFrameOutputTransportMessageFrame!OutputTransportMessageUrgentFrame StartFrame)FrameSerializercseZdZdZGdddejZddeeffdd Zdefd d Z de d e e BdBfd d Z de e Bd e dBfddZZS)VonageFrameSerializeraSerializer for Vonage Video API Audio Connector WebSocket protocol. This serializer converts between Pipecat frames and the Vonage Audio Connector WebSocket streaming protocol. Note: Ref docs: https://developer.vonage.com/en/video/guides/audio-connector c@s.eZdZUdZdZeed<dZeeed<dS)z!VonageFrameSerializer.InputParamsa}Configuration parameters for VonageFrameSerializer. Parameters: vonage_sample_rate: Sample rate used by Vonage, defaults to 16000 Hz. Common values: 8000, 16000, 24000 Hz. sample_rate: Optional override for pipeline input sample rate. ignore_rtvi_messages: Inherited from base FrameSerializer, defaults to True. i>vonage_sample_rateN sample_rate) __name__ __module__ __qualname____doc__rint__annotations__rrrrN/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/serializers/vonage.py InputParams's  rNparamscs8t|pt|jj|_d|_t|_ t|_ dS)zkInitialize the VonageFrameSerializer. Args: params: Configuration parameters. rN) super__init__rr_paramsr_vonage_sample_rate _sample_rater_input_resampler_output_resampler)selfr __class__rrr4s   zVonageFrameSerializer.__init__framecs|jjp|j|_dS)zSets up the serializer with pipeline configuration. Args: frame: The StartFrame containing pipeline configuration. N)rraudio_in_sample_rater )r#r&rrrsetupBszVonageFrameSerializer.setupreturncst|trddi}t|St|tr2|j}|j||j|j IdH}|dus.t |dkr0dS|St|t t frF| |r@dSt|jSdS)aPSerializes a Pipecat frame to Vonage WebSocket format. Handles conversion of various frame types to Vonage WebSocket messages. Args: frame: The Pipecat frame to serialize. Returns: Serialized data as string (JSON commands) or bytes (audio), or None if the frame isn't handled. actionclearNr) isinstancer jsondumpsraudior"resamplerrlenr r should_ignore_framemessage)r#r&answerdataserialized_datarrr serializeJs"       zVonageFrameSerializer.serializer5cst|tr+|}|j||j|jIdH}|dust|dkr!dSt|d|jd}|Szt |}| d}|dkrHt d| dWdS|d krTt d WdS|d krft d | d WdS|dkr| dpw| di d}|durt d|WdSt|}t d|ztt|WWStyt d|YWdSwt d|WdStjyt d|YdSw)aDeserializes Vonage WebSocket data to Pipecat frames. Handles conversion of Vonage events to appropriate Pipecat frames. - Binary messages contain audio data (16-bit linear PCM) - Text messages contain JSON events (websocket:connected, websocket:cleared, dtmf, etc.) Args: data: The raw WebSocket data from Vonage. Returns: A Pipecat frame corresponding to the Vonage event, or None if unhandled. Nr)r/ num_channelsreventzwebsocket:connectedz)Vonage WebSocket connected: content-type=z content-typezwebsocket:clearedzVonage audio buffer clearedzwebsocket:notifyzVonage notify event: payloadzwebsocket:dtmfdigitdtmfz(DTMF event received but no digit found: zReceived DTMF digit: zInvalid DTMF digit received: zVonage event: z*Failed to parse JSON message from Vonage: )r,bytesr!r0rr r1rr-loadsgetrdebugwarningstrr r ValueErrorJSONDecodeError)r#r5r;deserialized_data audio_framer3r:r<rrr deserializensb      z!VonageFrameSerializer.deserialize)N)rrrrrrrrr r(rrCr>r7rH __classcell__rrr$rrs  "$r)rr-typingrlogururpipecat.audio.dtmf.typesrpipecat.audio.utilsrpipecat.frames.framesrrrr r r r r #pipecat.serializers.base_serializerrrrrrrs    (