o 2h]@sVddlmZmZddlmZddlmZmZddlm Z m Z m Z m Z ddl m Z ddlZddlZddlmZeddlmZdd lmZdd lmZdd lmZdd lmZdd lmZeddddZejedgddgdgdeZ eZ!ee e!Z"ee e!Z#ee e!Z$GdddeZ%GdddeZ&GdddeZ'GdddeZ(GdddeZ)Gdd d eZ*Gd!d"d"eZ+Gd#d$d$eZ,Gd%d&d&eZ-Gd'd(d(eZ.Gd)d*d*eZ/Gd+d,d,eZ0Gd-d.d.eZ1Gd/d0d0eZ2e3d1d2d3Z4e3d4d5d6Z5ej6d7e*d8d9e)fd:d;Z7d_d9e)dd?Z9d@e:dAe:d=e:fdBdCZ;dDe:dEedHed=e e e:e:ffdKdLZ?deZdZUeed<eed<dZeeeefed<e ed<dS)EndSessionResponsemessagerENr@rK) r r!r"r#r$r@r rr r=r%r%r%r&rbrs  rbc@eZdZUeed<eed<dS)EndSessionRequestrErFNrHr%r%r%r&rexr'rec@rd)CleanupRequestrEis_adminN)r r!r"r#r$r6r%r%r%r&rf|r'rfc@s.eZdZUeed<eed<eeefed<dS)CleanupResponsercrEcleanup_statusN)r r!r"r#r$rboolr%r%r%r&rhs rh/cs dddS)zHealth check endpointrrunning)rcstatusr%r%r%r%r&roots rnz/healthc szr9r:r;)r9r:r;r?r@rrru)r>r?r@rA) groq_servicerget_token_countsr=r8)r ai_service token_datar%r%r&rs,          rz/roleplay_scenarioc sz[|jjdk}td|jtd|rdnddt|j|jj}|r9tdt|}t|}t||dWStd t ||}|sJt d d d t|}t|}td t||dWSt ydt y}ztdt |ddl }|t d dt |d d}~ww)z9Generate and return formatted preview data as slide arrayrz(Processing preview request for session: rrrz for preview generationz#Using existing scenario for preview)rarKz!Creating new scenario for previewrwz%Failed to create scenario for previewrxz%Preview slides generated successfullyz(Error in get_roleplay_scenario_preview: rNzError generating preview: )rFr5rrErr4rrr`rrrr#rr)rrexisting_scenarioslides_responserKrrrr%r%r&get_roleplay_scenario_previews8    rz /end_sessionc sz|jjdk}td|jd|jjtd|rdnddt|j|jj}|s1tdd d t|j|jj}|sBtdd d t |j|jj}|sStd d d td|jt |j|||jj|}td|dut |}td||rt nt}|td|rtdtj|j|jj}td|d|j|d} |r|| d<tdi| WStyty} z td dt| d d} ~ ww)zBEnd a roleplay session and automatically generate skill assessmentrzDEBUG end_session: session_id='z', roleplay_data.isAdmin=rrrz for assessment generationzSession/Scenario not foundrxzNo conversation history foundrwzFailed to end sessionz0DEBUG: About to generate assessment for session zDEBUG: Assessment generated: Nz(DEBUG: Final token counts before reset: z3DEBUG: Token counts reset to zero after end_sessionz8DEBUG: Deleting conversation after successful assessmentzDEBUG: Conversation deleted: zSession ended successfully)rcrErKr@zError ending session: r%)rFr5rrEr4rrr%get_conversation_turns_for_assessment end_sessionskill_analyzeranalyze_sessionrrrreset_token_counts json_handlerdelete_conversationto_dictrbrr#) rrrconversation_turnssuccessr@final_token_countsrconversation_deleted response_datarr%r%r&rsN       rz/cleanupc sztd|jdddd}ztj|j|j}||d<td|Wnty?}z td|WYd}~nd}~wwztj|j|j}||d<td |Wntyo}z td |WYd}~nd}~wwzt j |j|j}||d <td |Wnty}z td |WYd}~nd}~wwtd|jt d|j|dWSty}ztdt |t ddt |dd}~ww)zKDelete all session-related data after assessment has been saved to databasez,DEBUG: Cleaning up session data for session F)scenario_deletedassessment_deletedrrz!DEBUG: Scenario deletion result: zError deleting scenario: Nrz#DEBUG: Assessment deletion result: zError deleting assessment: rz%DEBUG: Conversation deletion result: zError deleting conversation: zCLEANUP COMPLETED - Session: z$Session data cleaned up successfully)rcrErizError in cleanup_session_data: rwz Error cleaning up session data: rx)rrErrdelete_scenariorgrrdelete_assessmentrrrhr#r)rrirrrrr%r%r&cleanup_session_datasTrz/session/{session_id}/historyc s^zt||}|durtddd||dWSty.}z tddt|dd}~ww)z&Get conversation history for a sessionNrzSession not foundrx)rErrwzError retrieving history: )rrrrr#)rErghistoryrr%r%r&get_session_history@s   r__main__z0.0.0.0i@)hostport)T)r)Hfastapirrfastapi.middleware.corsrpydanticrrtypingrrr r r uuidr}dotenvr services.scenario_generatorr services.roleplay_enginerservices.groq_servicerservices.ollama_servicerservices.skill_analyzerrmodels.scenariorappadd_middlewarerrrrrrr(r8r=rCrIrLrRrVr`rbrerfrhrrnrpostrrjrr#rr6rrrrrrrrr uvicornrunr%r%r%r&s                 X' /!  % ;0