Update app.py

app.py

@@ -102,14 +102,7 @@ col1, col2 = st.columns([1, 2])
 with col1:
     st.subheader("🔧 Define Your Function")
     
-    # Use a temporary storage for the function input
-    func_input = st.text_input(
-        "Enter a function of 'x':", 
-        st.session_state.func_input, 
-        key="func_input", 
-        on_change=reset_state
-    )
-
+    # Display the predefined function buttons
     st.write("Or choose a predefined function:")
     cols = st.columns(4)
     for i, (btn_label, func_value) in enumerate(predefined_functions.items()):
@@ -119,6 +112,14 @@ with col1:
                 reset_state()  # Ensure that the reset function is called to update the state
                 st.experimental_rerun()  # Re-run to ensure everything updates properly
 
+    # Use text input for the user to define a function
+    func_input = st.text_input(
+        "Enter a function of 'x':", 
+        st.session_state.func_input, 
+        key="func_input", 
+        on_change=reset_state
+    )
+
     st.subheader("⚙️ Gradient Descent Parameters")
     starting_point = st.number_input(
         "Starting Point (X₀)",