Update app.py

app.py

@@ -4,13 +4,10 @@ from pages import page_1, page_2
 
 def go_to_main():
     st.session_state.page = 'main'
-
 def go_to_page_1():
     st.session_state.page = 'page_1'
-
 def go_to_page_2():
     st.session_state.page = 'page_2'
-
 if 'page' not in st.session_state:
     st.session_state.page = 'main'
 
@@ -18,9 +15,7 @@ if st.session_state.page == 'main':
     st.markdown("# Main Page 🏠")
     st.button('Go to Page 1', on_click=go_to_page_1)
     st.button('Go to Page 2', on_click=go_to_page_2)
-
 elif st.session_state.page == 'page_1':
     page_1.run(go_to_main, go_to_page_2)
-
 elif st.session_state.page == 'page_2':
     page_2.run(go_to_main, go_to_page_1)