Update app.py

app.py

@@ -148,9 +148,9 @@ with st.chat_message("assistant"):
 # React to user input
 if user_message := st.chat_input("Enter message"):
     # Display user message in chat message container
-    st.chat_message("user").markdown(prompt)
+    st.chat_message("user").markdown(user_message)
     # Add user message to chat history
-    st.session_state.messages.append({"role": "user", "content": prompt})
+    st.session_state.messages.append({"role": "user", "content": user_message})
 
     response = handle_submission
     # Display the agent's response