Update app.py

app.py

@@ -89,10 +89,17 @@ if selected_model not in st.session_state:
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
+def remove_message(position):
+    st.toast("remove message no:" + position)
+
 # Display chat messages from history on app rerun
+pos = 0
 for message in st.session_state.messages:
+    pos++
     with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+        cols = st.columns([9,1])
+        cols[1].markdown(message["content"])
+        cols[2].button("remove", key=button_remove_message, args=[pos], on_click=remove_message)
 
 
 if "remove" not in st.session_state: