Update app.py

app.py

@@ -64,8 +64,9 @@ st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
 st.sidebar.write(f"You're now chatting with **{selected_model}**")
 st.sidebar.markdown("*Generated content may be inaccurate or false.*")
 
-#if "prev_option" not in st.session_state:
-#    st.session_state.prev_option = selected_model
+def remove_message(position):
+    st.toast("try to remove message no: " + str(position-1) + " and "+ str(position))
+    del st.session_state.messages[position-2:position]
 
 st.subheader(f'{selected_model}')
 
@@ -73,17 +74,6 @@ st.subheader(f'{selected_model}')
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-
-def remove_message(position):
-    st.toast("try to remove message no: " + str(position-1) + " and "+ str(position))
-    del st.session_state.messages[position-2:position]
-
-if "remove" not in st.session_state:
-    st.session_state.remove= False
-
-def remove_click():
-    st.session_state.remove= True
-
 # Display chat messages from history on app rerun
 pos = 0
 for message in st.session_state.messages: