Update app.py

app.py

@@ -28,49 +28,71 @@ Important guidelines:
 
 Please respond in Indonesian language unless the user specifically asks for English."""
 
-def is_dental_related(message):
-    dental_keywords = [
-        'gigi', 'mulut', 'sakit gigi', 'berlubang', 'taring', 'gusi', 'karang gigi',
-        'behel', 'kawat gigi', 'tambal gigi', 'cabut gigi', 'gigi tiruan', 'gigi palsu',
-        'pembersihan', 'scaling', 'bleaching', 'veneer', 'mahkota gigi', 'implan',
-        'odontologi', 'dokter gigi', 'klinik gigi', 'obat gigi', 'pasta gigi',
-        'sikat gigi', 'benang gigi', 'kumur', 'mulut kering', 'bau mulut',
-        'radang gusi', 'gingivitis', 'periodontitis', 'karies', 'plak',
-        'tongue', 'lidah', 'saliva', 'liur', 'email gigi', 'dentin',
-        'saraf gigi', 'akar gigi', 'gigi berlubang', 'gigi goyang', 'gigi tanggal',
-        'gigi susu', 'gigi dewasa', 'gigi bungsu', 'wisdom tooth', 'orthodonti',
-        'bracket', 'alat ortodonti', 'retainer', 'gigi rapat', 'gigi renggang',
-        'gigi maju', 'gigi tonggos', 'gigi nongol', 'anestesi', 'bius',
-        'x-ray gigi', 'radiologi', 'panoramic', 'cek up', 'kontrol', 'perawatan',
-        'dental', 'tooth', 'teeth', 'gum', 'enamel', 'cavity', 'braces',
-        'orthodontist', 'dentist', 'oral', 'mouth', 'jaw', 'rahang'
-    ]
-    return any(keyword in message.lower() for keyword in dental_keywords)
+NON_MEDICAL_KEYWORDS = [
+    'masak', 'resep', 'film', 'musik', 'lagu', 'game', 'politik', 'bisnis',
+    'saham', 'crypto', 'olahraga', 'bola', 'basket', 'travel', 'wisata',
+    'fashion', 'cuaca', 'berita', 'gosip', 'artis', 'coding', 'programming',
+    'hukum', 'harga', 'beli', 'jual', 'toko', 'belanja'
+]
+
+def extract_text(content):
+    """Ekstrak teks dari content yang bisa berupa string, list, atau dict"""
+    if isinstance(content, str):
+        return content
+    elif isinstance(content, list):
+        texts = []
+        for item in content:
+            if isinstance(item, dict):
+                texts.append(item.get('text', '') or item.get('content', ''))
+            elif isinstance(item, str):
+                texts.append(item)
+        return ' '.join(filter(None, texts))
+    elif isinstance(content, dict):
+        return content.get('text', '') or content.get('content', '') or str(content)
+    return str(content) if content else ''
+
+def is_medical_related(message):
+    """Izinkan semua topik medis, blokir hanya yang jelas non-medis"""
+    msg_lower = message.lower()
+    # Blokir jika mengandung keyword non-medis yang jelas
+    non_medical_count = sum(1 for kw in NON_MEDICAL_KEYWORDS if kw in msg_lower)
+    # Jika mengandung 2+ keyword non-medis dan tidak ada konteks medis, tolak
+    medical_hints = ['sakit', 'nyeri', 'obat', 'dokter', 'gejala', 'penyakit',
+                     'kesehatan', 'medis', 'klinik', 'rumah sakit', 'operasi',
+                     'terapi', 'diagnosis', 'treatment', 'health', 'pain', 'disease']
+    has_medical = any(kw in msg_lower for kw in medical_hints)
+    if non_medical_count >= 2 and not has_medical:
+        return False
+    return True
 
 def dental_chat(message, history):
-    if not message.strip():
-        history.append({"role": "user", "content": message})
+    message_text = extract_text(message)
+
+    if not message_text.strip():
+        history.append({"role": "user", "content": message_text})
         history.append({"role": "assistant", "content": "Mohon masukkan pertanyaan yang valid."})
         yield "", history
         return
 
-    if not is_dental_related(message):
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": "Maaf, saya hanya bisa membantu dengan pertanyaan seputar kesehatan gigi dan mulut. Silakan ajukan pertanyaan tentang topik kedokteran gigi."})
+    if not is_medical_related(message_text):
+        history.append({"role": "user", "content": message_text})
+        history.append({"role": "assistant", "content": "Maaf, saya hanya bisa membantu dengan pertanyaan seputar kesehatan dan kedokteran gigi. Silakan ajukan pertanyaan medis."})
         yield "", history
         return
 
     try:
         mistral_messages = [ChatMessage(role="system", content=DENTAL_SYSTEM_PROMPT)]
+
         for msg in history:
-            mistral_messages.append(ChatMessage(role=msg["role"], content=msg["content"]))
-        mistral_messages.append(ChatMessage(role="user", content=message))
+            content_text = extract_text(msg["content"])
+            if content_text.strip():
+                mistral_messages.append(ChatMessage(role=msg["role"], content=content_text))
 
-        # Tambahkan user message dan placeholder kosong untuk streaming
-        history.append({"role": "user", "content": message})
+        mistral_messages.append(ChatMessage(role="user", content=message_text))
+
+        history.append({"role": "user", "content": message_text})
         history.append({"role": "assistant", "content": ""})
 
-        # Streaming response
         response_text = ""
         for chunk in client.chat_stream(
             model="mistral-tiny",
@@ -86,13 +108,29 @@ def dental_chat(message, history):
 
     except Exception as e:
         print(f"Error: {str(e)}")
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": f"Maaf, terjadi kesalahan: {str(e)}. Silakan coba lagi."})
+        if history and history[-1]["content"] == "":
+            history[-1]["content"] = f"Maaf, terjadi kesalahan: {str(e)}. Silakan coba lagi."
+        else:
+            history.append({"role": "assistant", "content": f"Maaf, terjadi kesalahan: {str(e)}. Silakan coba lagi."})
         yield "", history
 
 css = """
 footer { display: none !important; }
 
+/* Ganti loading dots dengan teks Thinking... */
+.progress-text span,
+.eta-bar,
+.generating span {
+    display: none !important;
+}
+
+.generating::after {
+    content: 'Thinking...' !important;
+    color: #94a3b8 !important;
+    font-style: italic !important;
+    font-size: 0.9rem !important;
+}
+
 body, .gradio-container {
     background-color: #0a0e27 !important;
 }