Update pages/chatbot.py

pages/chatbot.py

@@ -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.rerun()
+    # Rerun the app to display the feedback buttons
+    st.rerun()
 
 # Sidebar for additional controls
 with st.sidebar: