Update app.py

app.py

@@ -64,14 +64,8 @@ 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
-
-#if st.session_state.prev_option != selected_model:
-#    st.session_state.messages = []
-    # st.write(f"Changed to {selected_model}")
+#if "prev_option" not in st.session_state:
 #    st.session_state.prev_option = selected_model
-#    reset_conversation()
 
 st.subheader(f'{selected_model}')
 
@@ -81,7 +75,9 @@ if "messages" not in st.session_state:
 
 
 def remove_message(position):
-    st.toast("try to remove message no: " + str(position) + " and "+ str(position+1))
+    st.toast("try to remove message no: " + str(position-1) + " and "+ str(position))
+    del st.session_state.messages[position-1:position]
+    
 
 if "remove" not in st.session_state:
     st.session_state.remove= False