Update app.py

app.py

@@ -94,19 +94,6 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-if "retry" not in st.session_state:
-    st.session_state.retry= False
-
-#def retry_click():
-#    st.session_state.retry= True
-
-if st.session_state.retry:
-    lastmessage = st.session_state.messages.pop()
-    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
-    st.rerun()
 
 if "remove" not in st.session_state:
     st.session_state.remove= False
@@ -118,8 +105,7 @@ 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()
+#    st.rerun()
 
 
 
@@ -142,9 +128,7 @@ 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:
-    col1, col2 = st.columns(2)
-    col1.button("retry", key="retry")
-    col2.button("remove", key="remove")
+    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)