Update app.py

app.py

@@ -32,8 +32,8 @@ def reset_session_state():
     st.session_state.x_points = [st.session_state.initial_point]
     st.session_state.y_points = [evaluate_function(st.session_state.math_function, st.session_state.initial_point)]
 
-# Create two-column grid layout for the left side
-left_col, right_col = st.columns([2, 1])  # 2 for left, 1 for right grid proportion
+# Create two-column grid layout for the left side (more space for the right graph)
+left_col, right_col = st.columns([1, 2])  # 1 for left, 2 for right grid proportion
 
 # Left side content (Function Input and Gradient Descent Parameters)
 with left_col:
@@ -51,6 +51,10 @@ with left_col:
         "Learning Rate", value=0.1, step=0.01, format="%.2f", key="learning_rate", on_change=reset_session_state
     )
     st.markdown("---")
+    
+    # Reset Button
+    if st.button("🔄 Reset"):
+        reset_session_state()
 
 # Right side content (Gradient Descent Updates and Progress)
 with right_col:
@@ -71,22 +75,6 @@ with right_col:
         except Exception as e:
             st.error(f"Error: {str(e)}")
 
-    # Gradient Descent Progress Section with Different Style
-    st.subheader("Gradient Descent Progress")
-    st.markdown(f"**Iteration:** {st.session_state.iter_count}")
-    st.markdown(f"**Current x Value:** {st.session_state.x_current:.4f}")
-    st.markdown(f"**Current Function Value (f(x)):** {st.session_state.y_points[-1]:.4f}")
-    st.markdown("---")
-    
-    # Styling the updates section
-    st.markdown(
-        f'<div style="background-color:#f0f0f0; padding: 10px; border-radius: 5px;">'
-        f'<strong>Iteration: </strong>{st.session_state.iter_count} <br>'
-        f'<strong>Current x Value: </strong>{st.session_state.x_current:.4f} <br>'
-        f'<strong>Current f(x): </strong>{st.session_state.y_points[-1]:.4f}</div>',
-        unsafe_allow_html=True
-    )
-
 # Generate plot data
 x_vals = np.linspace(-10, 10, 400)
 y_vals = [evaluate_function(function_input, x) for x in x_vals]
@@ -136,5 +124,5 @@ plot.update_layout(
     legend=dict(bgcolor="rgba(255,255,255,0.5)", bordercolor="gray", borderwidth=1),
 )
 
-# Display the plot in the left side
+# Display the plot in the right side (with more space now)
 st.plotly_chart(plot, use_container_width=True)