Update app.py

app.py

@@ -139,6 +139,7 @@ if st.session_state.retry:
     response = get_assistant_aswer(model_links[selected_model], st.session_state.messages, temp_values, max_token_value)
     st.session_state.messages.append({"role": "assistant", "content": response})
     st.session_state.retry= False
+    st.rerun()
 
 if "remove" not in st.session_state:
     st.session_state.remove= False
@@ -150,7 +151,8 @@ 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= True
+    st.session_state.remove= False
+    st.rerun()