Update app.py

app.py

@@ -75,6 +75,11 @@ def get_assistant_aswer(st_model, st_messages, st_temp_value, st_max_tokens):
     return response
 
 def retry_last():
+    response = get_assistant_aswer(model_links[selected_model], st.session_state.messages[:-1], 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