Update app.py

app.py

@@ -132,11 +132,13 @@ def answer_question(question, history):
     if not question.strip():
         return history, ""
     if collection.count() == 0:
-        history.append([question, "⚠️ Chưa có tài liệu. Vui lòng upload trước."])
+        history.append({"role": "user",      "content": question})
+        history.append({"role": "assistant", "content": "⚠️ Chưa có tài liệu. Vui lòng upload trước."})
         return history, ""
     chunks = retrieve(question)
     if not chunks:
-        history.append([question, "Không tìm thấy thông tin liên quan."])
+        history.append({"role": "user",      "content": question})
+        history.append({"role": "assistant", "content": "Không tìm thấy thông tin liên quan."})
         return history, ""
     context = "\n\n---\n\n".join(f"[{c['source']}]\n{c['text']}" for c in chunks)
     system  = ("Bạn là trợ lý AI. Trả lời câu hỏi DỰA TRÊN ngữ cảnh. "
@@ -151,7 +153,8 @@ def answer_question(question, history):
     answer  = outputs[0]["generated_text"].strip()
     sources = list({c["source"] for c in chunks})
     src_text = "  \n📄 *Nguồn: " + " · ".join(sources) + "*"
-    history.append([question, answer + src_text])
+    history.append({"role": "user",      "content": question})
+    history.append({"role": "assistant", "content": answer + src_text})
     return history, ""
 
 def reset_index():
@@ -322,7 +325,7 @@ SUGS = [
 ]
 
 # ── UI ────────────────────────────────────────────────────────────────────────
-with gr.Blocks(title="RAG · Qwen2.5-0.5B") as demo:
+with gr.Blocks(title="RAG · Qwen2.5-0.5B", theme=gr.themes.Base(), css=CSS) as demo:
 
     # ── Sidebar ───────────────────────────────────────────────────────────────
     with gr.Sidebar(elem_classes="sidebar-content"):
@@ -356,6 +359,7 @@ with gr.Blocks(title="RAG · Qwen2.5-0.5B") as demo:
             height=460,
             show_label=False,
             render_markdown=True,
+            type="messages",
         )
 
     with gr.Row(elem_classes="sug-row"):
@@ -383,4 +387,4 @@ with gr.Blocks(title="RAG · Qwen2.5-0.5B") as demo:
     demo.load(fn=get_count, outputs=chunk_vis)
 
 if __name__ == "__main__":
-    demo.launch(css=CSS, theme=gr.themes.Base(), server_name="0.0.0.0", ssr_mode=False)
+    demo.launch(server_name="0.0.0.0", ssr_mode=False)