Update app.py

app.py

@@ -7,11 +7,7 @@ from oauth2client.service_account import ServiceAccountCredentials
 # =============== 1. Cache Google Sheets ===============
 cached_text_data = None
 
-def read_google_sheets():
-    global cached_text_data
-    if cached_text_data is not None:
-        return cached_text_data
-
+def search_google_sheets(user_query):
     try:
         scope = ["https://www.googleapis.com/auth/spreadsheets", "https://www.googleapis.com/auth/drive"]
         creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
@@ -20,26 +16,22 @@ def read_google_sheets():
         SPREADSHEET_ID = "1e_cNMhwF-QYpyYUpqQh-XCw-OdhWS6EuYsoBUsVtdNg"
         sheet_names = ["datatarget", "datacuti", "dataabsen", "datalembur", "pkb"]
 
-        all_data = []
+        matched_data = []
         spreadsheet = client.open_by_key(SPREADSHEET_ID)
 
         for sheet_name in sheet_names:
-            try:
-                sheet = spreadsheet.worksheet(sheet_name)
-                data = sheet.get_all_values()
-                all_data.append(f"=== Data dari {sheet_name.upper()} ===")
-                all_data.extend([" | ".join(row) for row in data])
-                all_data.append("\n")
-            except gspread.exceptions.WorksheetNotFound:
-                all_data.append(f"❌ Worksheet {sheet_name} tidak ditemukan.")
-
-        cached_text_data = "\n".join(all_data).strip()
-        return cached_text_data
+            sheet = spreadsheet.worksheet(sheet_name)
+            data = sheet.get_all_values()
+            for row in data:
+                row_text = " | ".join(row)
+                if user_query.lower() in row_text.lower():
+                    matched_data.append(row_text)
+
+        return "\n".join(matched_data) if matched_data else "Maaf, saya tidak menemukan informasi yang relevan."
 
     except Exception as e:
         return f"❌ ERROR: {str(e)}"
 
-
 # =============== 2. Load Model Transformers ===============
 def load_model():
     model_id = "HuggingFaceH4/zephyr-7b-beta"  # Bisa ganti ke Zephyr juga
@@ -65,7 +57,7 @@ def load_model():
 def generate_prompt(user_message, context_data):
     prompt = f"""
 ### SISTEM:
-Anda adalah asisten HRD yang membantu karyawan memahami administrasi perusahaan. Jawablah dengan Bahasa Indonesia yang profesional dan ramah. Jika tidak tahu, katakan tidak tahu dengan sopan. Jangan jawab dalam bahasa Inggris.
+Anda adalah chatbot HRD yang membantu karyawan memahami administrasi perusahaan. Jangan menjawab menggunakan Bahasa Inggris. Gunakan Bahasa Indonesia dengan gaya profesional dan ramah. Jika informasi tidak tersedia dalam dokumen, katakan dengan sopan bahwa Anda tidak tahu. Jawaban harus singkat, jelas, dan sesuai konteks. Jangan memberikan jawaban untuk pertanyaan yang tidak diajukan oleh pengguna. Jangan menyertakan rekomendasi pertanyaan lain.
 ### DATA:
 {context_data}
 ### PERTANYAAN:
@@ -77,17 +69,15 @@ Anda adalah asisten HRD yang membantu karyawan memahami administrasi perusahaan.
 
 # =============== 4. Generate Response ===============
 def generate_response(message, history, pipe):
-    context = read_google_sheets()
+    context = search_google_sheets(message)  # 🔍 Cari hanya bagian yang relevan
     full_prompt = generate_prompt(message, context)
     response = pipe(full_prompt)[0]["generated_text"]
 
-    # Bersihkan jawaban: ambil bagian setelah ### JAWABAN:
     cleaned = response.split("### JAWABAN:")[-1].strip()
     history = history or []
     history.append((message, cleaned))
     return cleaned
 
-
 # =============== 5. Jalankan Gradio ===============
 def main():
     pipe = load_model()