Update app.py

app.py

@@ -185,7 +185,7 @@ with st.spinner("Loading, please be patient with us ... 🙏"):
         metadatas=[{"type": "support"} for _ in range(0, L)],
     )
     db=collection
-    if curr_database:
+    if (curr_database in st.session_state) & (st.session_state.curr_database == dataset):
         st.write("True")
     else:
         st.write("False")
@@ -194,7 +194,7 @@ with st.spinner("Loading, please be patient with us ... 🙏"):
     # st.session_state.messages.append({"role": "assistant", "content": init_message})
 
 # Current database variable
-curr_database = collection
+st.session_state.curr_database = dataset
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages: