Update app.py

app.py

@@ -98,14 +98,14 @@ for message in st.session_state.messages:
 if "remove" not in st.session_state:
     st.session_state.remove= False
 
-#def remove_click():
-#    st.session_state.remove= True
+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.rerun()
+    st.rerun()
 
 
 
@@ -128,7 +128,4 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     st.session_state.messages.append({"role": "assistant", "content": response})
 
 if len(st.session_state.messages)>0:
-    st.button("remove last message", key="remove")
-
-#    col1.button("retry", key="retryButton", on_click=retry_click)
-#    col2.button("remove", key="removeButton", on_click=remove_click)
+    st.button("remove last conversation", key="removeButton", on_click=remove_click)