o e<“hã@sFddlZddlZddlmZmZmZmZddlmZGdd„dƒZ dS)éN)ÚDictÚListÚAnyÚOptional)ÚPathc@seZdZddefdd„Zdd„Zdeeefdefd d „Z dede eeeffdd „Zdeeeeffdd„Z dedeeefdefdd„Zdede eeeffdd„Zdeeefdefdd„Zdede eeeffdd„Zdeeeeffdd„ZdS)ÚJSONHandlerÚstorageÚ base_pathcCs:t|ƒ|_|jd|_|jd|_|jd|_| ¡dS)NÚ scenariosÚ conversationsÚassessments)rr Úscenarios_pathÚconversations_pathÚassessments_pathÚ_create_directories)Úselfr ©rúS/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/prod/utils/json_handler.pyÚ__init__s zJSONHandler.__init__cCs4|jjddd|jjddd|jjddddS)z.Create storage directories if they don't existT)ÚparentsÚexist_okN)r Úmkdirrr)rrrrrszJSONHandler._create_directoriesÚ scenario_dataÚreturnc Cóšz2d|d›d}|j|}t|ddd}tj||ddd Wd ƒWdS1s+wYWdStyL}ztd|›ƒWYd }~dSd }~ww) zSave scenario to JSON fileÚ scenario_Úidú.jsonÚwúutf-8©ÚencodingéF©ÚindentÚensure_asciiNTzError saving scenario: )r ÚopenÚjsonÚdumpÚ ExceptionÚprint)rrÚfilenameÚfilepathÚfÚerrrÚ save_scenarioó ÿþ€þzJSONHandler.save_scenarioÚscenario_idc Cóšz2d|›d}|j|}| ¡sWdSt|ddd}t |¡WdƒWS1s+wYWdStyL}ztd|›ƒWYd}~dSd}~ww)zLoad scenario from JSON filerrNÚrrr zError loading scenario: )r Úexistsr&r'Úloadr)r*)rr1r+r,r-r.rrrÚ load_scenario#ó (ÿ€þzJSONHandler.load_scenarioc Cs¼g}z9|j d¡D]0}t|ddd}t |¡}| |d|d|d|dd œ¡Wd ƒn1s4wYq WntyT}z td|›ƒWYd }~nd }~wwt|dd „ddS)zList all available scenarioszscenario_*.jsonr3rr rÚcategoryÚ objectiveÚ created_at)rr8r9r:NzError listing scenarios: cSó|dS©Nr:r©ÚxrrrÚBóz,JSONHandler.list_scenarios..T©ÚkeyÚreverse) r Úglobr&r'r5Úappendr)r*Úsorted)rr r,r-Úscenarior.rrrÚlist_scenarios2s& üþ€ÿ €ÿzJSONHandler.list_scenariosÚ session_idÚconversation_datac Cs–z0d|›d}|j|}t|ddd}tj||dddWd ƒWd S1s)wYWd StyJ}ztd|›ƒWYd }~dSd }~ww)zSave conversation to JSON fileÚ conversation_rrrr r"Fr#NTzError saving conversation: )rr&r'r(r)r*)rrIrJr+r,r-r.rrrÚsave_conversationDs ÿþ€þzJSONHandler.save_conversationc Cr2)z Load conversation from JSON filerKrNr3rr zError loading conversation: )rr4r&r'r5r)r*©rrIr+r,r-r.rrrÚload_conversationQr7zJSONHandler.load_conversationÚassessment_datac Cr) zSave assessment to JSON fileÚassessment_rIrrrr r"Fr#NTzError saving assessment: )rr&r'r(r)r*)rrOr+r,r-r.rrrÚsave_assessment`r0zJSONHandler.save_assessmentc Cr2)zLoad assessment from JSON filerPrNr3rr zError loading assessment: )rr4r&r'r5r)r*rMrrrÚload_assessmentmr7zJSONHandler.load_assessmentc CsÂg}z<|j d¡D]3}t|ddd"}t |¡}| |d|d|d|d|d d œ¡Wdƒn1s7wYq WntyW}z td|›ƒWYd}~nd}~wwt|d d„ddS)z&List all assessments with summary infozassessment_*.jsonr3rr rIr1Ú overall_scoreÚperformance_levelr:)rIr1rSrTr:NzError listing assessments: cSr;r<rr=rrrr?r@z.JSONHandler.list_assessments..TrA) rrDr&r'r5rEr)r*rF)rrr,r-Ú assessmentr.rrrÚlist_assessments|s( ûþ€ÿ €ÿzJSONHandler.list_assessmentsN)r)Ú__name__Ú __module__Ú__qualname__ÚstrrrrrÚboolr/rr6rrHrLrNrQrRrVrrrrrs r) r'ÚosÚtypingrrrrÚpathlibrrrrrrÚs