Update app.py

app.py

@@ -98,7 +98,7 @@ with col1:
     learning_rate = st.number_input("Learning Rate:", value=0.1, step=0.01, format="%.2f", key="learning_rate", on_change=reset_session)
 
     # Buttons for actions with styling
-    col1.button("🔄 Next Step", key="next", use_container_width=True, help="Perform one step of gradient descent", on_click=None, args=None, kwargs=None)
+    col1.button("🔄 Next Step", key="next", use_container_width=True, help="Perform one step of gradient descent", on_click=None)
     col1.button("🔁 Reset", key="reset", use_container_width=True, help="Reset the gradient descent", on_click=reset_session)
     iteration_data_button = col1.button("📊 Iteration Data", key="iteration_data", use_container_width=True)
 
@@ -143,6 +143,22 @@ with col2:
 
     st.plotly_chart(plot)
 
+# Perform gradient descent operation when button is clicked
+if st.button("🔄 Next Step", key="next", use_container_width=True):
+    try:
+        grad = compute_derivative(st.session_state.func_input, st.session_state.x_current)
+        st.session_state.x_current = st.session_state.x_current - st.session_state.learning_rate * grad
+        st.session_state.iter_count += 1
+
+        # Add current iteration data
+        st.session_state.iter_data.append({
+            "Iteration": st.session_state.iter_count,
+            "x": st.session_state.x_current,
+            "f(x)": evaluate_function(st.session_state.func_input, st.session_state.x_current),
+        })
+    except Exception as e:
+        st.error(f"Error: {e}")
+
 # Show iteration data in a new section when the button is clicked
 if iteration_data_button:
     st.subheader("📊 Iteration Data")