Update app.py

app.py

@@ -1084,6 +1084,21 @@ st.sidebar.markdown("### You")
 username = st.sidebar.text_input("Your name", value=st.session_state.get("username",""), key="sidebar_username")
 avatar = st.sidebar.selectbox("Avatar", ["🎮","🐱","🐶","🦄","👽","🎩"], index=0, key="sidebar_avatar")
 
+# ---------------- PAGE CONTROL ----------------
+if "page" not in st.session_state:
+    st.session_state["page"] = "Home"
+
+pages = ["Home","Create Game","Join Game","Play","Friends","Inbox","Leaderboard"]
+
+page = st.sidebar.selectbox(
+    "Page",
+    pages,
+    index=pages.index(st.session_state["page"])
+)
+
+# Sync
+st.session_state["page"] = page
+
 # Save to session_state
 st.session_state['username'] = username or st.session_state.get("username","")
 st.session_state['avatar'] = avatar or st.session_state.get("avatar","🎮")
@@ -1092,7 +1107,6 @@ st.session_state['avatar'] = avatar or st.session_state.get("avatar","🎮")
 if st.sidebar.button("Refresh"):
     st.rerun()
 
-page = st.sidebar.selectbox("Page", ["Home","Create Game","Join Game","Play","Friends","Inbox","Leaderboard"], index=0)
 st.title("AI Quiz Game — Online/Offline (Friends & Chat)")
 
 def render_copy_button(val, key):
@@ -1275,6 +1289,8 @@ def create_game(host=None, topics=[], num_questions=5, auto_close=True, ai_topic
     st.session_state['question_started_at'] = None
 
     st.success(f"Game created: {gid} with {len(questions)} questions.")
+    st.session_state["page"] = "Join Game"
+    st.rerun()
     return gid
 # -------------------------
 #playgame
@@ -1629,6 +1645,7 @@ def join_game_page():
             st.session_state['username'] = username
             st.session_state['avatar'] = avatar
 
+            st.session_state["page"] = "Play"
             st.rerun()
 
         else:
@@ -1662,7 +1679,7 @@ def join_game_page():
         st.success("🚀 Game Started!")
         
         if st.button("Go To Play Page"):
-            st.session_state.page = "Play"
+            st.session_state["page"] = "Play"
             st.rerun()