@i}`SSKJrJrJrJr SSKJr SSKJr SSK J r SSK J r "SS5r g) )DictListAnyOptional)RoleplayScenario) GroqService) OllamaService) JSONHandlerc"\rSrSrSSjrS\4SjrSS\S\S\S \\\ 4S \S \S \ \S \ S\S\ \\\4S\S\S\S\ S\ S\ \ 4 SjjrS\S \ S\ \ 4SjrS \ S\ \\\ 44SjrS\S\ \4SjrSrg)ScenarioGeneratorNcU=(d [5UlU=(d [5Ul[ 5UlgN)r groq_servicer ollama_servicer json_handler)selfrrs M/var/www/eduai.edurigo.com/roleplay/production/services/scenario_generator.py__init__ScenarioGenerator.__init__s*(9KM,? 'Muse_groqc@U(a UR$UR$)z9Get the appropriate AI service based on groqRoleplay flag)rr)rrs r_get_ai_service!ScenarioGenerator._get_ai_services$,t  E$2E2EEr session_idcategory objectivedetailsai_role learner_roleskills_to_assessis_adminroleplay_questions roleplay_namerole difficultyduration max_attemptsreturncH^# [U5S:aUSSnUUUUUUU S.nURU 5nURU5IShvN mT(dg/SQn[U4SjU55(d[ SU35 g[ R "S0SU_SU_S U_S U_S U_S U_S U_STS_STS_STS_STS_STS_SU _SU _SU =(d U_SU _SU_SU_6nURRUR5X5(aU$[ S5 gN7f)zCreate a new roleplay scenarioN)rrrr r!r"r$)scenario_setup ai_characterscenario_introconversation_startersuccess_criteriac3,># UH oT;v M g7fr).0key scenario_datas r 4ScenarioGenerator.create_scenario..1sA=C-'=sz*Generated scenario missing required keys: rrrrr r!r"r-r.r/r0r1r$r%r&r'r(r)zFailed to save scenarior3) lenrgenerate_scenarioallprintr create_newr save_scenarioto_dict)rrrrrr r!r"r#rr$r%r&r'r(r) admin_input ai_service required_keysscenarior6s @rcreate_scenario!ScenarioGenerator.create_scenarios  1 $/3 !"( 0"4 ))(3 (::;GG y A=AAA >}oN O$.. !        &  . ))9: '~6 ))9: "//E!F ++=>  2 ( % "! "# $&% ,    * *8+;+;+=x T TO + ,MHsAD"D CD"ctURRX5nU(a[R"U5$g)zLoad an existing scenarioN)r load_scenarior from_dict)rrr#r6s rrGScenarioGenerator.load_scenarioRs/))77 M #--m< <rc8URRU5$)zList all available scenarios)rlist_scenarios)rr#s rlist_available_scenarios*ScenarioGenerator.list_available_scenariosYs  //99rcN/SQ/SQ/SQ/SQ/SQS.nURU/SQ5$)z'Get default skills for a given category)zPolicy AdherenceEmpathyzCommunication ClarityzConflict ResolutionProblem Solving)zNeeds AssessmentzProduct Knowledge PersuasionzObjection HandlingzRelationship Building)zDecision MakingzTeam Communication Motivation DelegationzConflict Management) PreparationzActive ListeningzCreative Problem Solving CompromisezRelationship Management)zProblem DiagnosiszTechnical CommunicationPatience Documentationz Follow-up)zCustomer ServiceSales Leadership NegotiationzTechnical Support) CommunicationrPProfessionalism AdaptabilityzResults Orientation)get)rrskill_mappingss rget_default_skills_for_category1ScenarioGenerator.get_default_skills_for_category]s?!";$ L!!(-   r)rrr)NN)TNrbmediumi,)__name__ __module__ __qualname____firstlineno__rboolrstrrrrintrrrDrGrLr`__static_attributes__r3rrr r s0* FF Z^RZ@A >>s>s>]abegjbj]k> #>36>JNs)>_b>!%>BFtCQTH~BV>'*>7:>MP>"% >;> >GOO_F` >@sxHX?Y::d38n9M:. . S . rr N)typingrrrrmodels.scenariorservices.groq_servicerservices.ollama_servicer utils.json_handlerr r r3rrrrs#,,,-1*D D r