qi#dZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZmZddlmZGd d eZy) z5Plivo WebSocket frame serializer for audio streaming.N)Optional)logger) KeypadEntry)create_stream_resampler pcm_to_ulaw ulaw_to_pcm) AudioRawFrame CancelFrameEndFrameFrameInputAudioRawFrameInputDTMFFrameInterruptionFrameOutputTransportMessageFrame!OutputTransportMessageUrgentFrame StartFrame)FrameSerializerc eZdZdZGddej Z ddedeedeedeed eef fd Zd e fd Z d e d ee zdzfdZ dZdee zd e dzfdZxZS)PlivoFrameSerializeraSerializer for Plivo Audio Streaming WebSocket protocol. This serializer handles converting between Pipecat frames and Plivo's WebSocket audio streaming protocol. It supports audio conversion, DTMF events, and automatic call termination. When auto_hang_up is enabled (default), the serializer will automatically terminate the Plivo call when an EndFrame or CancelFrame is processed, but requires Plivo credentials to be provided. cBeZdZUdZdZeed<dZeeed<dZ e ed<y) PlivoFrameSerializer.InputParamsaConfiguration parameters for PlivoFrameSerializer. Parameters: plivo_sample_rate: Sample rate used by Plivo, defaults to 8000 Hz. sample_rate: Optional override for pipeline input sample rate. auto_hang_up: Whether to automatically terminate call on EndFrame. ignore_rtvi_messages: Inherited from base FrameSerializer, defaults to True. i@plivo_sample_rateN sample_rateT auto_hang_up) __name__ __module__ __qualname____doc__rint__annotations__rrrboolK/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/serializers/plivo.py InputParamsr,s, "&3%%) Xc])! d!r#r%N stream_idcall_idauth_id auth_tokenparamsct||xstj||_||_||_||_|jj|_ d|_ t|_ t|_d|_y)ayInitialize the PlivoFrameSerializer. Args: stream_id: The Plivo Stream ID. call_id: The associated Plivo Call ID (optional, but required for auto hang-up). auth_id: Plivo auth ID (required for auto hang-up). auth_token: Plivo auth token (required for auto hang-up). params: Configuration parameters. rFN)super__init__rr% _stream_id_call_id_auth_id _auth_token_paramsr_plivo_sample_rate _sample_rater_input_resampler_output_resampler_hangup_attempted)selfr&r'r(r)r* __class__s r$r-zPlivoFrameSerializer.__init__:sv" E#7#C#C#EF#  %"&,,"@"@ 7 9!8!:!&r#framec^K|jjxs |j|_yw)zSets up the serializer with pipeline configuration. Args: frame: The StartFrame containing pipeline configuration. N)r2raudio_in_sample_rater4)r8r:s r$setupzPlivoFrameSerializer.setupYs% !LL44R8R8Rs+-returnc,K|jjrB|js6t|tt fr d|_|j d{yt|tr$d|jd}tj|St|tr|j}t||j|j|j d{}|t#|dk(ryt%j&|j)d}dd|j|d |jd }tj|St|t*t,fr1|j/|rytj|j0Sy7:7w) aSerializes a Pipecat frame to Plivo WebSocket format. Handles conversion of various frame types to Plivo WebSocket messages. For EndFrames, initiates call termination if auto_hang_up is enabled. Args: frame: The Pipecat frame to serialize. Returns: Serialized data as string or bytes, or None if the frame isn't handled. TN clearAudio)eventstreamIdrzutf-8 playAudioz audio/x-mulaw) contentType sampleRatepayload)rAmediarB)r2rr7 isinstancer r _hang_up_callrr.jsondumpsr audiorrr3r6lenbase64 b64encodedecoderrshould_ignore_framemessage)r8r:answerdataserialized_datarFs r$ serializezPlivoFrameSerializer.serializeasa LL % %**58["9:%)D "$$& & & 0 1+IF::f% % } -;;D%0e'')@)@$BXBX%O&#o*>!*C&&7>>wGG$#2"&"9"9& !OOF::f% %  ;=^_ `''.::emm, ,E 's&AFFBFFB5FFc K ddl}|j}|j}|j}|r|r|scg}|s|j d|s|j d|s|j dt j ddj|yd|d |d }|j||}|j4d{}|j|| 4d{} | jd k(rt jd |ni| jdk(rt jd|dn@| jd{} t jd|d| jd| dddd{dddd{y777Q7#1d{7swY+xYw7"#1d{7swYyxYw#t$r"} t jd| Yd} ~ yd} ~ wwxYww)z.Hang up the Plivo call using Plivo's REST API.rNr'r(r)z8Cannot hang up Plivo call: missing required parameters: z, z!https://api.plivo.com/v1/Account/z/Call//)authz#Successfully terminated Plivo call iz Plivo call z already terminatedzFailed to terminate Plivo call z : Status z , Response: zFailed to hang up Plivo call: )aiohttpr0r1r/appendrwarningjoin BasicAuth ClientSessiondeletestatusdebugtexterror Exception) r8r[r(r)r'missingendpointrYsessionresponse error_textes r$rIz"PlivoFrameSerializer._hang_up_calls+ ? mmG))JmmG'NN9-NN9-!NN<0NtyyY`OaNbc;7)6'RSTH$$Wj9D,,.  '">>(>>  (#- 'J7)%TU!C/ {7);N%OP,4==?%:  =gYG&&.oo%6l:,P      &;         ? LL9!= > > ?sG9BG G9/G FG F6 F!F6$A%F F ,F6 F6FF6 G F4G G9G F6FF6F1 %F(&F1 -F64G 6G<F? =GG G9G G6G1,G91G66G9rTcK tj|}|j ddk(r|j di}|j d}|syt j|}t||j|j|jd{}|t|dk(ryt|d|j}|S|j dd k(r;|j d i}|j d } | r tt| Syy#tj$rtjd|YywxYw7#t $rtjd | YywxYww) a8Deserializes Plivo WebSocket data to Pipecat frames. Handles conversion of Plivo media events to appropriate Pipecat frames. Args: data: The raw WebSocket data from Plivo. Returns: A Pipecat frame corresponding to the Plivo event, or None if unhandled. zFailed to parse JSON message: NrArGrFr)rL num_channelsrdtmfdigitzInvalid DTMF digit received: )rJloadsJSONDecodeErrorrr]getrN b64decoderr3r4r5rMr rr ValueError) r8rTrRrGpayload_base64rFdeserialized_data audio_frame dtmf_datarqs r$ deserializez PlivoFrameSerializer.deserializesp jj&G ;;w 7 *KK,E"YYy1N!&&~6G'200$2C2CTEZEZ'! !(C0A,Ba,G,'aTEVEVK  [[ !V + FB/IMM'*E )+e*<==I##  NN;D6B C !"" NN%B5'#JK sYE1DA>E1EA(E1E E1+EE1EE1 !E.+E1-E..E1)NNNN)rrrrrr%strrr-rr=r bytesrVrIr{ __classcell__)r9s@r$rr s  "o11 """&!%$((, ''#'# ' SM ' % '>SS4U4sU{T/A4l-?^1cEk1edl1r#r)rrNrJtypingrlogururpipecat.audio.dtmf.typesrpipecat.audio.utilsrrrpipecat.frames.framesr r r r r rrrrr#pipecat.serializers.base_serializerrrr"r#r$rsB< 0QQ   @W?Wr#