Update app.py

app.py

@@ -1,4 +1,4 @@
-# Import Library yang Diperlukan
+import json
 import gradio as gr
 import gspread
 from oauth2client.service_account import ServiceAccountCredentials
@@ -13,7 +13,30 @@ from llama_index.core.chat_engine.condense_plus_context import CondensePlusConte
 from llama_index.core.schema import Document
 
 # ===================================
-# 1️⃣ Fungsi untuk Membaca Google Spreadsheet dari Beberapa Worksheet
+# 1️⃣ Fungsi Membaca Data PKB.json
+# ===================================
+def read_pkb_json():
+    try:
+        with open("pkb.json", "r", encoding="utf-8") as file:
+            data = json.load(file)
+        
+        pkb_text = "=== Perjanjian Kerja Bersama ===\n"
+        for bab, content in data["perjanjian_kerja_bersama"].items():
+            pkb_text += f"\n## {content['judul']} ##\n"
+            for pasal, pasal_data in content.items():
+                if pasal != "judul":
+                    pkb_text += f"\n### {pasal_data['judul']} ###\n"
+                    for item in pasal_data["isi"]:
+                        if isinstance(item, dict):
+                            pkb_text += f"- {item['istilah']}: {item['definisi']}\n"
+                        else:
+                            pkb_text += f"- {item}\n"
+        return pkb_text
+    except Exception as e:
+        return f"❌ ERROR membaca PKB.json: {str(e)}"
+
+# ===================================
+# 2️⃣ Fungsi Membaca Data Google Spreadsheet
 # ===================================
 def read_google_sheets():
     try:
@@ -82,7 +105,7 @@ def initialize_settings(model_path):
 # 4️⃣ Inisialisasi Index dari Data Spreadsheet
 # ===================================
 def initialize_index():
-    text_data = read_google_sheets()
+    text_data = read_google_sheets() + "\n" + read_pkb_json()
     document = Document(text=text_data)
     documents = [document]
     
@@ -113,7 +136,9 @@ def generate_response(message, history, chat_engine):
     if history is None:
         history = []
     
-    text_data = read_google_sheets()
+    # 🔹 Baca ulang data dari Google Sheets & PKB
+    text_data = read_google_sheets() + "\n" + read_pkb_json()
+    
     document = Document(text=text_data)
     documents = [document]
     
@@ -128,17 +153,17 @@ def generate_response(message, history, chat_engine):
     )
     
     chat_messages = [
-    ChatMessage(
-        role="system",
-        content=(
-            "Anda adalah chatbot yang dirancang khusus untuk berbicara dalam Bahasa Indonesia. "
-            "Anda tidak diperbolehkan menjawab dalam bahasa lain, termasuk Inggris. "
-            "Gunakan gaya bahasa profesional tetapi tetap ramah. "
-            "Jika informasi tidak tersedia dalam dokumen, katakan dengan sopan bahwa Anda tidak tahu. "
-            "Pastikan setiap jawaban diberikan secara ringkas, jelas, dan sesuai konteks."
+        ChatMessage(
+            role="system",
+            content=(
+                "Anda adalah chatbot yang dirancang khusus untuk berbicara dalam Bahasa Indonesia. "
+                "Anda tidak diperbolehkan menjawab dalam bahasa lain, termasuk Inggris. "
+                "Gunakan gaya bahasa profesional tetapi tetap ramah. "
+                "Jika informasi tidak tersedia dalam dokumen, katakan dengan sopan bahwa Anda tidak tahu. "
+                "Pastikan setiap jawaban diberikan secara ringkas, jelas, dan sesuai konteks."
+            ),
         ),
-    ),
-]
+    ]
     
     response = chat_engine.stream_chat(message)
     text = "".join(response.response_gen)