Update app.py

app.py

@@ -127,7 +127,7 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 if "retry" not in st.session_state:
-        st.session_state.retry= False
+    st.session_state.retry= False
 
 def retry_click():
     st.session_state.retry= True
@@ -138,18 +138,19 @@ if st.session_state.retry:
     st.toast("popped msg: " + lastmessage["content"] + " // model: " + model_links[selected_model])
     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
 
 if "remove" not in st.session_state:
-        st.session_state.remove= False
+    st.session_state.remove= False
 
 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= True