Update pages/chatbot.py

pages/chatbot.py

@@ -2,7 +2,7 @@ import streamlit as st
 from app import rag_query, process_feedback
 
 
-st.title("RAG Chatbot")
+st.title("🛡️ RAG Chatbot 🛡️")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -12,16 +12,16 @@ if "messages" not in st.session_state:
 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)
-                   # st.session_state.messages.pop()  # Remove the last assistant message
-                    #st.rerun()  # Rerun the app to regenerate the response
+        # 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)
+        #            # st.session_state.messages.pop()  # Remove the last assistant message
+        #             #st.rerun()  # Rerun the app to regenerate the response
 
 # React to user input
 if prompt := st.chat_input("Bạn cần tư vấn về điều gì? Hãy chia sẻ nhu cầu và thông tin của bạn nhé!"):
@@ -38,8 +38,8 @@ if prompt := st.chat_input("Bạn cần tư vấn về điều gì? Hãy chia s
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
 
-    # Rerun the app to display the feedback buttons
-    st.experimental_rerun()
+    # # Rerun the app to display the feedback buttons
+    # st.rerun()
 
 # Sidebar for additional controls
 with st.sidebar: