o @i}@sZddlmZmZmZmZddlmZddlmZddl m Z ddl m Z GdddZ dS) )DictListAnyOptional)RoleplayScenario) GroqService) OllamaService) JSONHandlerc"@seZdZd#ddZdefddZ  d$d ed ededeeefdedede ede dede eeefdededede de de e f ddZ d ede de e fddZde de eeeffdd Zd ede efd!d"ZdS)%ScenarioGeneratorNcCs$|pt|_|p t|_t|_dSN)r groq_servicerollama_servicer json_handler)selfr r rM/var/www/eduai.edurigo.com/roleplay/production/services/scenario_generator.py__init__s   zScenarioGenerator.__init__use_groqcCs|r|jS|jS)z9Get the appropriate AI service based on groqRoleplay flag)r r )rrrrr_get_ai_servicesz!ScenarioGenerator._get_ai_serviceTmedium, session_idcategory objectivedetailsai_role learner_roleskills_to_assessis_adminroleplay_questions roleplay_namerole difficultyduration max_attemptsreturncs4t|dkr |dd}||||||| d}|| }||IdHs(dSgd}tfdd|Ds@td|dStjdid|d |d |d |d |d |d|ddddddddddd| d| d| p||d| d|d|}|j| ||r|StddS)zCreate a new roleplay scenarioN)rrrrrrr!)scenario_setup ai_characterscenario_introconversation_startersuccess_criteriac3s|]}|vVqdSr r).0key scenario_datarr 1sz4ScenarioGenerator.create_scenario..z*Generated scenario missing required keys: rrrrrrrr)r*r+r,r-r!r"r#r$r%r&zFailed to save scenarior) lenrgenerate_scenarioallprintr create_newr save_scenarioto_dict)rrrrrrrrr rr!r"r#r$r%r& admin_input ai_service required_keysscenariorr0rcreate_scenariosz         z!ScenarioGenerator.create_scenariocCs |j||}|rt|SdS)zLoad an existing scenarioN)r load_scenarior from_dict)rrr r1rrrr?Rs zScenarioGenerator.load_scenariocCs |j|S)zList all available scenarios)rlist_scenarios)rr rrrlist_available_scenariosYs z*ScenarioGenerator.list_available_scenarioscCs4gdgdgdgdgdd}||gdS)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) CommunicationrDProfessionalism AdaptabilityzResults Orientation)get)rrskill_mappingsrrrget_default_skills_for_category]s&z1ScenarioGenerator.get_default_skills_for_category)NN)TNrrrrr)__name__ __module__ __qualname__rboolrstrrrrintrrr>r?rBrTrrrrr sD  @r N)typingrrrrmodels.scenariorservices.groq_servicerservices.ollama_servicerutils.json_handlerr r rrrrs