Update app.py

app.py

@@ -112,10 +112,12 @@ def get_rag_results(prompt):
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
-    with st.chat_message("assistant"):
-        st.markdown(initial_message)
     st.session_state.messages.append({"role": "assistant", "content": 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?"):
     st.session_state.messages.append({"role": "user", "content": prompt})