Update app.py

app.py

@@ -70,17 +70,6 @@ def get_assistant_aswer(st_model, st_messages, st_temp_value, st_max_tokens):
 
     return response
 
-def retry_last():
-    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)
-#    with st.chat_message("assistant"):
-#        st.write(response)
-#        st.button("retry", on_click=retry_last())
-#    st.session_state.messages.append({"role": "assistant", "content": response})
-    return None
-
-
 # Define the available models
 models =[key for key in model_links.keys()]
 
@@ -138,8 +127,16 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 
-if st.button("retry", key=retry):
-    retry_last()
+if st.button("retry", key="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)
+#    with st.chat_message("assistant"):
+#        st.write(response)
+#        st.button("retry", on_click=retry_last())
+#    st.session_state.messages.append({"role": "assistant", "content": response})
+
+
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):