Update app.py

app.py

@@ -859,104 +859,147 @@ def home_page():
     st.write("Create games, invite friends, play and climb the leaderboard.")
 
     # ---------------- TOTAL ONLINE PLAYERS ----------------
-    total_online = get_total_online_players()
+    players_map = unified_get("players") or {}
+    total_online = sum(len(v) for v in players_map.values())
     st.metric("🟢 Players Online", total_online)
 
     # ---------------- SHOW LAST SCORE ----------------
-    last_score = st.session_state.get('last_score')
-    last_game = st.session_state.get('last_game')
-    if last_score is not None and last_game:
+    last_score = st.session_state.get("last_score")
+    last_game = st.session_state.get("last_game")
+    if last_score is not None:
         st.success(f"Your last score: {last_score} (Game {last_game})")
 
     # ---------------- LOAD GAMES ----------------
     games = unified_get("games") or {}
-    if not games:
-        st.info("No games exist yet. Please create a game first.")
-        return
+    st.subheader("Recent games")
+
+    for g in sorted(
+        games.values(),
+        key=lambda x: x.get("created_at", ""),
+        reverse=True
+    )[:10]:
 
-    st.subheader("Recent Games")
-    for g in sorted(games.values(), key=lambda x: x.get("created_at", ""), reverse=True)[:10]:
         gid = g.get("game_id")
-        st.markdown(f"### 🎮 Game: **{gid}** {'(Closed)' if g.get('closed') else ''}")
+        players = g.get("players", [])
+        submissions = g.get("submissions", {})
+
+        st.markdown(
+            f"### 🎮 Game: **{gid}** {'(Closed)' if g.get('closed') else ''}"
+        )
         st.write(f"Host: {g.get('host')} — Topics: {', '.join(g.get('topics', []))}")
         st.write(f"Created: {g.get('created_at')}")
 
-        # Players & submissions
-        players = g.get("players", [])
-        submissions = g.get("submissions", {})
+        joined = len(players)
+        submitted = len(submissions)
 
-        st.write(f"Players joined: **{len(players)}**")
-        st.write(f"Submitted: **{len(submissions)} / {len(players)}**")
+        st.write(f"Players joined: **{joined}**")
+        st.write(f"Submitted: **{submitted} / {joined}**")
 
-        # Top 3 players
+        # ---------------- TOP 3 PLAYERS ----------------
         if submissions:
             st.markdown("🏆 **Top 3 Players**")
-            top_players = sorted(submissions.items(), key=lambda x: x[1], reverse=True)[:3]
-            for i, (uname, score) in enumerate(top_players, start=1):
-                st.write(f"{i}. 🎮 **{uname}** — {score} pts")
+            top_players = sorted(
+                submissions.items(),
+                key=lambda x: x[1],
+                reverse=True
+            )[:3]
+
+            for i, (uname_p, score) in enumerate(top_players, start=1):
+                st.write(f"{i}. 🎮 **{uname_p}** — {score} pts")
 
-        # Player status
+        # ---------------- PLAYER STATUS ----------------
         if players:
             st.markdown("**👥 Player Status**")
-            for uname in players:
-                if uname in submissions:
-                    st.write(f"🎮 **{uname}** — ✅ Submitted ({submissions[uname]} pts)")
+            for uname_p in players:
+                if uname_p in submissions:
+                    st.write(
+                        f"🎮 **{uname_p}** — ✅ Submitted ({submissions[uname_p]} pts)"
+                    )
                 else:
-                    st.write(f"🎮 **{uname}** — ⏳ Playing")
+                    st.write(f"🎮 **{uname_p}** — ⏳ Playing")
 
         st.markdown("---")
 
-        # Invite & challenge friends
-        if not g.get("closed") and st.session_state.get("username"):
+        # ---------------- INVITE & CHALLENGE ----------------
+        if not g.get("closed"):
             st.info(f"Share this Game ID: {gid}")
             render_copy_button(gid, gid)
 
-            if st.button(f"Invite friends to {gid}", key=f"invite_{gid}"):
-                friends = get_friends_map().get(st.session_state["username"], [])
-                if not friends:
-                    st.warning("No friends to invite.")
-                else:
-                    for f in friends:
-                        send_game_invite(st.session_state["username"], f, gid)
-                    st.success("Invites sent to friends.")
-
-            if st.button(f"Challenge friends with a new game like {gid}", key=f"challenge_{gid}"):
-                new_id = create_game(
-                    host=st.session_state.get("username", "Host"),
-                    topics=g.get("topics", []),
-                    num_questions=len(g.get("questions", []))
-                )
-                st.success(f"Challenge created: {new_id}")
-                st.session_state['active_game_id'] = new_id
-                st.session_state['game_questions'] = games.get(new_id, {}).get("questions", [])
-                st.experimental_rerun()
+            if st.session_state.get("username"):
+                if st.button(f"Invite your friends to {gid}", key=f"invite_{gid}"):
+                    friends = get_friends_map().get(
+                        st.session_state["username"], []
+                    )
+                    if not friends:
+                        st.warning("No friends to invite.")
+                    else:
+                        for f in friends:
+                            send_game_invite(
+                                st.session_state["username"], f, gid
+                            )
+                        st.success("Invites sent to friends.")
+
+                if st.button(
+                    f"Challenge friends with a new game like {gid}",
+                    key=f"challenge_{gid}"
+                ):
+                    new_id = create_game(
+                        st.session_state.get("username", "Host"),
+                        g.get("topics", []),
+                        num_questions=len(g.get("questions", []))
+                    )
+
+                    st.session_state["active_game_id"] = new_id
+                    st.session_state["game_questions"] = games.get(
+                        new_id, {}
+                    ).get("questions", [])
+
+                    st.success(f"Challenge created: {new_id}")
+                    st.experimental_rerun()
 
     # ---------------- WEEKLY LEADERBOARD ----------------
     st.subheader("🏆 Weekly Leaderboard (Top 10)")
     weekly = get_weekly_leaderboard()
+
     if not weekly:
         st.info("No scores yet this week.")
     else:
         for i, r in enumerate(weekly, 1):
-            st.write(f"{i}. {r.get('avatar','🎮')} **{r['name']}** (Game {r['game_id']}) — {r['score']} pts")
-
-    # ---------------- UPDATE ACTIVE GAME SUBMISSIONS ----------------
-    active_game_id = st.session_state.get('active_game')
-    username = st.session_state.get('username')
-    if active_game_id and 'final_score' in st.session_state and username:
-        games = unified_get("games") or {}
-        game = games.get(active_game_id, {})
-        if game:
-            players = game.get("players", [])
-            submissions = game.get("submissions", {})
-
-            if username not in players:
-                players.append(username)
-            submissions[username] = st.session_state['final_score']
-
-            game["players"] = players
-            game["submissions"] = submissions
-            unified_set("games", {active_game_id: game})
+            st.write(
+                f"{i}. {r.get('avatar', '🎮')} "
+                f"**{r['name']}** (Game {r['game_id']}) — {r['score']} pts"
+            )
+
+    # ---------------- PLAY PAGE SETUP ----------------
+    if "active_game" not in st.session_state:
+        if games:
+            st.session_state["active_game"] = list(games.keys())[0]
+        else:
+            st.warning("No games exist yet. Please create a game first.")
+
+    if "username" not in st.session_state:
+        st.session_state["username"] = "Guest"
+
+    # ---------------- SAFE SCORE UPDATE ----------------
+    active_game_id = st.session_state.get("active_game")
+
+    if active_game_id and "final_score" in st.session_state:
+        game = unified_get(f"games/{active_game_id}") or {}
+        players = game.get("players", [])
+        submissions = game.get("submissions", {})
+
+        username = st.session_state["username"]
+
+        if username not in players:
+            players.append(username)
+
+        submissions[username] = st.session_state["final_score"]
+
+        game["players"] = players
+        game["submissions"] = submissions
+
+        unified_set(f"games/{active_game_id}", game)
+
 
 
 # -------------------------