o h@D@sxddlmZmZddlmZddlmZmZddlm Z m Z m Z m Z ddl m Z ddlZddlZddlmZeddlmZdd lmZdd lmZdd lmZed d ddZejedgddgdgdeZeZeZGdddeZGdddeZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$Gdd d eZ%Gd!d"d"eZ&e'd#d$d%Z(e'd&d'd(Z)ej*d)e"d*d+e!fd,d-Z+d+e!d.e efd/d0Z,d1e-d2e-d.e-fd3d4Z.d5e-d.e efd6d7Z/d8ed.e%fd9d:Z0d8ed.e-fd;d<Z1ej*d=e&d*d+e!fd>d?Z2e*d@d5e-fdAdBZ3e'dCd5e-fdDdEZ4e5dFkr:ddl6Z6e6j7edGdHdIdSdS)J)FastAPI HTTPException)CORSMiddleware) BaseModelField)ListDictAnyOptional)datetimeN) load_dotenv)ScenarioGenerator)RoleplayEngine) GroqService)RoleplayScenarioAI Roleplay Servicez,Simple AI service for roleplay functionalityz1.0.0)title descriptionversion*T) allow_originsallow_credentials allow_methods allow_headersc@eZdZUeed<eed<dS) SkillDataskill_id skill_nameN__name__ __module__ __qualname__str__annotations__r$r$R/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/prod/roleplay_fast_api.pyr(  rc@sZeZdZUeed<eed<edddZeed<eed<edd dZeed <ee ed <d S) RoleplayDatacategory objective.z Learner role)alias learner_roleadditional_infozConstraints/Policiescompany_policiesskills_for_roleplayN) rr r!r"r#rr+r.rrr$r$r$r%r',s r'c@s.eZdZUeed<eed<eed<eed<dS)RoleplayRequest client_id session_id roleplay_dataqueryN)rr r!r"r#r'r$r$r$r%r04s  r0c@r)RoleplayResponser2responseNrr$r$r$r%r5:r&r5c@s6eZdZUeed<eed<eed<eed<eed<dS)CharacterDetailsname personalitygoals backgroundemotional_stateNrr$r$r$r%r7>s  r7c@s&eZdZUeed<eed<eed<dS) ScenarioSetupcontext environment constraintsNrr$r$r$r%r=Es  r=c@szeZdZUeed<eed<eed<eed<eed<eeed<eed<eed<eed <eeefed <eed <eed <d S)PreviewResponse scenario_idr(r)r+ai_roleskills_to_assessscenario_setupcharacter_detailsconversation_startersuccess_criteriadifficulty_levelbackground_infoN) rr r!r"r#rr=r7rr$r$r$r%rAJs   rAc@seZdZUeed<dS)ScenarioPreviewResponser6Nrr$r$r$r%rKXs rK/cs dddS)zHealth check endpointrrunning)messagestatusr$r$r$r$r%root[s rPz/healthc s`ztd}|sdddWSdtdWSty/}z tddt|d d }~ww) z Health check GROQ_API_KEYerrorzGROQ_API_KEY not configured)rOrNhealthy)rO timestampzHealth check failed:  status_codedetailN)osgetenvr now isoformat Exceptionrr")api_keyer$r$r% health_check`s  r`z /roleplay)response_modelrequestc sz|j}td|t|}td|du|ssz0create_scenario_from_request..zRoleplay conversationMedium)r;r@r?rI)r(r)detailsrCr+rDzError creating scenario: N)r3r/r,r.generate_ai_roler(scenario_generatorcreate_scenarior)r+r2id json_handler save_scenarioto_dictr]rc)rbr3rDrvrCrmr_r$r$r%res4 rer(r,cCs~dddddd}||d}d|vrd |Sd |vs'd |vr,d |Sd |vs8d|vr=d|S|S)z:Generate appropriate AI role based on category and contextzPotential CustomerzCustomer with Issuez Team MemberzNegotiation PartnerzUser with Technical Problem)saleszcustomer service leadership negotiationztechnical supportzConversation Partner enterprisez Enterprise budgetpricezBudget-Conscious frustrated complaintz Frustrated )getlower)r(r,category_roles base_roler$r$r%rws    rwr2c CsHzt|WSty#}ztd|d|WYd}~dSd}~ww)z"Load scenario for existing sessionz#Error loading scenario for session z: N)rx load_scenarior]rc)r2r_r$r$r%rds rdrmcCszmt|jdd|jdd|jdd|jdd|jddd}t|jd d|jd d|jd dd }i}t|jtrE|j}nd t|ji}t |j |j |j |j |j|j|||j||jdd|jddd WSty}ztd|d}~ww)z.Convert scenario to formatted preview responser8 AI Characterr9 Not specifiedr:r;r<)r8r9r:r;r<r>r?r@)r>r?r@generalrIru) rBr(r)r+rCrDrErFrGrHrIrJz'Error formatting scenario for preview: N)r7 ai_characterrr=rE isinstancerHdictr"rArzr(r)r+rCrDrGrvr]rc)rmrFrErHr_r$r$r%format_scenario_for_previewsH           rcCszid|jd|jd|jd|jd|jd|jddd|jddd |jd dd |jd d d |jddd|jddd|jddd|jddd|jd|j d|j ddd|j dd}d|d}t |}|sd|jd|jd|jd dd |jd!|jd"|jdd#d!|jdd$d%|jWS| WSty}z+td&|d|jd|jd|jd dd |jd!|jd%|j WYd'}~Sd'}~ww)(z4Convert scenario to natural paragraph text using LLMr(r)r+rCrDr>rr?r@ai_character_namer8rai_personalityr9 ai_backgroundr;ai_goalsr:ai_emotional_stater<rGrHrJrIruaConvert the following roleplay scenario data into natural, flowing paragraphs without any headings, bullet points, or markdown formatting. Write it as a cohesive narrative description that explains everything about the scenario in a conversational manner. Scenario Data: a Requirements: - Write in natural paragraphs only (no headings, no bullet points, no markdown) - Make it flow like a story or description - Include all the important information about the scenario, roles, characters, objectives, and success criteria - Keep it engaging and easy to understand - Don't use any formatting symbols like **, ##, -, etc. - Write it as if you're explaining the scenario to someone in person Write the complete scenario description in paragraph form:z This is a z. roleplay scenario where you play the role of z and interact with zan AI characterz who is a z. z The scenario takes place in za roleplay settingr-z# The conversation will begin with: z1Error formatting scenario as paragraph with LLM: N)r(r)r+rCrDrErrrGrHrv groq_serviceget_completionstripr]rc)rm scenario_datapromptparagraph_responser_r$r$r%format_scenario_as_paragraph0s`      ^ Hrz/roleplay_scenarioc sz;td|jt|j}|rtdt|}t|dWStdt|}|s/tdddt|}tdt|dWStyDtyk}ztd t|d d l }| tdd t|dd }~ww) zsr           M) . 7!