Update app.py

app.py

@@ -4,34 +4,38 @@ import random
 # Global variables
 number = 0
 mode = ""
+current_page = "main"
 
 def main_page():
     st.title("Before and After Game")
     st.write("Welcome to the Before and After Game! Click 'Play' to get started.")
 
     if st.button("Play", key="play_button"):
-        select_mode()
+        global current_page
+        current_page = "select_mode"
 
 def select_mode():
     st.title("Choose a Mode")
     st.write("Select 'Before' to find the number before a given number or 'After' to find the number after it.")
 
     col1, col2 = st.columns(2)
+    global current_page, mode
     with col1:
         if st.button("Before", key="before_button"):
-            start_game("before")
+            mode = "before"
+            current_page = "game"
     with col2:
         if st.button("After", key="after_button"):
-            start_game("after")
+            mode = "after"
+            current_page = "game"
 
-def start_game(selected_mode):
-    global mode, number
-    mode = selected_mode
+def start_game():
+    global number
     number = random.randint(1, 100)
     ask_question()
 
 def ask_question():
-    global number, mode
+    global number, mode, current_page
     st.title(f"What is {mode.upper()} {number}?")
     st.write("Choose the correct answer from the options below.")
 
@@ -50,20 +54,31 @@ def ask_question():
         if i % 2 == 0:
             with col1:
                 if st.button(str(option), key=f"option_{option}"):
-                    check_answer(option, correct_answer)
+                    if option == correct_answer:
+                        st.success("Great Job! 🎉")
+                        st.balloons()
+                        current_page = "game"
+                    else:
+                        st.error("Oops! Try again!")
         else:
             with col2:
                 if st.button(str(option), key=f"option_{option}"):
-                    check_answer(option, correct_answer)
+                    if option == correct_answer:
+                        st.success("Great Job! 🎉")
+                        st.balloons()
+                        current_page = "game"
+                    else:
+                        st.error("Oops! Try again!")
 
-def check_answer(selected, correct):
-    if selected == correct:
-        st.success("Great Job! 🎉")
-        st.balloons()
-        st.button("Next Question", on_click=ask_question)
-    else:
-        st.error("Oops! Try again!")
+def app():
+    global current_page
+    if current_page == "main":
+        main_page()
+    elif current_page == "select_mode":
+        select_mode()
+    elif current_page == "game":
+        start_game()
 
 # Run the application
 if __name__ == "__main__":
-    main_page()
+    app()