Update app.py

app.py

@@ -124,17 +124,64 @@ def process_feedback(query, response, feedback):
 
 # Streamlit interface
 
-st.title("Chào mừng bạn đã đến với MBAL Chatbot")
-st.markdown("***")
-st.info('''
-        Tôi sẽ giải đáp các thắc mắc của bạn liên quan đến các sản phẩm bảo hiểm nhân thọ của MB Ageas Life''')
+# st.title("Chào mừng bạn đã đến với MBAL Chatbot")
+# st.markdown("***")
+# st.info('''
+#         Tôi sẽ giải đáp các thắc mắc của bạn liên quan đến các sản phẩm bảo hiểm nhân thọ của MB Ageas Life''')
 
-col1, col2 = st.columns(2)
+# col1, col2 = st.columns(2)
 
-with col1:
-    chat = st.button("Chat")
-    if chat:
-        st.switch_page("pages/chatbot.py")
+# with col1:
+#     chat = st.button("Chat")
+#     if chat:
+#         st.switch_page("pages/chatbot.py")
+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()
 
 
 st.markdown("<div style='text-align:center;'></div>", unsafe_allow_html=True)