Update app.py

app.py

@@ -1,26 +1,15 @@
 import streamlit as st
+from pages import main, new_page
 
-# Main page content
-st.title("Main Page")
-
-# Button to navigate to another page
-if st.button("Go to New Page"):
-    st.session_state.page = "new_page"
-    st.experimental_rerun()  # Refresh the app to load the new page
-
-
-
+# Initialize session state
 if "page" not in st.session_state:
     st.session_state.page = "main"
 
-# Dynamically load pages
-if st.session_state.page == "main":
-    st.title("Main Page")
+# Page routing
+pages = {
+    "main": main.run,
+    "new_page": new_page.run,
+}
 
-    # Button to navigate to the new page
-    if st.button("Go to New Page"):
-        st.session_state.page = "new_page"
-        st.experimental_rerun()
-elif st.session_state.page == "new_page":
-    from new_page import run_new_page
-    run_new_page()
+# Run the selected page
+pages[st.session_state.page]()