Update app.py

app.py

@@ -60,7 +60,8 @@ if "x" not in st.session_state:
 
 # Function to handle iteration movement
 def update_iteration(step):
-    if st.session_state.iteration + step >= 0 and st.session_state.iteration + step < len(st.session_state.x_vals):
+    """Moves to next or previous iteration when the buttons are clicked."""
+    if 0 <= st.session_state.iteration + step < len(st.session_state.x_vals):
         st.session_state.iteration += step
         st.session_state.x = st.session_state.x_vals[st.session_state.iteration]
         st.session_state.y_vals = [safe_eval(func_input, st.session_state.x)]
@@ -81,6 +82,20 @@ with col3:
     if st.button("⏩ Next", key="next", on_click=update_iteration, args=(1,)):
         pass
 
+# Buttons to run descent step and reset
+col4, col5 = st.columns([1, 1])
+with col4:
+    if st.button("🔄 Run Descent Step"):
+        grad = derivative(func_input, st.session_state.x)
+        st.session_state.x -= learning_rate * grad
+        st.session_state.iteration += 1
+        st.session_state.x_vals.append(st.session_state.x)
+        st.session_state.y_vals.append(safe_eval(func_input, st.session_state.x))
+
+with col5:
+    if st.button("🔄 Reset", on_click=reset_state):
+        pass
+
 # Plotting
 x_range = np.linspace(-2, 6, 400)  # Zoomed-in x-axis range for better visualization
 y_range = [safe_eval(func_input, x) for x in x_range]