Update ui.py

ui.py

@@ -1,23 +1,100 @@
-# ui.py – Gradio UI
+# ui.py – Gradio UI (Student + Teacher tabs)
 
 import gradio as gr
+
 from rag import answer_question
+from data.manual_answers import add_manual_qa
+
 
+# ========= Student side =========
 
 def chat_fn(message: str, history: list):
     # history is ignored for now
     return answer_question(message)
 
 
+# ========= Teacher side =========
+
+def save_teacher_qa(question: str, answer: str) -> str:
+    """
+    Called when teacher clicks 'Save Q&A'.
+    """
+    if not question.strip() or not answer.strip():
+        return "❗ ກະລຸນາພິມທັງຄໍາຖາມ ແລະຄໍາຕອບ."
+
+    try:
+        norm_key = add_manual_qa(question, answer)
+        return (
+            "✅ ບັນທຶກສໍາເລັດ! ຈາກນີ້ chatbot ຈະໃຊ້ຄໍາຕອບນີ້ທັນທີ.\n"
+            f"🔑 ຄີທີ່ໃຊ້ຄົ້ນຫາ (normalized): `{norm_key}`"
+        )
+    except Exception as e:
+        return f"⚠️ ບັນທຶກບໍ່ສໍາເລັດ: {e}"
+
+
 def create_app():
-    return gr.ChatInterface(
-        fn=chat_fn,
-        title="Laos History Chatbot (Lao language)",
-        description="ຖາມຂໍ້ມູນກ່ຽວກັບປະຫວັດສາດຂອງປະເທດລາວ (ສໍາລັບນັກຮຽນ ມ.1)",
-        examples=[
-            "ປະຫວັດສາດແມ່ນຫຍັງ?",
-            "ປະຫວັດສາດມີຄວາມສໍາຄັນຈັ່ງໃດ໋?",
-            "ຫຼັກຖານປະຫວັດສາດມີຫຍັງແດ່?",
-            "ປະຕິທິນຈັນທະລະຄະຕິແມ່ນຫຍັງ?",
-        ],
-    )
+    with gr.Blocks() as demo:
+        gr.Markdown(
+            """
+# 📚 Laos History Chatbot (M1)
+
+- 👩‍🎓 **Student Chat**: ນັກຮຽນຖາມຄໍາຖາມປະຫວັດສາດພາສາລາວ  
+- 👩‍🏫 **Teacher Panel**: ຄູເພີ່ມ Q&A ທີ່ຖືກຕ້ອງເອງ ເພື່ອໃຫ້ chatbot ຕອບໄດ້ດີຂຶ້ນ
+"""
+        )
+
+        # ---------- Tab 1: Student Chat ----------
+        with gr.Tab("👩‍🎓 Student Chat"):
+            gr.ChatInterface(
+                fn=chat_fn,
+                title="Laos History Chatbot (Lao language)",
+                description="ຖາມຂໍ້ມູນກ່ຽວກັບປະຫວັດສາດຂອງປະເທດລາວ (ສໍາລັບນັກຮຽນ ມ.1)",
+                examples=[
+                    "ປະຫວັດສາດແມ່ນຫຍັງ?",
+                    "ປະຫວັດສາດມີຄວາມສໍາຄັນຈັ່ງໃດ໋?",
+                    "ຫຼັກຖານປະຫວັດສາດມີຫຍັງແດ່?",
+                    "ປະຕິທິນຈັນທະລະຄະຕິແມ່ນຫຍັງ?",
+                ],
+            )
+
+        # ---------- Tab 2: Teacher Panel ----------
+        with gr.Tab("👩‍🏫 Teacher Panel"):
+            gr.Markdown(
+                """
+### ✍️ ເພີ່ມ Q&A ສໍາລັບຄູ
+
+- ພິມ **ຄໍາຖາມ** ແລະ **ຄໍາຕອບ** ພາສາລາວ  
+- ກົດ **"💾 Save Q&A"**  
+- Chatbot ຈະໃຊ້ຄໍາຕອບນີ້ທັນທີເມື່ອນັກຮຽນຖາມຄໍາຖາມຄ້າຍກັນ
+"""
+            )
+
+            q_box = gr.Textbox(
+                label="ຄໍາຖາມ (ພາສາລາວ)",
+                lines=2,
+                placeholder="ຕົວຢ່າງ: ປະຫວັດສາດມີຄວາມສໍາຄັນຈັ່ງໃດ໋?",
+            )
+
+            a_box = gr.Textbox(
+                label="ຄໍາຕອບ (ພາສາລາວ)",
+                lines=4,
+                placeholder="ຕົວຢ່າງ: ປະຫວັດສາດມີຄວາມສໍາຄັນເພາະ...",
+            )
+
+            save_btn = gr.Button("💾 Save Q&A")
+            status = gr.Markdown()
+
+            save_btn.click(
+                fn=save_teacher_qa,
+                inputs=[q_box, a_box],
+                outputs=status,
+            )
+
+            gr.Markdown(
+                """
+> ℹ️ ໝາຍເຫດ: Q&A ທີ່ເພີ່ມໃໝ່ຈະຖືກບັນທຶກໃນ `data/manual_qa.jsonl`.  
+> ມັນຈະໃຊ້ງານໄດ້ທັນທີໃນ chatbot ໂດຍບໍ່ຕ້ອງ restart Space.
+"""
+            )
+
+    return demo