o Ai.@stddlZddlZddlmZddlmZddlmZddlZGdddZe de deZ e d e ddS) N)datetime)Path)Optionalc@seZdZdZd&defddZ d'deded ed ed ed eef d dZddZ d'dedee fddZ  d(dedee dee fddZ  d(dee dee fddZ ddZ dd Zd!d"Zd#efd$d%ZdS))ConversationLoggerzHLogs conversation details including session config and message exchangeslogslogs_dircCsjt||_|jjdddd|_t|_td|jtd|j td|j dS)NT)exist_okparentsz0ConversationLogger initialized. Logs directory: zDirectory exists: zDirectory writable: ) rabsolutermkdircurrent_session threadingLocklockprintexistsis_dir)selfrrV/var/www/html/livekit_frontend/BackEnd/agent-starter-python/src/conversation_logger.py__init__ s zConversationLogger.__init__N room_name instructionsvoice_idlanguage llm_model first_messagec Cs|jC|jrtd||dttd}||||d}|r)||d<||t|gd|_td||WdS1sIwYdS) z Start a new conversation sessionzKWarning: Starting new session while one is active. Ending previous session._i)rrrrr) session_idr start_timeconfigmessagesz"Started new conversation session: N) rr r_end_session_unsafeinttimernow isoformat) rrrrrrrrr rrr start_sessions*  $z ConversationLogger.start_sessionc Cs|jsdS|jd}|j|d}zHt|jd<t|jd}t|jd}|||jd<t|ddd }tj |j|d d d Wdn1sQwYt |WSt y|}zt d |ddl }|WYd}~dSd}~ww)z@Save the current session to file immediately (internal, no lock)Nr.jsonend_timertotal_durationwutf-8encodingF)indent ensure_asciizERROR saving session: r)r rrr%r& fromisoformat total_secondsopenjsondumpstr Exceptionr traceback print_exc)rrlog_filestartendfer9rrr_save_current_session8s(  z(ConversationLogger._save_current_session transcriptdurationcCs|js tddS|jEtd||d}|jd|td|dddt|jdd |}|rItd |WddSWddS1sTwYdS) zLog a user messagez6Warning: log_user_message called but no active sessionNuser) timestamptyperArBr!zLogged user message: 2... (total messages: )Auto-saved to: ) r rrrr%r&appendlenr@)rrArBmessager;rrrlog_user_messageSs" ( "z#ConversationLogger.log_user_messageresponsellm_mscCs|js tddS|jPtd||d}|dur#t|d|d<|jd|td|dd d t|jdd | }|rTtd |WddSWddS1s_wYdS) zhLog a bot response. Optionally include LLM latency (ms). TTS/e2e can be set via update_last_bot_latency.z5Warning: log_bot_message called but no active sessionNbot)rDrErNrBrlatency_llm_msr!zLogged bot message: rFrGrHrI) r rrrr%r&roundrJrKr@)rrNrBrOrLr;rrrlog_bot_messagejs& ("z"ConversationLogger.log_bot_messagetts_mse2e_mscCs|jr|jds dS|jAt|jdD]0}|ddkrEd|vrE|dur-t|d|d<|dur8t|d|d<|WddSqWddS1sQwYdS)zAdd TTS and/or end-to-end latency (ms) to the most recent bot message that does not yet have latency_tts_ms. This avoids race with log_bot_message (thread pool): we attach to the message for this turn, not the previous one.r!NrErPlatency_tts_msrlatency_e2e_ms)r rreversedgetrRr@)rrTrUmrrrupdate_last_bot_latencys"z*ConversationLogger.update_last_bot_latencycCsb|jsdS|}|r,td|tdt|jdtd|jdddd d|_|S) z5Internal method to end session without acquiring lockNzSession ended - Final save: z - Messages: r!z - Duration: r*rz.2fs)r r@rrKrY)rr;rrrr"sz&ConversationLogger._end_session_unsafecCsD|js tddS|j |WdS1swYdS)z(End the current session and save to filez1Warning: end_session called but no active sessionN)r rrr")rrrr end_sessions $zConversationLogger.end_sessionc Csg}|jdD]?}z!t|ddd}|t|Wdn1s%wYWqtyG}ztd|d|WYd}~qd}~ww|jdd d d |S) zGet all conversation sessionsz*.jsonrr,r-NzError reading z: cSs |ddS)Nr)rY)xrrrs z5ConversationLogger.get_all_sessions..T)keyreverse) rglobr4rJr5loadr8rsort)rsessionsr;r>r?rrrget_all_sessionss z#ConversationLogger.get_all_sessionsrcCsT|j|d}|r(t|ddd }t|WdS1s#wYdS)zGet a specific session by IDr(r^r,r-N)rrr4r5re)rrr;r>rrr get_sessions  zConversationLogger.get_session)r)N)NN)__name__ __module__ __qualname____doc__r7rrr'r@floatrMrSr[r"r]rhrirrrrr sX #    rz<============================================================z"Initializing ConversationLogger...zConversationLogger ready) r5r$rpathlibrtypingrr rrloggerrrrrs   C