Update app.py

app.py

@@ -66,7 +66,7 @@ if "x" not in st.session_state:
     st.session_state.y_vals = [safe_eval(func_input, starting_point)]
 
 # Perform one iteration when the button is pressed
-if st.sidebar.button("🔄 Perform Iteration"):
+if st.sidebar.button("🔄 Run Descent Step"):
     try:
         grad = derivative(func_input, st.session_state.x)
         st.session_state.x -= learning_rate * grad
@@ -76,6 +76,10 @@ if st.sidebar.button("🔄 Perform Iteration"):
     except Exception as e:
         st.sidebar.error(f"Error: {str(e)}")
 
+# Reset button
+if st.sidebar.button("🔄 Reset"):
+    reset_state()
+
 # Display gradient descent progress
 st.subheader("🧮 Gradient Descent Progress")
 st.write(f"*Iteration:* {st.session_state.iteration}")
@@ -90,12 +94,12 @@ y_range = [safe_eval(func_input, x) for x in x_range]
 fig = go.Figure()
 
 # Plot the function with a cool gradient color
-fig.add_trace(go.Scatter(x=x_range, y=y_range, mode="lines", line=dict(color="royalblue", width=3), name="Function"))
+fig.add_trace(go.Scatter(x=x_range, y=y_range, mode="lines", line=dict(color="mediumslateblue", width=3), name="Function"))
 
 # Plot gradient descent points with purple color
 fig.add_trace(go.Scatter(
     x=st.session_state.x_vals, y=st.session_state.y_vals,
-    mode="markers", marker=dict(color="mediumvioletred", size=10, symbol="circle"), name="Gradient Descent Points"
+    mode="markers", marker=dict(color="crimson", size=10, symbol="circle"), name="Gradient Descent Points"
 ))
 
 # Plot tangent line at current point with dotted line
@@ -108,7 +112,7 @@ tangent_y = tangent_line(func_input, current_x, tangent_x)
 
 fig.add_trace(go.Scatter(
     x=tangent_x, y=tangent_y, mode="lines",
-    line=dict(color="orange", dash="dot", width=2), name="Tangent Line"
+    line=dict(color="gold", dash="dot", width=2), name="Tangent Line"
 ))
 
 # Customize the layout with a soft gradient background
@@ -117,8 +121,8 @@ fig.update_layout(
     xaxis=dict(title="x", range=[-2, 6], showgrid=False),
     yaxis=dict(title="f(x)", showgrid=False),
     template="plotly",
-    plot_bgcolor="rgb(243, 243, 243)",  # Soft background color
-    paper_bgcolor="rgb(243, 243, 243)",  # Soft background color
+    plot_bgcolor="rgb(250, 250, 250)",  # Light background for a soft look
+    paper_bgcolor="rgb(250, 250, 250)",  # Consistent soft background color
     font=dict(family="Arial, sans-serif", size=14, color="black"),
     legend=dict(orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1),
 )