Update app.py

app.py

@@ -59,9 +59,6 @@ units = st.slider(
 #         st.markdown(message["content"])
 
 
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
 
 assistant = st.chat_message("assistant")
 initial_message = "How can I help you today?"
@@ -84,6 +81,10 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
     assistant_response(initial_message)
 
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
 if prompt := st.chat_input("What kind of class are you looking for?"):
     with st.chat_message("user"):
         st.markdown(prompt)