Update app.py

app.py

@@ -76,8 +76,7 @@ if "messages" not in st.session_state:
 
 def remove_message(position):
     st.toast("try to remove message no: " + str(position-1) + " and "+ str(position))
-    del st.session_state.messages[position-1:position]
-    
+    del st.session_state.messages[position-2:position]
 
 if "remove" not in st.session_state:
     st.session_state.remove= False
@@ -85,14 +84,6 @@ if "remove" not in st.session_state:
 def remove_click():
     st.session_state.remove= True
 
-if st.session_state.remove:
-    lastmessage = st.session_state.messages.pop()
-    prelastmessage = st.session_state.messages.pop()
-    st.toast("popped msg: " + lastmessage["content"] + " // model: " + model_links[selected_model])
-    st.session_state.remove = False
-    st.rerun()
-
-
 # Display chat messages from history on app rerun
 pos = 0
 for message in st.session_state.messages:
@@ -128,6 +119,3 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             col2.button("remove-"+str(pos), key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
 
     st.session_state.messages.append({"role": "assistant", "content": response})
-
-if len(st.session_state.messages)>0:
-    st.button("remove last conversation", key="removeButton", on_click=remove_click)