Update app.py

app.py

@@ -113,9 +113,8 @@ with right_col:
         go.Scatter(x=x_vals, y=y_vals, mode="lines", line=dict(color="green", width=3), name="Function Curve")
     )
     
-    # Add gradient descent points
-    x_points = [point[0] for point in st.session_state.history]
-    y_points = [point[1] for point in st.session_state.history]
+    # Add gradient descent points up to the current index
+    x_points, y_points = zip(*st.session_state.history[:st.session_state.current_index + 1])
     plot.add_trace(
         go.Scatter(
             x=x_points,
@@ -126,13 +125,9 @@ with right_col:
         )
     )
     
-    # 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)
-    
+    # Add tangent line at the selected point
+    tangent_x = np.linspace(selected_x - 2, selected_x + 2, 100)
+    tangent_y = calculate_tangent(function_input, selected_x, tangent_x)
     plot.add_trace(
         go.Scatter(
             x=tangent_x,
@@ -153,5 +148,5 @@ with right_col:
         height=500,  # Reduce the graph height for better fitting
     )
     
-    # Display the plot in the right side (with more space now)
+    # Display the plot in the right side
     st.plotly_chart(plot, use_container_width=True)