Update app.py

app.py

@@ -1,191 +1,6 @@
-import streamlit as st
-import numpy as np
-import plotly.graph_objects as go
+# Gradient Descent Visualization with Updated Axes and Color Scheme
 
-# Safe function evaluation
-def safe_eval(func_str, x_val):
-    """ Safely evaluates the function at a given x value. """
-    allowed_names = {"x": x_val, "np": np}
-    try:
-        return eval(func_str, {"__builtins__": None}, allowed_names)
-    except Exception as e:
-        raise ValueError(f"Error evaluating the function: {e}")
-
-# Function derivative using finite difference method
-def derivative(func_str, x_val, h=1e-5):
-    """ Numerically compute the derivative of the function at x using finite differences. """
-    return (safe_eval(func_str, x_val + h) - safe_eval(func_str, x_val - h)) / (2 * h)
-
-# Tangent line equation
-def tangent_line(func_str, x_val, x_range):
-    """ Compute the tangent line at a given x value. """
-    y_val = safe_eval(func_str, x_val)
-    slope = derivative(func_str, x_val)
-    return slope * (x_range - x_val) + y_val
-
-# Callback to reset session state
-def reset_state():
-    st.session_state.x = st.session_state.starting_point
-    st.session_state.iteration = 0
-    st.session_state.x_vals = [st.session_state.starting_point]
-    st.session_state.y_vals = [safe_eval(st.session_state.func_input, st.session_state.starting_point)]
-
-# Initialize session state variables
-if "func_input" not in st.session_state:
-    st.session_state.func_input = "x**2 + x"
-if "x" not in st.session_state:
-    st.session_state.x = 4.0
-    st.session_state.iteration = 0
-    st.session_state.x_vals = [4.0]
-    st.session_state.y_vals = [safe_eval(st.session_state.func_input, 4.0)]
-if "learning_rate" not in st.session_state:
-    st.session_state.learning_rate = 0.25
-
-# Full-width layout
-st.set_page_config(layout="wide")
-
-# CSS Styles for Borders, Font, Reduced Padding, and Custom Border Color
-st.markdown(
-    """
-    <style>
-    * {
-        font-family: Cambria, Arial, sans-serif !important;
-    }
-    h1, h2, h3, h4, h5 {
-        text-align: center;
-        margin-top: 0;
-    }
-    input, .stButton button, .stDownloadButton button {
-        border: 2px solid #ea445a;
-        border-radius: 5px;
-        padding: 10px;
-    }
-    .stInfo, .stSuccess {
-        border: 2px solid #ea445a;
-        border-radius: 5px;
-        padding: 10px;
-    }
-    .stButton {
-        margin-top: 10px;
-    }
-    /* Reduced Padding at the top */
-    .css-1d391kg {
-        padding-top: 0.5rem;
-    }
-    /* Centering the legend in the plot */
-    .stPlotlyChart {
-        display: block;
-        margin: 0 auto;
-    }
-    /* Adjusting for full width without scrolling */
-    .css-1lcbvhc {
-        padding-left: 0;
-        padding-right: 0;
-    }
-    /* Custom borders for input fields */
-    .stTextInput input, .stNumberInput input {
-        border: 2px solid #001A6E;
-        border-radius: 5px;
-        padding: 10px;
-    }
-    /* Tooltip styling */
-    .tooltip {
-        position: relative;
-        display: inline-block;
-        cursor: pointer;
-    }
-    .tooltip .tooltiptext {
-        visibility: hidden;
-        opacity: 0;
-        width: 300px;
-        background-color: #001A6E;
-        color: #fff;
-        text-align: center;
-        border-radius: 5px;
-        padding: 5px;
-        position: absolute;
-        z-index: 1;
-        bottom: 125%; /* Position the tooltip above */
-        left: 50%;
-        margin-left: -150px;
-        transition: opacity 0.3s;
-    }
-    .tooltip:hover .tooltiptext {
-        visibility: visible;
-        opacity: 1;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
-
-# Page Layout
-st.title("🌟 Gradient Descent Interactive Tool 🌟")
-
-col1, col2 = st.columns([1, 2])
-
-# Left Section: User Input
-with col1:
-    st.subheader("🔧 Define Your Function")
-
-    st.markdown(
-        """
-        <div class="tooltip">
-            <label for="func_input">Enter a function of 'x':</label>
-            <span class="tooltiptext">
-                **How to input your function:**
-                - Please give the inputs as mentioned below                
-                - x^n as  x**n,                
-                - sin(x) as np.sin(x)                
-                - log(x) as np.log(x),                
-                - e^x or exp(x) as np.exp(x).
-            </span>
-        </div>
-        """,
-        unsafe_allow_html=True
-    )
-    
-    func_input = st.text_input(
-        "👇", 
-        key="func_input", 
-        on_change=reset_state
-    )
-
-    st.subheader("⚙️ Gradient Descent Parameters")
-    starting_point = st.number_input(
-        "Starting Point (X₀)", 
-        value=4.0, 
-        step=0.1, 
-        format="%.2f", 
-        key="starting_point", 
-        on_change=reset_state
-    )
-    new_learning_rate = st.number_input(
-        "Learning Rate (ŋ)", 
-        value=st.session_state.learning_rate, 
-        step=0.01, 
-        format="%.2f"
-    )
-    # Update the learning rate without resetting progress
-    if new_learning_rate != st.session_state.learning_rate:
-        st.session_state.learning_rate = new_learning_rate
-
-    col3, col4 = st.columns(2)
-    with col3:
-        if st.button("🔄 Set Up Function"):
-            reset_state()
-    with col4:
-        if st.button("▶️ Next Iteration"):
-            try:
-                grad = derivative(st.session_state.func_input, st.session_state.x)
-                st.session_state.x = st.session_state.x - st.session_state.learning_rate * grad
-                st.session_state.iteration += 1
-                st.session_state.x_vals.append(st.session_state.x)
-                st.session_state.y_vals.append(safe_eval(st.session_state.func_input, st.session_state.x))
-            except Exception as e:
-                st.error(f"⚠️ Error: {str(e)}")
-
-# Right Section: Visualization
+# Visualization Section in Streamlit
 with col2:
     st.subheader("📊 Gradient Descent Visualization")
     try:
@@ -196,9 +11,9 @@ with col2:
 
         # Function curve
         fig.add_trace(go.Scatter(
-            x=x_plot, 
-            y=y_plot, 
-            mode="lines", 
+            x=x_plot,
+            y=y_plot,
+            mode="lines",
             line=dict(color="blue", width=2),
             name="Function"
         ))
@@ -224,9 +39,30 @@ with col2:
             name="Tangent Line"
         ))
 
+        # Update layout to show negative y-axis and white axes
         fig.update_layout(
-            xaxis=dict(title="x-axis"),
-            yaxis=dict(title="y-axis"),
+            xaxis=dict(
+                title="x-axis",
+                zeroline=True,
+                zerolinecolor="white",
+                zerolinewidth=2,
+                showgrid=True,
+                gridcolor="lightgray",
+                color="white"
+            ),
+            yaxis=dict(
+                title="y-axis",
+                zeroline=True,
+                zerolinecolor="white",
+                zerolinewidth=2,
+                showgrid=True,
+                gridcolor="lightgray",
+                range=[min(y_plot) - 10, max(y_plot) + 10],  # Adjust to show negative y-axis
+                color="white"
+            ),
+            plot_bgcolor="black",
+            paper_bgcolor="black",
+            font=dict(color="white"),
             title="Gradient Descent Visualization",
             width=800,
             height=400,
@@ -237,7 +73,3 @@ with col2:
 
     except Exception as e:
         st.error(f"⚠️ Error in visualization: {str(e)}")
-
-    col5, col6 = st.columns(2)
-    col5.info(f"🧑‍💻 Iteration: {st.session_state.iteration}")
-    col6.success(f"✅ Current x: {st.session_state.x:.4f}, Current f(x): {st.session_state.y_vals[-1]:.4f}")