Update app.py

app.py

@@ -56,9 +56,6 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# -------------------------------
-# your original code below 👇
-# -------------------------------
 
 st.title("Gradient Descent Visualizer")
 
@@ -127,4 +124,8 @@ if 'func' in st.session_state and len(st.session_state.points) > 0:
         )
         
         fig = go.Figure(data=[trace1, trace2, trace3], layout=layout)
-        st.plotly_chart(fig,_
+        st.plotly_chart(fig, use_container_width=True)
+        st.success(f"Current Point = {iter_points[-1]}")
+
+    except Exception as e:
+        st.error(f"Plot error: {e}")