improve search similarity

app_gradio.py

@@ -65,18 +65,36 @@ class BPJSService:
         return SequenceMatcher(None, a.lower(), b.lower()).ratio()
 
     def search_faq(self, question: str, threshold: float = 0.3):
-        """Search for relevant FAQ based on question"""
+        """
+        Search for relevant FAQ based on question
+
+        Args:
+            question: User's question
+            threshold: Minimum similarity score (0.0-1.0). Default 0.3.
+                      - Lower = more results but less accurate (e.g., 0.2)
+                      - Higher = fewer but more accurate results (e.g., 0.5)
+        """
         best_match = None
         best_score = 0
+        question_lower = question.lower()
 
         for faq in self.faq_data.get("data", []):
             question_text = faq.get("pertanyaan", "")
             title_text = faq.get("title", "")
 
+            # Calculate base similarity scores
             question_score = self.similarity(question, question_text)
             title_score = self.similarity(question, title_text)
 
-            score = max(question_score, title_score)
+            # Bonus score for exact keyword matches (helps find specific FAQs)
+            keywords = ['denda', 'iuran', 'bayar', 'keterlambatan', 'rawat inap', 'phk', 'klaim']
+            keyword_bonus = 0
+            for keyword in keywords:
+                if keyword in question_lower and keyword in question_text.lower():
+                    keyword_bonus += 0.1
+
+            # Combine scores
+            score = max(question_score, title_score) + keyword_bonus
 
             if score > best_score and score >= threshold:
                 best_score = score
@@ -314,13 +332,10 @@ class BPJSService:
             state["step"] = "tanggal_lahir"
 
             response = f"""✅ Nama: **{data['nama']}**
-
 **Langkah 2 dari 4**
 Sekarang masukkan **tanggal lahir** Anda:
-
 Format: DD/MM/YYYY
 _Contoh: 15/08/1990_
-
 _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
 
         elif current_step == "tanggal_lahir":
@@ -334,14 +349,11 @@ _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
             state["step"] = "jenis_kelamin"
 
             response = f"""✅ Tanggal Lahir: **{data['tanggal_lahir']}**
-
 **Langkah 3 dari 4**
 Pilih **jenis kelamin** Anda:
-
 Ketik:
 - **L** atau **Laki-laki**
 - **P** atau **Perempuan**
-
 _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
 
         elif current_step == "jenis_kelamin":
@@ -357,12 +369,9 @@ _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
             state["step"] = "provinsi"
 
             response = f"""✅ Jenis Kelamin: **{data['jenis_kelamin']}**
-
 **Langkah 4 dari 4 (Terakhir!)**
 Masukkan **provinsi** tempat tinggal Anda:
-
 _Contoh: DKI Jakarta, Jawa Barat, Jawa Timur, dll._
-
 _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
 
         elif current_step == "provinsi":
@@ -373,20 +382,16 @@ _(Ketik 'batal' jika ingin membatalkan registrasi)_"""
             state["step"] = "konfirmasi"
 
             response = f"""## 📋 Konfirmasi Data Registrasi
-
 Mohon periksa data Anda:
-
 | Field | Value |
 |-------|-------|
 | **Nama** | {data['nama']} |
 | **Tanggal Lahir** | {data['tanggal_lahir']} |
 | **Jenis Kelamin** | {data['jenis_kelamin']} |
 | **Provinsi** | {data['provinsi']} |
-
 Apakah data sudah benar?
 - Ketik **YA** untuk melanjutkan registrasi
 - Ketik **TIDAK** atau **BATAL** untuk membatalkan
-
 _(Data ini akan disimpan ke database)_"""
 
         elif current_step == "konfirmasi":
@@ -406,23 +411,18 @@ _(Data ini akan disimpan ke database)_"""
 
                 if success:
                     response = f"""## 🎉 Registrasi BPJS Berhasil!
-
 **NOMOR PESERTA ANDA:** `{nomor_peserta}`
-
 ### 📋 Data yang Tersimpan:
 - **Nama:** {data['nama']}
 - **Tanggal Lahir:** {data['tanggal_lahir']}
 - **Jenis Kelamin:** {data['jenis_kelamin']}
 - **Provinsi:** {data['provinsi']}
 - **Status:** AKTIF ✅
-
 ⚠️ **PENTING:** Simpan nomor peserta Anda dengan baik!
-
 Anda sekarang bisa:
 - Mengecek status dengan: "Cek peserta {nomor_peserta}"
 - Bertanya tentang FAQ BPJS
 - Bertanya tentang layanan kesehatan
-
 Ada yang bisa saya bantu lagi? 😊"""
                 else:
                     response = f"❌ **Error:** {msg}\n\nSilakan coba daftar lagi dengan ketik 'daftar BPJS'."
@@ -492,13 +492,10 @@ Ada yang bisa saya bantu lagi? 😊"""
                 "data": {}
             }
             response = """## 📝 Registrasi Peserta BPJS Baru
-
 Baik! Saya akan membantu Anda mendaftar sebagai peserta BPJS.
 Prosesnya akan saya tanyakan satu per satu ya.
-
 **Langkah 1 dari 4**
 Silakan masukkan **nama lengkap** Anda:
-
 _Contoh: Budi Santoso_"""
 
         elif question_type == "status_check":
@@ -618,13 +615,11 @@ demo = gr.ChatInterface(
     title="🏥 BPJS Chatbot",
     description="""
     Chatbot untuk FAQ BPJS dan layanan kepesertaan menggunakan Groq LLM.
-
     **Features:**
     - 📋 FAQ BPJS dengan pencarian cerdas
     - 👤 Cek status kepesertaan dari database
     - 📝 Registrasi peserta baru (step-by-step)
     - 💬 Conversational AI dengan context awareness
-
     **Contoh pertanyaan:**
     - "Bagaimana cara bayar iuran BPJS?"
     - "Cek peserta 12345678"
@@ -657,4 +652,4 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False
-    )
+    )