o GiH6@sUdZddlZddlZddlZddlZddlmZddlmZmZddl m Z m Z ddl Z ej ej dedZedZd ZiZeeefed <iZeeefed <gZeed <d ddddiidZiZeeefed<GdddZde jde jfddZde jde jfddZ de jde jfddZ!de jde jfddZ"de jde jfd d!Z#de j$fd"d#Z%e&d$kre'ed%d&Z(e)d'e(e%Z*e j+e*d(e(d)dSdS)*zh ElevenLabs ConvAI Server Provides WebSocket endpoint for real-time voice conversations with ElevenLabs N)datetime)DictSet)web WSMsgType)levelelevenlabs_serverELEVENLABS_API_KEYz#https://api.elevenlabs.io/v1/convaiactive_sessionssession_metricsmetrics_historygQ?gQ?)eleven_monolingualeleven_multilingualultrag{Gz?)ttssttllm session_costsc@s.eZdZdZddZdejdejfddZdS) ElevenLabsConvAIHandlerz6Handles WebSocket connections to ElevenLabs ConvAI APIcCst|_d|_dS)Nr)setactive_connections session_count)selfr^/var/www/html/vikas/2025-Nov-21-Voicebot/BackEnd/agent-starter-python/src/elevenlabs_server.py__init__1s z ElevenLabsConvAIHandler.__init__requestreturnc sZt|IdHd|j|jd7_tddt<dddddddddt<|j t dt d t}zz_t4IdHJ}||4IdH+d td <fd d }fdd}t||IdHWdIdHn 1IdHswYWdIdHn 1IdHswYWn(ty}zt d|dtd <t|td<WYd}~nd}~wwW|j tvrdtd <ttd<titd<t dS|j tvr$dtd <ttd<titd<t dw)z.forward_to_elevenlabscs2z3dHW}|jtjkrzt|j}tvrgt<tt | dd|d| ddks.forward_to_clientzWebSocket error: error disconnectedended_atcostszWebSocket closed: )rWebSocketResponsepreparerrrCrDr rraddloggerinfoELEVENLABS_CONVAI_URLr aiohttp ClientSession ws_connectasynciogather ExceptionrLstrdiscardrE)rrelevenlabs_ws_urlsessionr7rKerrIrhandle_websocket5sj      9 *(O     z(ElevenLabsConvAIHandler.handle_websocketN) __name__ __module__ __qualname____doc__rrRequestrPrarrrrr.srrrcstdtdS)zHealth check endpointok)r"r9)r json_responserrCrDrrrr health_checks  rjc sJt}d}d}tD]\}}|D]}d|divr)||dd7}|d7}qq |dkr3||nd}d}d} d} d} tD]$\}} || dd7}| | dd7} | | dd7} | | dd7} qAi} tD]8\}} t| ddd t| ddd t| ddd t| d dd | dd| d d| d dd | |<qltddt Dtt| t|d| t|d t| d t| d t|| | d dd| dkrt|| d nd| dkrt| | d nd| dkrt| | d nd| dkrt|| | | d ndddt| d}t |tt dkr t dt|S)zEMetrics endpoint showing active sessions, latency, and cost breakdownrlatencyr.rr#r$r%r'r&r(r))r#r$r%r&r'r(r)cSsg|] }|ddkr|qS)r"r+)rE).0srrr sz#metrics_handler..USD)rrrtotalcurrency)rrrrr)rrper_turn_average)r total_sessions total_turnsaverage_end_to_end_latency_msr9cost_breakdownsessionsrd)rrCr itemsrErroundlenr valuesrDr rBpoprrh)r current_time total_latency latency_countrJmetricsmetric avg_latencytotal_stt_costtotal_tts_costtotal_llm_costrv cost_datasession_cost_breakdown metrics_datarrrmetrics_handlersn     "   rc std}tr dtini}z[t4IdHE}|j||d4IdH"}|IdH}t|WdIdHWdIdHWS1IdHsKwYWdIdHWdS1IdHsbwYWdSty}zt d|tjdt |iddWYd}~Sd}~ww) z#List available LLMs from ElevenLabs /llm/list xi-api-keyN)headerszError fetching LLM list: rLr") rUr rVrWrEr@rrhr[rSrLr\)rurlrr_responser.r`rrr list_llmss" 2"rc s@z |IdH}Wni}Ytd}trdtini}z\t4IdHF}|j|||d4IdH"}|IdH}t|WdIdHWdIdHWS1IdHs\wYWdIdHWdS1IdHsswYWdSty}zt d|tjdt |iddWYd}~Sd}~ww) zCalculate LLM usage/costN/llm-usage/calculater)r@rzError calculating LLM usage: rLrr) r@rUr rVrWpostrrhr[rSrLr\)rr.rrr_rresultr`rrrcalculate_llm_usages* 2"rc s"tddddddddd d S) z!Index handler showing server infozElevenLabs ConvAI Serverz1.0.0/ws//health/metricsrr) websockethealthrrcalculate_usagerunning)nameversion endpointsr")rrhrirrr index_handler(srcCsht}t}|jdt|jdt|jdt|jdt|j dt |jd|j |S)z,Create and configure the aiohttp application/rrrrr) r Applicationrrouteradd_getrrjrradd_postrra)apphandlerrrr create_app8sr__main__PORT8080z*Starting ElevenLabs ConvAI Server on port z0.0.0.0)hostport),rerYr@osloggingrtypingrrrVrr basicConfigINFO getLoggerrSgetenvr rUr r\dict__annotations__r listr rFrrrfResponserjrrrrrrrbintrrTrrun_apprrrrsL    J