spinner organized

app.py

@@ -86,7 +86,7 @@ collection = client.create_collection(combined_string)
 
 
 # Embed and store the first N supports for this demo
-with st.spinner("Loading database, please be patient with us ... 🙏"):
+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
@@ -97,13 +97,13 @@ with st.spinner("Loading database, please be patient with us ... 🙏"):
 
 # React to user input
 if prompt := st.chat_input("Tell me about YSA"):
-    # Display user message in chat message container
-    st.chat_message("user").markdown(prompt)
-    # Add user message to chat history
-    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.spinner("Loading, please be patient with us ... 🙏"):
+        # Display user message in chat message container
+        st.chat_message("user").markdown(prompt)
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
 
-    question = prompt
-    with st.spinner("Thinking ..."):
+        question = prompt
         results = collection.query(query_texts=question, n_results=5)
         idx = results["ids"][0]
         idx = [int(i) for i in idx]