Update app.py

app.py

@@ -227,22 +227,22 @@ init_messages = {
 }
 
 # Embed and store the first N supports for this demo
-    with st.spinner("Loading, please be patient with us ... 🙏"):
-        L = len(dataset["train"]["questions"])
+with st.spinner("Loading, please be patient with us ... 🙏"):
+    L = len(dataset["train"]["questions"])
+    
+    collection.add(
+        ids=[str(i) for i in range(0, L)],  # IDs are just strings
+        documents=dataset["train"]["questions"],  # Enter questions here
+        metadatas=[{"type": "support"} for _ in range(0, L)],
+    )
+
+    if st.session_state.curr_domain != domain:
+        st.session_state.messages = []
+        
+        init_message = init_messages[domain]
+        st.session_state.messages.append({"role": "assistant", "content": init_message})
         
-        collection.add(
-            ids=[str(i) for i in range(0, L)],  # IDs are just strings
-            documents=dataset["train"]["questions"],  # Enter questions here
-            metadatas=[{"type": "support"} for _ in range(0, L)],
-        )
-
-        if st.session_state.curr_domain != domain:
-            st.session_state.messages = []
-            
-            init_message = init_messages[domain]
-            st.session_state.messages.append({"role": "assistant", "content": init_message})
-            
-            st.session_state.curr_domain = domain
+        st.session_state.curr_domain = domain
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages: