Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 from streamlit_option_menu import option_menu
 from markup import app_intro
 import langchain
-from langchain.cache import InMemoryCache
 from query_data import chat_chain
+from memory import msgs
+
 
-langchain.llm_cache = InMemoryCache()
 
 def tab1():
     st.header("CIMA Chatbot")
@@ -21,17 +21,13 @@ unique_metadata_list = []
 seen = set()
 
 def tab4():    
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
-
     st.header("🗣️ Chat with the AI about the ingested documents! 📚")
 
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+    for i, msg in enumerate(msgs.messages):
+        name = "user" if i % 2 == 0 else "assistant"
+        st.chat_message(name).markdown(msg.content)
 
     if user_input := st.chat_input("User Input"):
-        st.session_state.messages.append({"role": "user", "content": user_input})
 
         with st.chat_message("user"):
             st.markdown(user_input)
@@ -57,8 +53,6 @@ def tab4():
 
                 st.write(answer)
                 st.write(unique_metadata_list)
-                
-            st.session_state.messages.append({"role": "assistant", "content": answer})
 
 
 def main():