Update app.py

app.py

@@ -124,64 +124,27 @@ 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()
+
+
+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)