Update app.py

app.py

@@ -82,55 +82,55 @@ with right_col:
     st.markdown(f"**Current Function Value (f(x)):** {st.session_state.y_points[-1]:.4f}")
     st.markdown("---")
 
-# Generate plot data
-x_vals = np.linspace(-10, 10, 400)
-y_vals = [evaluate_function(function_input, x) for x in x_vals]
-
-# Create the plot
-plot = go.Figure()
-
-# Add function plot
-plot.add_trace(
-    go.Scatter(x=x_vals, y=y_vals, mode="lines", line=dict(color="green", width=3), name="Function Curve")
-)
-
-# Add gradient descent points
-plot.add_trace(
-    go.Scatter(
-        x=st.session_state.x_points,
-        y=st.session_state.y_points,
-        mode="markers",
-        marker=dict(color="red", size=10, symbol="diamond"),
-        name="Descent Steps",
+    # Generate plot data
+    x_vals = np.linspace(-10, 10, 400)
+    y_vals = [evaluate_function(function_input, x) for x in x_vals]
+    
+    # Create the plot
+    plot = go.Figure()
+    
+    # Add function plot
+    plot.add_trace(
+        go.Scatter(x=x_vals, y=y_vals, mode="lines", line=dict(color="green", width=3), name="Function Curve")
     )
-)
-
-# Add tangent line
-current_x = st.session_state.x_current
-current_y = evaluate_function(function_input, current_x)
-slope = compute_derivative(function_input, current_x)
-tangent_x = np.linspace(current_x - 2, current_x + 2, 100)
-tangent_y = calculate_tangent(function_input, current_x, tangent_x)
-
-plot.add_trace(
-    go.Scatter(
-        x=tangent_x,
-        y=tangent_y,
-        mode="lines",
-        line=dict(color="blue", width=2, dash="dash"),
-        name="Tangent Line",
+    
+    # Add gradient descent points
+    plot.add_trace(
+        go.Scatter(
+            x=st.session_state.x_points,
+            y=st.session_state.y_points,
+            mode="markers",
+            marker=dict(color="red", size=10, symbol="diamond"),
+            name="Descent Steps",
+        )
     )
-)
-
-# Update plot layout
-plot.update_layout(
-    title="Interactive Gradient Descent with Tangent Visualization",
-    xaxis_title="x",
-    yaxis_title="f(x)",
-    template="plotly_dark",
-    legend=dict(bgcolor="rgba(255,255,255,0.5)", bordercolor="gray", borderwidth=1),
-    height=300,  # Reduce the graph height for better fitting
-)
-
-# Display the plot in the right side (with more space now)
-st.plotly_chart(plot, use_container_width=True)
+    
+    # Add tangent line
+    current_x = st.session_state.x_current
+    current_y = evaluate_function(function_input, current_x)
+    slope = compute_derivative(function_input, current_x)
+    tangent_x = np.linspace(current_x - 2, current_x + 2, 100)
+    tangent_y = calculate_tangent(function_input, current_x, tangent_x)
+    
+    plot.add_trace(
+        go.Scatter(
+            x=tangent_x,
+            y=tangent_y,
+            mode="lines",
+            line=dict(color="blue", width=2, dash="dash"),
+            name="Tangent Line",
+        )
+    )
+    
+    # Update plot layout
+    plot.update_layout(
+        title="Interactive Gradient Descent with Tangent Visualization",
+        xaxis_title="x",
+        yaxis_title="f(x)",
+        template="plotly_dark",
+        legend=dict(bgcolor="rgba(255,255,255,0.5)", bordercolor="gray", borderwidth=1),
+        height=200,  # Reduce the graph height for better fitting
+    )
+    
+    # Display the plot in the right side (with more space now)
+    st.plotly_chart(plot, use_container_width=True)