Update app.py

app.py

@@ -165,7 +165,7 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 
 if "curr_database" not in st.session_state:
-    st.session_state.curr_database = None
+    st.session_state.curr_domain = ""
 
 init_messages = {
     "About YSA": '''
@@ -201,14 +201,12 @@ with st.spinner("Loading, please be patient with us ... 🙏"):
     )
     # db=collection
     
-    if not st.session_state.curr_database.equals(dataset):
+    if st.session_state.curr_domain != domain:
         print("True")
         st.session_state.messages = []
         init_message = init_messages[domain]
         st.session_state.messages.append({"role": "assistant", "content": init_message})
-
-# Current database variable
-st.session_state.curr_database = dataset
+        st.session_state.curr_domain = domain
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages: