add compare trigger

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from chatbot.core import get_chat_response
+from chatbot.core import get_chat_response, memory, cmp
 
 # Cấu hình giao diện Streamlit
 st.set_page_config(page_title="RangDong Chatbot", layout="wide")
@@ -29,8 +29,10 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # Ô nhập liệu chat
+
 user_input = st.chat_input("Nhập tin nhắn của bạn...")
 
+
 if user_input:
     # Hiển thị tin nhắn của user
     st.session_state.messages.append({"role": "user", "content": user_input})
@@ -44,3 +46,13 @@ if user_input:
     st.session_state.messages.append({"role": "assistant", "content": response})
     with st.chat_message("assistant"):
         st.markdown(response)
+
+if memory.cache:
+    cmp_button = st.button("So sánh các sản phẩm tương đồng")
+
+    if cmp_button:
+        response = cmp()
+
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        with st.chat_message("assistant"):
+            st.markdown(response)

chatbot/core.py

@@ -49,7 +49,6 @@ qa_chain2 = MyCustomMemoryRetrievalChain(
 #     output_key="result"
 # )
 
-
 def get_chat_response(user_input: str) -> str:
     restriction = classify_query(user_input.strip())
 
@@ -68,3 +67,9 @@ def get_chat_response(user_input: str) -> str:
     memory.add(user_input, response["result"])
     print(memory.restrict)
     return response["result"]
+
+
+def cmp():
+    prompt = "So sánh các sản phẩm trên"
+    response = qa_chain2({"question": prompt + memory.restrict , "memory": memory.get_memory_text()})
+    return response["result"]