Update app.py

app.py

@@ -64,7 +64,7 @@ assistant = st.chat_message("assistant")
 assistant.write("How can I help you today?")
 
 def assistant_response(response):
-    message_placeholder = st.empty()
+    message_placeholder = assistant.empty()
     full_response = ""
     assistant_response = response
     # Simulate stream of response with milliseconds delay
@@ -80,7 +80,9 @@ def assistant_response(response):
 if prompt := st.chat_input("What kind of class are you looking for?"):
     with st.chat_message("user"):
         st.markdown(prompt)
+    st.sessionstate.messages.append({"role": "user", "content": prompt})
     assistant_response("Yah I'm tired af right now boi")
+    
 
 
 # if prompt := st.chat_input("What kind of class are you looking for?"):