Update app.py

app.py

@@ -75,7 +75,8 @@ def get_assistant_aswer(st_model, st_messages, st_temp_value, st_max_tokens):
     return response
 
 def retry_last():
-    st.session_state.messages.pop()
+    lastmessage = st.session_state.messages.pop()
+    st.info(lastmessage[content])
 #    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)
@@ -152,8 +153,8 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     
     # Display assistant response in chat message container
     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})
     with st.chat_message("assistant"):
         st.write(response)
         st.button("retry", on_click=retry_last())
-    st.session_state.messages.append({"role": "assistant", "content": response})