Update app.py

app.py

@@ -149,20 +149,20 @@ with col2:
                 zerolinewidth=2,
                 showgrid=True,
                 gridcolor="lightgray",
-                range=[0, max(y_plot) + 10],  # Show non-negative y-axis only
+                range=[0, max(y_plot) + 10],
                 color="white"
             ),
             plot_bgcolor="black",
             paper_bgcolor="black",
             font=dict(color="white"),
             legend=dict(
-                x=0.6,  # Legend slightly left for border visibility
+                x=0.6,
                 y=1.0,
                 bgcolor="black",
                 bordercolor="#001A6E",
                 borderwidth=2
             ),
-            margin=dict(l=10, r=80, t=10, b=10),  # Expand right border
+            margin=dict(l=10, r=80, t=10, b=10),
             width=800,
             height=400,
             showlegend=True
@@ -173,8 +173,25 @@ with col2:
     except Exception as e:
         st.error(f"⚠️ Error in visualization: {str(e)}")
 
-# Display iteration and current point info
 col5, col6, col7 = st.columns(3)
-col5.info(f"🧑‍💻 Iteration: {st.session_state.iteration}")
-# col6.success(f"✅ Current x: {st.session_state.x:.4f}")
-col7.warning(f"📍 Current Point: ({st.session_state.x:.4f}, {st.session_state.y_vals[-1]:.4f})")
+
+# Custom HTML/CSS for iteration and current point display with pink background and black borders
+col5.markdown(
+    f"""
+    <div style="background-color:#FFB6C1; border: 2px solid black; border-radius: 8px; padding: 10px; text-align: center;">
+        <span style="color:black; font-weight: bold;">🧑‍💻 Iteration:</span> <br> 
+        <span style="color:black;">{st.session_state.iteration}</span>
+    </div>
+    """,
+    unsafe_allow_html=True,
+)
+
+col7.markdown(
+    f"""
+    <div style="background-color:#FFB6C1; border: 2px solid black; border-radius: 8px; padding: 10px; text-align: center;">
+        <span style="color:black; font-weight: bold;">📍 Current Point:</span> <br> 
+        <span style="color:black;">({st.session_state.x:.4f}, {st.session_state.y_vals[-1]:.4f})</span>
+    </div>
+    """,
+    unsafe_allow_html=True,
+)