|
|
|
|
|
import os |
|
|
from google import genai |
|
|
from .nlp_handler import MBTI_EXPLANATIONS |
|
|
|
|
|
class MBTIChatbot: |
|
|
def __init__(self): |
|
|
print("[INIT] Initializing MBTI Chatbot (Lite Version)...") |
|
|
|
|
|
|
|
|
|
|
|
api_key = os.getenv("GEMINI_API_KEY") |
|
|
if not api_key: |
|
|
print("[WARN] GEMINI_API_KEY not found in .env.") |
|
|
self.client = None |
|
|
else: |
|
|
try: |
|
|
self.client = genai.Client(api_key=api_key) |
|
|
except Exception as e: |
|
|
print(f"[ERR] Gemini Client Init Failed: {e}") |
|
|
self.client = None |
|
|
|
|
|
|
|
|
self.model_name = 'gemini-2.0-flash' |
|
|
|
|
|
def generate_response(self, user_query, lang="en"): |
|
|
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." |
|
|
|
|
|
|
|
|
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: |
|
|
if not self.client: |
|
|
return "Maaf, kunci otak saya (API Key) belum dipasang atau salah." |
|
|
|
|
|
response = self.client.models.generate_content( |
|
|
model=self.model_name, |
|
|
contents=system_prompt |
|
|
) |
|
|
return response.text |
|
|
except Exception as e: |
|
|
return f"Maaf, saya sedang mengalami gangguan koneksi ke otak AI saya. (Error: {str(e)})" |
|
|
|
