Update app.py

app.py

@@ -60,8 +60,12 @@ units = st.slider(
 #     with st.chat_message(message["role"]):
 #         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")
-assistant.write("How can I help you today?")
+initial_message = "How can I help you today?"
 
 def assistant_response(response):
     message_placeholder = assistant.empty()
@@ -77,6 +81,8 @@ def assistant_response(response):
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": full_response})
 
+assistant_response(initial_message)
+
 if prompt := st.chat_input("What kind of class are you looking for?"):
     with st.chat_message("user"):
         st.markdown(prompt)