o 0hn@sNddlmZmZmZmZddlmZddlmZddl m Z GdddZ dS))DictListAnyOptional)RoleplayScenario) GroqService) JSONHandlerc@seZdZddZdededeeefdededeed ee fd d Z d ed ee fd dZ d eeeeffddZ ded eefddZ dS)ScenarioGeneratorcCst|_t|_dSN)r groq_servicer json_handlerselfr_/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/testing/services/scenario_generator.py__init__s zScenarioGenerator.__init__category objectivedetailsai_role learner_roleskills_to_assessreturnc st|dkr |dd}||||||d}|j|sdSgd}tfdd|Ds7td|dStj||||||dd d d d d } |j| r[| StddS)zCreate a new roleplay scenarioN)rrrrrr)scenario_setup ai_characterscenario_introconversation_startersuccess_criteriac3s|]}|vVqdSr r).0key scenario_datarr %sz4ScenarioGenerator.create_scenario..z*Generated scenario missing required keys: rrrrr) rrrrrrrrrrrzFailed to save scenario) lenr generate_scenarioallprintr create_newr save_scenarioto_dict) rrrrrrr admin_input required_keysscenariorr!rcreate_scenario sB   z!ScenarioGenerator.create_scenario scenario_idcCs|j|}|r t|SdS)zLoad an existing scenarioN)r load_scenarior from_dict)rr/r"rrrr0?s  zScenarioGenerator.load_scenariocCs |jS)zList all available scenarios)r list_scenariosr rrrlist_available_scenariosFs 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) Communicationr5Professionalism AdaptabilityzResults Orientation)get)rrskill_mappingsrrrget_default_skills_for_categoryJs&z1ScenarioGenerator.get_default_skills_for_categoryN)__name__ __module__ __qualname__rstrrrrrrr.r0r3rErrrrr s 4r N) typingrrrrmodels.scenariorservices.groq_servicerutils.json_handlerrr rrrrs