Update app.py

app.py

@@ -27,9 +27,12 @@ def calculate_tangent(expression, x_value, x_range):
 
 # Reset state
 def reset_session_state():
+    """Resets the session state for a fresh start."""
     st.session_state.x_current = st.session_state.initial_point
     st.session_state.iter_count = 0
-    st.session_state.history = [(st.session_state.initial_point, evaluate_function(st.session_state.math_function, st.session_state.initial_point))]
+    st.session_state.history = [
+        (st.session_state.initial_point, evaluate_function(st.session_state.math_function, st.session_state.initial_point))
+    ]
     st.session_state.current_index = 0
 
 # Initialize session state variables
@@ -59,9 +62,10 @@ with left_col:
     initial_point = st.number_input(
         "Initial Value of x", value=4.0, step=0.1, format="%.2f", key="initial_point", on_change=reset_session_state
     )
-    learning_rate = st.number_input(
-        "Learning Rate", value=0.1, step=0.01, format="%.2f", key="learning_rate", on_change=reset_session_state
-    )
+    st.number_input(
+        "Learning Rate", value=st.session_state.learning_rate, step=0.01, format="%.2f", 
+        key="learning_rate"
+    )  # Updates session state directly without reset
     st.markdown("---")
     
     # Reset Button
@@ -74,7 +78,7 @@ with right_col:
     if st.button("🔄 Run Descent Step", type="primary"):
         try:
             gradient = compute_derivative(function_input, st.session_state.x_current)
-            st.session_state.x_current -= learning_rate * gradient
+            st.session_state.x_current -= st.session_state.learning_rate * gradient
             st.session_state.iter_count += 1
             st.session_state.history.append(
                 (st.session_state.x_current, evaluate_function(function_input, st.session_state.x_current))