Update app.py

app.py

@@ -48,7 +48,8 @@ def reset_conversation():
     st.session_state.messages = []
     return None
     
-
+def retry_last():
+    return None
 
 
 # Define the available models
@@ -133,6 +134,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             )
     
             response = st.write_stream(stream)
+            st.button("retry", on_click=retry_last())
 
         except Exception as e:
             # st.empty()
@@ -153,4 +155,5 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
 
             
 
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})
+