Sync from GitHub Actions: 73ae4457dcc243cc5a3129137c4f8d7c50f310b1

api/core/chatbot.py

@@ -1,10 +1,12 @@
 # api/core/chatbot.py
 import os
 import google.generativeai as genai
+from .nlp_handler import MBTI_EXPLANATIONS
 
 class MBTIChatbot:
     def __init__(self):
         print("[INIT] Initializing MBTI Chatbot (Lite Version)...")
+        # ... (rest of init)
         
         # 1. Setup Google Gemini
         api_key = os.getenv("GEMINI_API_KEY")
@@ -14,30 +16,40 @@ class MBTIChatbot:
             genai.configure(api_key=api_key)
             
         try:
-            # Pake Gemini 2.0 Flash (Standard)
             self.model = genai.GenerativeModel('gemini-2.0-flash')
         except Exception:
             print("[WARN] 2.0 Flash failed, fallback to Lite")
             self.model = genai.GenerativeModel('gemini-2.0-flash-lite')
         
     def generate_response(self, user_query, lang="en"):
-        # Versi Lite: Gak pake RAG (Database lokal), langsung pake knowledge LLM yang luas.
-        
-        lang_instruction = "Answer in English." if lang == "en" else "Jawab dalam Bahasa Indonesia gaul (Slang Jakarta/Lo-Gue), maskulin, santai, dan to the point. Panggil user 'bro' atau 'bre'. JANGAN panggil 'bestie', 'kak', atau 'gan'. Gaya bicara tongkrongan cowok tapi tetap edukatif soal MBTI."
+        lang_instruction = "Answer in English Slang." if lang == "en" else "Jawab dalam Bahasa Indonesia gaul (Slang Jakarta/Lo-Gue), maskulin, santai, dan to the point. Panggil user 'bro' atau 'bre'. JANGAN panggil 'bestie', 'kak', atau 'gan'. Gaya bicara tongkrongan cowok tapi tetap edukatif soal MBTI."
+
+        # Prepare Knowledge Base String
+        knowledge_base = "Here are the official MBTI definitions used in Sentimind:\n"
+        for mbti, desc in MBTI_EXPLANATIONS.items():
+            d_text = desc.get(lang, desc['en'])
+            knowledge_base += f"- {mbti}: {d_text}\n"
 
         system_prompt = f"""
 You are Sentimind AI, an expert in MBTI personality types and mental health.
 {lang_instruction}
 
+INTERNAL KNOWLEDGE BASE:
+{knowledge_base}
+
 USER QUERY:
 {user_query}
 
 INSTRUCTIONS:
 - Answer directly based on your extensive knowledge about MBTI and Psychology.
+- USE the Internal Knowledge Base for definitions if asked about specific types.
+- STRICTLY REFUSE to answer questions unrelated to MBTI, Personality, Psychology, or Mental Health. 
+  - If asked about coding, politics, math, or general trivia, say: "Maaf bro, gua cuma ahli soal MBTI dan Psikologi aja." (or English equivalent).
 - Be empathetic, insightful, and use formatting (bullet points) if helpful.
 - Keep answers concise (under 200 words) unless asked for details.
 - DO NOT use emojis in your response. Keep it clean and text-only.
 """
+
         try:
             response = self.model.generate_content(system_prompt)
             return response.text