Update app.py

app.py

@@ -81,16 +81,18 @@ 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)
     st.session_state.messages.append({"role": "user", "content": prompt})
     assistant_response("Yah I'm tired af right now boi")
     
+    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?"):