Update src/streamlit_app.py

src/streamlit_app.py

@@ -101,14 +101,21 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# Main chat input logic
 if prompt := st.chat_input("Ask about fee deadlines, scholarships, etc."):
+    # Add user message to session state and display it
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
 
+    # Get and display bot response
     with st.chat_message("assistant"):
-        with st.spinner("Searching Notion..."):
+        with st.spinner("Thinking..."):
+            # 1. Retrieve context
             context = get_context(prompt)
+            # 2. Generate response
             response = generate_response(prompt, context)
+            # 3. Display response INSIDE the bubble
             st.markdown(response)
+            # 4. Add the response to the history for the next re-run
             st.session_state.messages.append({"role": "assistant", "content": response})