Update pages/page_2.py

pages/page_2.py

@@ -1,14 +1,8 @@
-# page_2.py inside the 'pages' folder
+# page_2.py inside 'pages' folder
 import streamlit as st
 
-def run():
+def run(go_to_main, go_to_page_1):
     st.markdown("# Page 2 📊")
-    # Add the rest of your page 2 code here
-    # For example:
-    st.write("Welcome to Page 2 of the application!")
-    
-    # Navigation buttons (if needed)
-    if st.button('Go to Main Page'):
-        st.session_state.page = 'main'
-    if st.button('Go to Page 1'):
-        st.session_state.page = 'page_1'
+    st.button('Go to Main Page', on_click=go_to_main)
+    st.button('Go to Page 1', on_click=go_to_page_1)
+    # Rest of your page 2 code