o .i]M@svddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z m Z mZmZmZmZGdddZdS)N)DictListAnyOptional) AsyncGroq)SCENARIO_GENERATION_PROMPTCHARACTER_ROLEPLAY_PROMPTCHARACTER_CONCLUSION_PROMPT CHARACTER_GUIDED_ROLEPLAY_PROMPTSKILL_ANALYSIS_PROMPTget_skill_analysis_templatec@sDeZdZddZdefddZdeeeffddZd d Z d eeefde eeeffd d Z de eeefde eeefdefddZ d$deeefde eeefdedede eeefde ef ddZd%deeefde eeefdede eeeffddZdede efdd Zd!ede efd"d#ZdS)& GroqServicecCsddlm}|td}|s#tdtdtdd|_d|_n t|d|_d |_d |_dddd dddd dddd d |_ dS) Nr) load_dotenv GROQ_API_KEYu@⚠️ WARNING: GROQ_API_KEY not found in environment variablesuL ℹ️ Groq service will not be available. System will use Ollama only.uJ ℹ️ To use Groq, set GROQ_API_KEY in your .env file or environmentF)api_keyTzopenai/gpt-oss-20binputoutputtotal)preview conversation assessment) dotenvrosgetenvprintclient availablermodel token_counts)selfrrr!G/var/www/eduai.edurigo.com/roleplay/production/services/groq_service.py__init__s       zGroqService.__init__operation_typec Cs|rHt|drJt|drL|j}|j}|j}|j|d|7<|j|d|7<|j|d|7<td|d|d|d |d Sd Sd Sd S) z*Update token counts for specific operation prompt_tokenscompletion_tokensrrrzGROQ Token count for z: Input=z , Output=z, Total=N)hasattrr%r& total_tokensrr)r r$ usage_stats input_tokens output_tokensr(r!r!r"_update_token_count(s$ zGroqService._update_token_countreturncCs |jS)z+Get current token counts for all operations)rcopy)r r!r!r"get_token_counts5s zGroqService.get_token_countscCs"|jD] }dddd|j|<qdS)zReset all token countersrrN)r)r operationr!r!r"reset_token_counts9s zGroqService.reset_token_counts admin_inputc szd}|dD]\}}||d|d7}q d}|d}|rr4skillssystemzHYou are a training scenario expert. Always respond with valid JSON only.rolecontentuserffffff?imessagesr temperature max_tokensNrr```json```zJSON parsing error: Response text: zError generating scenario: )itemsget enumeraterformatjoinrchat completionscreaterr,usagechoicesmessagerEstrip startswithendswithjsonloadsJSONDecodeErrorr Exception) r r2 details_textkeyvaluequestions_textr7idxqpromptresponse response_texter!r!r"generate_scenario>sd         zGroqService.generate_scenarior7conversation_historyc s|sdSddd|Dg}t|dD]7\}}|dd}|dd }tfd d |D}|r:d nd } |d|d|ddd| qt|D]4\}}|dd}|dd }tfdd |D}|s|d|dd|ddnqSd|S)z Analyze conversation history to determine which guided questions have been addressed. Returns a formatted string explaining the progression status. z&No guided questions for this roleplay. cSsg|]}|ddqS)r[r3rR).0turnr!r!r" sz=GroqService._analyze_question_progression..r8r:r3Nc3$|] }t|dkr|vVqdSrtNlenrqkeywordconversation_textr!r" "z.u ✓ ADDRESSEDu⚠ NOT YET ADDRESSEDz Question r5 - c3rurvrwryr{r!r"r}r~z **NEXT FOCUS**: Question r6)rUlowerrSrRsplitanyappend) r r7rnprogression_linesrgrh question_textkeywords addressedstatusr!r{r"_analyze_question_progression~s& &" z)GroqService._analyze_question_progressionFNscenario user_message is_conclusionc szd}|D]}|ddkrdn|dd}||d|dd 7}qtd d |Dd } |d d} |dd} |dd} |dd} |dd}|rt}|dd|dd|dd|dd|ddd|dd|dd|dd||d }n|rt}d}t|d D]\}}||d |d!dd 7}q|||}id"|ddd|ddd|ddd|ddd|dddd|ddd|ddd#|d#dd$|d%iddd|ddd&|d'|d(|d)|d | d*| d| | || d+}nit}id"|ddd|ddd|ddd|ddd|dddd|ddd|ddd#|d#dd$|d%iddd|ddd(|d)|d | d*| d| d| d|d,| i}|jd9i|}|jj j j d-d.d/d0|d/g|j d1d2d3Id4H}| d5|j|jd6jjWSty}ztd7|WYd4}~d8Sd4}~ww):z&AI character responses during roleplayr3speakerlearnerLearner ai_characternamer5r[r6cSsg|] }|ddkr|qS)rrrp)rqtr!r!r"rssz.GroqService.play_character..r8 roleplay_namezRoleplay Sessionr> Characterr= difficultymediumdurationi, personalitygoals backgroundemotional_stateneutralscenario_setupcontext environment constraints) character_namerrrrrrrrnrr9r:rr<additional_infor4guided_questionsquestion_progress_analysisrnrcharacter_role)rr current_turnrrBa"You are a world-class roleplay AI. You MUST stay strictly in character as a human-like counterpart (client, customer, etc.). You NEVER break character. You MUST follow the provided 'MANDATORY GUIDED QUESTIONS' if they exist. Respond professionally, naturally, and concisely (1-3 sentences).rCrFg?irHNrrzError in character roleplay: z:I'm having trouble responding right now. Please try again.r!)rxrRr r rSrrrTrrVrWrXrr,rYrZr[rEr\rbr)r rrnrrr7 history_textrrrrrrr=rrprompt_template prompt_kwargsrfrgrhquestion_progressrirjrlr!r!r"play_characters                                          zGroqService.play_characterr3conversation_turnsengagement_metadatac stzd}|D]}|ddkrdnd}||d|dd7}qd |d d |d d |ddd|dd|ddd|ddd }|d}t|} tjd||||| d} |jjjjdddd| dg|jddd Id!H} | d"| j | j d#j j } zB| d$r| d%d!} n | d&r| d'd!} | d&r| d!d(} | } t| tfd)d*d+Dstd,WWd!SWWStjy} zJtd-| td.| d#d!l}|d/| |j}|rzt|td0WWYd!} ~ WStjytd1YnwWYd!} ~ Wd!Sd!} ~ wwty9} ztd2| WYd!} ~ d!Sd!} ~ ww)3z:Comprehensive skill analysis with engagement-aware scoringr3rrrz AI Characterr5r[r6z Category: r;z Objective: r<z Context: rrz Success Criteria: success_criteriaz AI Character: rrrrr@r?)rAscenario_contextrrskill_analysis_templaterBzYou are an expert skill assessor. Provide STRICT, ACCURATE analysis in valid JSON format only. Do NOT inflate scores. Follow the engagement-based score ceilings exactly.rCrFg333333?i rHNrrrLrMrNrtrOc3s|]}|vVqdS)Nr!)rqrdanalysisr!r"r}Qsz-GroqService.analyze_skills..)skill_analysisoverall_performanceconversation_analysisrecommendationszAnalysis missing required keysz JSON parsing error in analysis: rPz\{.*\}z0Successfully extracted JSON using regex fallbackzRegex fallback also failedzError analyzing skills: )r r rTrUrrVrWrXrr,rYrZr[rEr\r]r^r_r`allrraresearchDOTALLgrouprb)r rrrr|rrrrrAskill_templaterirjrkrlr json_matchr!rr"analyze_skillss             zGroqService.analyze_skillsric sz)|jjjjdddd|dg|jdddIdH}|d |j|jd jj WSt yD}zt d |WYd}~dSd}~ww) z)Get a simple text completion from the LLMrBz]You are a helpful assistant. Provide clear, natural responses without any formatting symbols.rCrFrGirHNrrzError getting completion: ) rrVrWrXrr,rYrZr[rEr\rbr)r rirjrlr!r!r"get_completionls*   zGroqService.get_completiontextc sf|jsdS|rt|dkrdS|dd}z~td}d}d|dd}d d |d d }td |dddtjdd4IdH}|j|||dIdH}WdIdHn 1IdHsawY|j dkr|j }|rt | d} tdt| d| WSWdStd|j d|jWdSty} ztdt| WYd} ~ dSd} ~ ww)z Generate human-like voice using Groq's TTS model. Uses direct HTTP request to avoid SDK version conflicts. Returns base64 encoded audio string. Nrirz+https://api.groq.com/openai/v1/audio/speechzBearer zapplication/json) Authorizationz Content-Typezcanopylabs/orpheus-v1-englishtaramp3)rvoicerresponse_formatz*INFO: Requesting TTS from Groq for text: 'z...'g>@)timeout)headersr_zutf-8z'SUCCESS: Generated TTS base64 (length: )zERROR: Groq TTS API returned r5zEXCEPTION in generate_tts: )rrxr\rrrhttpx AsyncClientpost status_coderEbase64 b64encodedecoderrbstr) r rrurlrpayloadrrj audio_data b64_audiorlr!r!r" generate_ttssF  ( zGroqService.generate_tts)FN)r3)__name__ __module__ __qualname__r#rr,rrr/r1rrmrrboolrrrrr!r!r!r"r s &.@H s