o 1hV@sZddlmZmZmZmZddlmZddlmZddl m Z ddl m Z GdddZ dS) )DictListAnyOptional)RoleplayScenario) GroqService) OllamaService) JSONHandlerc@seZdZdddZdefddZ dded ed ed eeefd ed ede ede dede e fddZ dede de e fddZde de eeeffddZd ede efddZdS)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_serviceT session_idcategory objectivedetailsai_role learner_roleskills_to_assessis_adminreturnc st|dkr |dd}||||||d} || } | | s#dSgd} tfdd| Ds;td| dStj|||||||dd d d d d } |j| ||rb| 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: rr r!r"r#) rrrrrrrrr r!r"r#zFailed to save scenario) lenrgenerate_scenarioallprintr create_newr save_scenarioto_dict)rrrrrrrrrr admin_input ai_service required_keysscenariorr&rcreate_scenariosF   z!ScenarioGenerator.create_scenariocCs |j||}|rt|SdS)zLoad an existing scenarioN)r load_scenarior from_dict)rrrr'rrrr5Hs zScenarioGenerator.load_scenariocCs |j|S)zList all available scenarios)rlist_scenarios)rrrrrlist_available_scenariosOs 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) Communicationr:Professionalism AdaptabilityzResults Orientation)get)rrskill_mappingsrrrget_default_skills_for_categorySs&z1ScenarioGenerator.get_default_skills_for_category)NN)T)__name__ __module__ __qualname__rboolrstrrrrintrrr4r5r8rJrrrrr s(  6r N)typingrrrrmodels.scenariorservices.groq_servicerservices.ollama_servicerutils.json_handlerr r rrrrs