o lWi#@sbddlZddlZddlZddlZddlmZejrddlmZGddde eZ GdddZ dS)N)Enum)ClientConnectionc@s\eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdS)RealtimeEventsz(Events emitted by the RealtimeConnectionopenclosesession_startedpartial_transcriptcommitted_transcript$committed_transcript_with_timestampserror auth_errorquota_exceededcommit_throttledtranscriber_errorunaccepted_terms_error rate_limited input_errorqueue_overflowresource_exhaustedsession_time_limit_exceededchunk_size_exceededinsufficient_audio_activityN)__name__ __module__ __qualname____doc__OPENCLOSESESSION_STARTEDPARTIAL_TRANSCRIPTCOMMITTED_TRANSCRIPT$COMMITTED_TRANSCRIPT_WITH_TIMESTAMPSERROR AUTH_ERRORQUOTA_EXCEEDEDCOMMIT_THROTTLEDTRANSCRIBER_ERRORUNACCEPTED_TERMS_ERROR RATE_LIMITED INPUT_ERRORQUEUE_OVERFLOWRESOURCE_EXHAUSTEDSESSION_TIME_LIMIT_EXCEEDEDCHUNK_SIZE_EXCEEDEDINSUFFICIENT_AUDIO_ACTIVITYr/r/b/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/elevenlabs/realtime/connection.pyr s*rc@seZdZdZddddedejejfddZ d e d ej d dfd d Z d e d dfddZ dddZdeje ejfd dfddZdddZdddZdddZdS)RealtimeConnectiona  A WebSocket connection for real-time speech-to-text transcription. This class handles bidirectional WebSocket communication with the ElevenLabs speech-to-text API, managing audio streaming and receiving transcription results. Example: ```python connection = await client.speech_to_text.realtime.connect({ "audio_format": AudioFormat.PCM_16000, "sample_rate": 16000 }) connection.on(RealtimeEvents.PARTIAL_TRANSCRIPT, lambda data: print(data)) connection.on(RealtimeEvents.COMMITTED_TRANSCRIPT, lambda data: print(data)) # Send audio connection.send({"audioBase64": audio_chunk}) # When done connection.commit() await connection.close() ``` N websocketrcurrent_sample_rateffmpeg_processcCs"||_||_||_i|_d|_dSN)r2r3r4_event_handlers _message_task)selfr2r3r4r/r/r0__init__=s  zRealtimeConnection.__init__eventcallbackreturncCs(||jvr g|j|<|j||dS)a Register an event handler for a specific event type. Args: event: The event type to listen for (from RealtimeEvents enum) callback: The function to call when the event occurs Example: ```python def handle_transcript(data): print(f"Transcript: {data['transcript']}") connection.on(RealtimeEvents.PARTIAL_TRANSCRIPT, handle_transcript) ``` N)r6append)r8r:r;r/r/r0onDs  zRealtimeConnection.onc Gsf||jvr/|j|D]&}z||Wq ty.}ztd|d|WYd}~q d}~wwdSdS)z(Emit an event to all registered handlerszError in event handler for z: N)r6 Exceptionprint)r8r:argshandlerer/r/r0_emitXs   zRealtimeConnection._emitc sXzzs|j2zl3dHW}zGt|}|d}z1t|}|||tjtjtjtj tj tj tj tj tjtjtjtjh }||vrG|tj|Wn tyQYnwWqtjyr}z|tjdd|iWYd}~qd}~ww6Wnty}z|tjdt|iWYd}~nd}~wwW|tjdSW|tjdS|tjw)z*Start handling incoming WebSocket messagesN message_typer zFailed to parse message: )r2jsonloadsgetrrDr#r$r%r&r'r(r)r*r+r,r-r.r" ValueErrorJSONDecodeErrorr?strr)r8messagedatarEr: error_eventsrCr/r/r0_start_message_handlerasT    $ "#z)RealtimeConnection._start_message_handlerrMcsL|jstdd|ddd|j|dd}|jt|IdHdS) a Send an audio chunk to the server for transcription. Args: data: Dictionary containing the following keys: - audio_base_64 (str): Base64-encoded audio data to transcribe - previous_text (str, optional): Previous transcript text to provide context for more accurate transcription Raises: RuntimeError: If the WebSocket connection is not open Example: ```python # Send audio chunk connection.send({ "audio_base_64": base64_encoded_audio }) # Send audio chunk with context - can only be sent with the first chunk of audio connection.send({ "audio_base_64": base64_encoded_audio, "previous_text": "Previously transcribed text for context" }) ``` WebSocket is not connectedinput_audio_chunk audio_base_64F previous_text)rErRcommit sample_raterTN)r2 RuntimeErrorrHr3sendrFdumps)r8rMrLr/r/r0rXs zRealtimeConnection.sendcs<|jstdddd|jd}|jt|IdHdS)a Commits the segment, triggering a COMMITTED_TRANSCRIPT event and clearing the buffer. It's recommend to commit often when using CommitStrategy.MANUAL to keep latency low. Raises: RuntimeError: If the WebSocket connection is not open Remarks: Only needed when using CommitStrategy.MANUAL. When using CommitStrategy.VAD, commits are handled automatically by the server. Example: ```python # Send all audio chunks for chunk in audio_chunks: connection.send({"audioBase64": chunk}) # Commit the audio segment await connection.commit() ``` rPrQrST)rErRrUrVN)r2rWr3rXrFrY)r8rLr/r/r0rUszRealtimeConnection.commitcsv|IdH|jr|jddIdH|jr7|js9|jz |jIdHWdStjy6YdSwdSdS)aG Closes the WebSocket connection and cleans up resources. This will terminate any ongoing transcription and stop ffmpeg processes if running. Remarks: After calling close(), this connection cannot be reused. Create a new connection if you need to start transcribing again. Example: ```python connection.on(RealtimeEvents.COMMITTED_TRANSCRIPT, async lambda data: ( print("Committed:", data["transcript"]), await connection.close() )) ``` NizUser ended conversation)_cleanupr2rr7donecancelasyncioCancelledErrorr8r/r/r0rs zRealtimeConnection.closecsR|jr'|jz |jjddWntjy!|jYnwd|_dSdS)z(Clean up resources like ffmpeg processes)timeoutN)r4killwait subprocessTimeoutExpiredr_r/r/r0rZs  zRealtimeConnection._cleanupr5)r<N)rrrrinttypingOptionalrdPopenr9rKCallabler>rDrODictAnyrXrUrrZr/r/r/r0r1#s  ( ( "r1) r]rFrdrgenumr TYPE_CHECKINGwebsockets.asyncio.clientrrKrr1r/r/r/r0s