o BW:i@sdZddlmZmZmZddlmZmZddlm Z ddl m Z ddl m Z GdddeZGd d d eZGd d d ZeZd S)z AI Service Factory for ElevenLabs WebSocket Service (Service Type 4) Simplified to only support service type 4 for better performance )OptionalDictAny)ABCabstractmethod)Config)Log)ElevenLabsWebSocketServicec @szeZdZdZeddedededeefddZed edee fd d Z ede fd d Z ede eeffddZdS)AIServiceInterfacezm Abstract interface for AI services. Ensures consistent interface across different AI providers. N user_input system_promptconversation_historyreturncdS)z'Generate text response from user input.Nselfr r r rrM/home/vmc/milan/call_management/calls/homebook/services/ai_service_factory.pygenerate_responsez$AIServiceInterface.generate_responsetextcr)zConvert text to speech audio.Nrrrrrrtext_to_speechrz!AIServiceInterface.text_to_speechcr)z!Handle conversation interruption.Nrrrrrhandle_interruptionrz&AIServiceInterface.handle_interruptioncCsdS)zGet service information.Nrrrrrget_service_info#sz#AIServiceInterface.get_service_infoN)__name__ __module__ __qualname____doc__rstrlistrrbytesrboolrrrrrrrrr s"r c @seZdZdZdeeeffddZddededed e efd d Z d ed e e fd dZ d e fddZd eeeffddZd efddZdS)ElevenLabsWebSocketWrapperz Wrapper for ElevenLabs WebSocket service to implement AIServiceInterface. Note: ElevenLabs WebSocket works with real-time audio streaming, not text-based methods. bot_configcCsD|r|dnd}|sttdrtjnd}t|d|_tddS)z Initialize ElevenLabs WebSocket wrapper. Args: bot_config: Bot configuration dictionary (may contain agent_id) elevenlabs_agent_idNELEVENLABS_AGENT_ID)agent_idu0🎙️ ElevenLabs WebSocket wrapper initialized)gethasattrrr(r elevenlabs_servicerinfo)rr&r)rrr__init__2s  z#ElevenLabsWebSocketWrapper.__init__Nr r r rctddS)z Generate response using ElevenLabs WebSocket. Note: This is a placeholder - ElevenLabs WebSocket works with real-time audio. T🎙️ ElevenLabs WebSocket doesn't use this method - it works with real-time audioNrdebugrrrrrA z,ElevenLabsWebSocketWrapper.generate_responsercr/)z Text-to-speech using ElevenLabs WebSocket. Note: This is a placeholder - ElevenLabs WebSocket works with real-time audio. r0Nr1rrrrrIr3z)ElevenLabsWebSocketWrapper.text_to_speechc sz+t|dr |jjstdWdS|jIdH}|r%td|WStd|WStyR}ztd|ddl }t| WYd}~dSd}~ww) z Handle conversation interruption by sending an interrupt signal to ElevenLabs. Returns: bool: True if interruption was handled successfully r,u;⚠️ Cannot handle interruption - WebSocket not connectedFNu0✅ Successfully sent interruption to ElevenLabsu0⚠️ Failed to send interruption to ElevenLabsu"❌ Error in handle_interruption: r) r+r, connectedrwarningsend_interruptionr- Exceptionerror traceback format_exc)rsuccesser9rrrrXs$   z.ElevenLabsWebSocketWrapper.handle_interruptioncCsdddddS)z* Get service information. Zelevenlabs_websocketT 200-500ms) service_typeZsupports_interruptionlatencyZ real_timerrrrrrss z+ElevenLabsWebSocketWrapper.get_service_infocCs|jS)z%Get the underlying WebSocket service.)r,rrrrget_websocket_service~sz0ElevenLabsWebSocketWrapper.get_websocket_servicer)rrrr rr!rr.r"rrr#rr$rrr r@rrrrr%,s  r%c@sfeZdZdZededeeefde fddZ ededefddZ ededeeeffd d Z d S) AIServiceFactoryzG Factory class for creating AI services based on service_type. r>r&rc Csj|dkrd|}td|t|z tdt|WSty4}z td|d}~ww)a Create AI service - only service type 4 (ElevenLabs WebSocket) is supported. Args: service_type: Service type (must be 4) bot_config: Bot configuration dictionary Returns: AI service instance (ElevenLabsWebSocketWrapper) Raises: ValueError: If service_type is not 4 z>Only service_type 4 (ElevenLabs WebSocket) is supported. Got: u❌ u-🎙️ Creating ElevenLabs WebSocket serviceu1❌ Error creating ElevenLabs WebSocket service: N)rr8 ValueErrorr-r%r7)r>r& error_msgr<rrrcreate_ai_services   z"AIServiceFactory.create_ai_servicecCs|dkrdSd|dS)z Get human-readable service name. Args: service_type: Service type number (must be 4) Returns: Service name string rBzElevenLabs WebSocketzUnsupported Service (z$) - Only service type 4 is supportedrr>rrrget_service_names  z!AIServiceFactory.get_service_namecCs|dkr ddddddSiS)z Get service capabilities information. Args: service_type: Service type number (must be 4) Returns: Dictionary with capabilities rBTr=ZMediumZ Excellent)Zreal_time_interruptionr?costqualityZsupports_streamingrrFrrrget_service_capabilitiess z)AIServiceFactory.get_service_capabilitiesN) rrrr staticmethodintrr!rr rErGrJrrrrrAs  rAN)r typingrrrabcrrconfigrservices.log_utilsr%services.elevenlabs_websocket_servicer r r%rAZai_service_factoryrrrrs   W F