Update ui.py

ui.py

@@ -3,7 +3,13 @@
 import gradio as gr
 
 from rag import answer_question
-from data.manual_answers import add_manual_qa
+from data.manual_answers import (
+    add_manual_qa,
+    list_manual_qa,
+    get_manual_qa,
+    update_manual_qa,
+    delete_manual_qa,
+)
 
 
 # ========= Student side =========
@@ -13,25 +19,145 @@ def chat_fn(message: str, history: list):
     return answer_question(message)
 
 
-# ========= Teacher side =========
+# ========= Teacher helpers =========
 
-def save_teacher_qa(question: str, answer: str) -> str:
+def _refresh_manual_data():
     """
-    Called when teacher clicks 'Save Q&A'.
+    Helper: get table data + dropdown choices.
+    """
+    rows = list_manual_qa()
+    table = [[r["norm_q"], r["q"], r["a"]] for r in rows]
+    keys = [r["norm_q"] for r in rows]
+    return table, keys
+
+
+def save_teacher_qa(question: str, answer: str):
+    """
+    Add a new Q&A (or overwrite existing same question).
+    Returns: status_text, table_data, dropdown_choices, editing_key
     """
     if not question.strip() or not answer.strip():
-        return "❗ ກະລຸນາພິມທັງຄໍາຖາມ ແລະຄໍາຕອບ."
+        table, keys = _refresh_manual_data()
+        return (
+            "❗ ກະລຸນາພິມທັງຄໍາຖາມ ແລະຄໍາຕອບ.",
+            table,
+            keys,
+            "",
+        )
 
     try:
         norm_key = add_manual_qa(question, answer)
+        table, keys = _refresh_manual_data()
         return (
             "✅ ບັນທຶກສໍາເລັດ! ຈາກນີ້ chatbot ຈະໃຊ້ຄໍາຕອບນີ້ທັນທີ.\n"
-            f"🔑 ຄີທີ່ໃຊ້ຄົ້ນຫາ (normalized): `{norm_key}`"
+            f"🔑 ຄີທີ່ໃຊ້ຄົ້ນຫາ (normalized): `{norm_key}`",
+            table,
+            keys,
+            norm_key,
         )
     except Exception as e:
-        return f"⚠️ ບັນທຶກບໍ່ສໍາເລັດ: {e}"
+        table, keys = _refresh_manual_data()
+        return (f"⚠️ ບັນທຶກບໍ່ສໍາເລັດ: {e}", table, keys, "")
+
+
+def load_for_edit(selected_key: str):
+    """
+    Load one row into the question/answer textboxes.
+    Returns: question, answer, editing_key
+    """
+    if not selected_key:
+        return "", "", ""
+
+    row = get_manual_qa(selected_key)
+    if not row:
+        return "", "", ""
+
+    return row["q"], row["a"], row["norm_q"]
 
 
+def update_teacher_qa(current_key: str, question: str, answer: str):
+    """
+    Update existing Q&A (identified by current_key).
+    Returns: status_text, table_data, dropdown_choices, new_editing_key
+    """
+    if not current_key:
+        table, keys = _refresh_manual_data()
+        return (
+            "❗ ກະລຸນາເລືອກ Q&A ຈາກຊ່ອງ Key ຫຼືຕາຕະລາງກ່ອນ.",
+            table,
+            keys,
+            current_key,
+        )
+
+    if not question.strip() or not answer.strip():
+        table, keys = _refresh_manual_data()
+        return (
+            "❗ ຄໍາຖາມ ແລະຄໍາຕອບຕ້ອງບໍ່ວ່າງ.",
+            table,
+            keys,
+            current_key,
+        )
+
+    try:
+        new_key = update_manual_qa(current_key, question, answer)
+        table, keys = _refresh_manual_data()
+        return (
+            "✅ ແກ້ໄຂ Q&A ສໍາເລັດ.",
+            table,
+            keys,
+            new_key,
+        )
+    except Exception as e:
+        table, keys = _refresh_manual_data()
+        return (
+            f"⚠️ ແກ້ໄຂບໍ່ສໍາເລັດ: {e}",
+            table,
+            keys,
+            current_key,
+        )
+
+
+def delete_teacher_qa(current_key: str):
+    """
+    Delete selected Q&A.
+    Returns: status_text, table_data, dropdown_choices, new_editing_key, q_text, a_text
+    """
+    if not current_key:
+        table, keys = _refresh_manual_data()
+        return (
+            "❗ ກະລຸນາເລືອກ Q&A ທີ່ຈະລົບກ່ອນ.",
+            table,
+            keys,
+            "",
+            "",
+            "",
+        )
+
+    try:
+        delete_manual_qa(current_key)
+        table, keys = _refresh_manual_data()
+        return (
+            "🗑️ ລົບ Q&A ສໍາເລັດ.",
+            table,
+            keys,
+            "",
+            "",
+            "",
+        )
+    except Exception as e:
+        table, keys = _refresh_manual_data()
+        return (
+            f"⚠️ ລົບບໍ່ສໍາເລັດ: {e}",
+            table,
+            keys,
+            current_key,
+            "",
+            "",
+        )
+
+
+# ========= Build Gradio app =========
+
 def create_app():
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -39,7 +165,7 @@ def create_app():
 # 📚 Laos History Chatbot (M1)
 
 - 👩‍🎓 **Student Chat**: ນັກຮຽນຖາມຄໍາຖາມປະຫວັດສາດພາສາລາວ  
-- 👩‍🏫 **Teacher Panel**: ຄູເພີ່ມ Q&A ທີ່ຖືກຕ້ອງເອງ ເພື່ອໃຫ້ chatbot ຕອບໄດ້ດີຂຶ້ນ
+- 👩‍🏫 **Teacher Panel**: ຄູເພີ່ມ / ແກ້ໄຂ / ລົບ Q&A ທີ່ຖືກຕ້ອງເອງ
 """
         )
 
@@ -61,14 +187,15 @@ def create_app():
         with gr.Tab("👩‍🏫 Teacher Panel"):
             gr.Markdown(
                 """
-### ✍️ ເພີ່ມ Q&A ສໍາລັບຄູ
+### ✍️ ເພີ່ມ / ແກ້ໄຂ / ລົບ Q&A ສໍາລັບຄູ
 
-- ພິມ **ຄໍາຖາມ** ແລະ **ຄໍາຕອບ** ພາສາລາວ  
-- ກົດ **"💾 Save Q&A"**  
-- Chatbot ຈະໃຊ້ຄໍາຕອບນີ້ທັນທີເມື່ອນັກຮຽນຖາມຄໍາຖາມຄ້າຍກັນ
+1. ພິມ **ຄໍາຖາມ** ແລະ **ຄໍາຕອບ** ແລ້ວກົດ **💾 Save Q&A** ເພື່ອເພີ່ມໃໝ່  
+2. ຫາກຈະແກ້ໄຂ: ເລືອກ Key, ກົດ **📥 Load**, ແກ້ໄຂ ແລ້ວກົດ **✏️ Update**  
+3. ຖ້າຕ້ອງການລົບ: ເລືອກ Key ແລ້ວກົດ **🗑️ Delete**
 """
             )
 
+            # Input boxes
             q_box = gr.Textbox(
                 label="ຄໍາຖາມ (ພາສາລາວ)",
                 lines=2,
@@ -81,20 +208,79 @@ def create_app():
                 placeholder="ຕົວຢ່າງ: ປະຫວັດສາດມີຄວາມສໍາຄັນເພາະ...",
             )
 
-            save_btn = gr.Button("💾 Save Q&A")
+            editing_key = gr.State("")  # current selected key for edit/delete
+
+            with gr.Row():
+                save_btn = gr.Button("💾 Save Q&A (Add / Overwrite)")
+                update_btn = gr.Button("✏️ Update current")
+                delete_btn = gr.Button("🗑️ Delete current")
+
             status = gr.Markdown()
 
+            gr.Markdown("### 📋 Manual Q&A List")
+
+            manual_df = gr.Dataframe(
+                headers=[
+                    "Key (normalized)",
+                    "Question (Lao)",
+                    "Answer (Lao)",
+                ],
+                interactive=False,
+                wrap=True,
+                row_count=(0, "dynamic"),
+                label="ລາຍການ Q&A ທີ່ chatbot ໃຊ້ຢູ່",
+            )
+
+            key_dropdown = gr.Dropdown(
+                label="ເລືອກ Q&A ເພື່ອແກ້ໄຂ / ລົບ (Key)",
+                choices=[],
+                interactive=True,
+            )
+
+            load_btn = gr.Button("📥 Load selected for edit")
+
+            # --- Wiring events ---
+
+            # Load selected row into text boxes
+            load_btn.click(
+                fn=load_for_edit,
+                inputs=key_dropdown,
+                outputs=[q_box, a_box, editing_key],
+            )
+
+            # Save new Q&A (or overwrite) and refresh table
             save_btn.click(
                 fn=save_teacher_qa,
                 inputs=[q_box, a_box],
-                outputs=status,
+                outputs=[status, manual_df, key_dropdown, editing_key],
+            )
+
+            # Update existing Q&A
+            update_btn.click(
+                fn=update_teacher_qa,
+                inputs=[editing_key, q_box, a_box],
+                outputs=[status, manual_df, key_dropdown, editing_key],
+            )
+
+            # Delete existing Q&A
+            delete_btn.click(
+                fn=delete_teacher_qa,
+                inputs=[editing_key],
+                outputs=[status, manual_df, key_dropdown, editing_key, q_box, a_box],
             )
 
             gr.Markdown(
                 """
-> ℹ️ ໝາຍເຫດ: Q&A ທີ່ເພີ່ມໃໝ່ຈະຖືກບັນທຶກໃນ `data/manual_qa.jsonl`.  
-> ມັນຈະໃຊ້ງານໄດ້ທັນທີໃນ chatbot ໂດຍບໍ່ຕ້ອງ restart Space.
+> ℹ️ ໝາຍເຫດ: Q&A ທັງໝົດຈະຖືກບັນທຶກໃນ `data/manual_qa.jsonl`.  
+> ເມື່ອເຈົ້າກົດ Save / Update / Delete chatbot ຈະອັບເດດທັນທີ (ບໍ່ຕ້ອງ restart Space).
 """
             )
 
+        # On app load, fill table + dropdown
+        demo.load(
+            fn=_refresh_manual_data,
+            inputs=None,
+            outputs=[manual_df, key_dropdown],
+        )
+
     return demo