Update app.py

app.py

@@ -129,18 +129,18 @@ def rag_query(query):
 
     return response
 
-# def process_feedback(query, response, feedback):
-#    # st.write(f"Feedback received: {'👍' if feedback else '👎'} for query: {query}")
-#     if feedback:
-#         # If thumbs up, store the response in memory buffer
-#         memory.chat_memory.add_ai_message(response)
-#     else:
-#         # If thumbs down, remove the response from memory buffer and regenerate the response
-#        # memory.chat_memory.messages = [msg for msg in memory.chat_memory.messages if msg.get("content") != response]
-#         new_query=f"{query}. Give better response"
-#         new_response = rag_query(new_query)
-#         st.markdown(new_response)
-#         memory.chat_memory.add_ai_message(new_response)
+def process_feedback(query, response, feedback):
+   # st.write(f"Feedback received: {'👍' if feedback else '👎'} for query: {query}")
+    if feedback:
+        # If thumbs up, store the response in memory buffer
+        memory.chat_memory.add_ai_message(response)
+    else:
+        # If thumbs down, remove the response from memory buffer and regenerate the response
+       # memory.chat_memory.messages = [msg for msg in memory.chat_memory.messages if msg.get("content") != response]
+        new_query=f"{query}. Give better response"
+        new_response = rag_query(new_query)
+        st.markdown(new_response)
+        memory.chat_memory.add_ai_message(new_response)
 
 # Streamlit interface
 
@@ -149,11 +149,11 @@ 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:
+with col1:
+    chat = st.button("Chat")
+    if chat:
 st.switch_page("pages/chatbot.py")