Update app.py

app.py

@@ -1,41 +1,42 @@
 import streamlit as st
 import random
 
-# Global variables
-number = 0
-mode = ""
-current_page = "main"
+# Initialize session state
+if "current_page" not in st.session_state:
+    st.session_state.current_page = "main"
+if "mode" not in st.session_state:
+    st.session_state.mode = ""
+if "number" not in st.session_state:
+    st.session_state.number = 0
 
 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"):
-        global current_page
-        current_page = "select_mode"
+        st.session_state.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"):
-            mode = "before"
-            current_page = "game"
+            st.session_state.mode = "before"
+            st.session_state.current_page = "game"
     with col2:
         if st.button("After", key="after_button"):
-            mode = "after"
-            current_page = "game"
+            st.session_state.mode = "after"
+            st.session_state.current_page = "game"
 
 def start_game():
-    global number
-    number = random.randint(1, 100)
+    st.session_state.number = random.randint(1, 100)
     ask_question()
 
 def ask_question():
-    global number, mode, current_page
+    number = st.session_state.number
+    mode = st.session_state.mode
     st.title(f"What is {mode.upper()} {number}?")
     st.write("Choose the correct answer from the options below.")
 
@@ -54,29 +55,27 @@ def ask_question():
         if i % 2 == 0:
             with col1:
                 if st.button(str(option), key=f"option_{option}"):
-                    if option == correct_answer:
-                        st.success("Great Job! 🎉")
-                        st.balloons()
-                        current_page = "game"
-                    else:
-                        st.error("Oops! Try again!")
+                    check_answer(option, correct_answer)
         else:
             with col2:
                 if st.button(str(option), key=f"option_{option}"):
-                    if option == correct_answer:
-                        st.success("Great Job! 🎉")
-                        st.balloons()
-                        current_page = "game"
-                    else:
-                        st.error("Oops! Try again!")
+                    check_answer(option, correct_answer)
+
+def check_answer(selected, correct):
+    if selected == correct:
+        st.success("Great Job! 🎉")
+        st.balloons()
+        if st.button("Next Question"):
+            start_game()
+    else:
+        st.error("Oops! Try again!")
 
 def app():
-    global current_page
-    if current_page == "main":
+    if st.session_state.current_page == "main":
         main_page()
-    elif current_page == "select_mode":
+    elif st.session_state.current_page == "select_mode":
         select_mode()
-    elif current_page == "game":
+    elif st.session_state.current_page == "game":
         start_game()
 
 # Run the application