Update pages/chatbot.py

pages/chatbot.py

@@ -1,47 +1,47 @@
-# import streamlit as st
-# from app import rag_query, memory, process_feedback
-
-# st.title("🛡️ RAG Chatbot MB Ageas Life 🛡️")
-
-# # Initialize session history
-# if "messages" not in st.session_state:
-#     st.session_state.messages = []
-
-# # Hiển thị lại tin nhắn cũ
-# for i, message in enumerate(st.session_state.messages):
-#     with st.chat_message(message["role"]):
-#         st.markdown(message["content"])
-#         if message["role"] == "assistant":
-#             col1, col2 = st.columns([1, 15])
-#             with col1:
-#                 if st.button("👍", key=f"thumbs_up_{i}"):
-#                     process_feedback(st.session_state.messages[i-1]["content"], message["content"], True)
-#             with col2:
-#                 if st.button("👎", key=f"thumbs_down_{i}"):
-#                     process_feedback(st.session_state.messages[i-1]["content"], message["content"], False)
-
-# # Nhận input người dùng
-# if prompt := st.chat_input("Ask me any question related to MBAL"):
-#     # Hiển thị tin nhắn người dùng
-#     st.chat_message("user").markdown(prompt)
-#     st.session_state.messages.append({"role": "user", "content": prompt})
-#     memory.chat_memory.add_user_message(prompt)
-
-#     # Gọi hàm RAG để trả lời
-#     response = rag_query(prompt)
-
-#     # Hiển thị tin nhắn chatbot
-#     with st.chat_message("assistant"):
-#         st.markdown(response)
-#     st.session_state.messages.append({"role": "assistant", "content": response})
-#     memory.chat_memory.add_ai_message(response)
-
-#     st.rerun()
-
-# # Sidebar
-# # with st.sidebar:
-# #     st.header("Lựa chọn khác")
-# #     if st.button("Xóa lịch sử chat"):
-# #         st.session_state.messages = []
-# #         memory.clear()
-#         # st.rerun()
+import streamlit as st
+from app import rag_query, memory, process_feedback
+
+st.title("🛡️ RAG Chatbot MB Ageas Life 🛡️")
+
+# Initialize session history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Hiển thị lại tin nhắn cũ
+for i, message in enumerate(st.session_state.messages):
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+        if message["role"] == "assistant":
+            col1, col2 = st.columns([1, 15])
+            with col1:
+                if st.button("👍", key=f"thumbs_up_{i}"):
+                    process_feedback(st.session_state.messages[i-1]["content"], message["content"], True)
+            with col2:
+                if st.button("👎", key=f"thumbs_down_{i}"):
+                    process_feedback(st.session_state.messages[i-1]["content"], message["content"], False)
+
+# Nhận input người dùng
+if prompt := st.chat_input("Ask me any question related to MBAL"):
+    # Hiển thị tin nhắn người dùng
+    st.chat_message("user").markdown(prompt)
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    memory.chat_memory.add_user_message(prompt)
+
+    # Gọi hàm RAG để trả lời
+    response = rag_query(prompt)
+
+    # Hiển thị tin nhắn chatbot
+    with st.chat_message("assistant"):
+        st.markdown(response)
+    st.session_state.messages.append({"role": "assistant", "content": response})
+    memory.chat_memory.add_ai_message(response)
+
+    st.rerun()
+
+# Sidebar
+# with st.sidebar:
+#     st.header("Lựa chọn khác")
+#     if st.button("Xóa lịch sử chat"):
+#         st.session_state.messages = []
+#         memory.clear()
+        # st.rerun()