Update app.py (#81)

- Update app.py (2053197d6b7302d519c720ed331b189dcabf545d)

app.py

@@ -687,8 +687,200 @@ def get_top3_and_player_count(game_id):
 # Home page
 
 HEARTBEAT_THRESHOLD_SECONDS = 60  # adjust if needed
-
 def home_page():
+    st.header("Home")
+    st.write("Create games, invite friends, play and climb the leaderboard.")
+
+    # Show last score
+    if st.session_state.get('last_score') is not None:
+        st.success(f"Your last score: {st.session_state['last_score']} (Game {st.session_state.get('last_game')})")
+
+    # Load games and players
+    games = unified_get("games") or {}
+    players_map = unified_get("players") or {}
+
+    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 ''}")
+        st.write(f"Host: {g.get('host')} — Topics: {', '.join(g.get('topics', []))}")
+        st.write(f"Created: {g.get('created_at')}")
+
+        # Players for this game
+        players_here = players_map.get(gid, {}) or {}
+        st.write(f"Players joined: **{len(players_here)}**")
+
+        # Submitted count
+        submitted_count = sum(1 for p in players_here.values() if p.get("submitted"))
+        st.write(f"Submitted: **{submitted_count} / {len(players_here)}**")
+
+        # Final scores
+        if players_here:
+            st.markdown("**📊 Final Scores**")
+            for uname_p, info in players_here.items():
+                if info.get("submitted"):
+                    st.write(
+                        f"{info.get('avatar','🎮')} **{uname_p}** — "
+                        f"Score: **{info.get('score', 0)}**, "
+                        f"Percentage: **{info.get('percentage', 0)}%**"
+                    )
+
+        # Player status
+        if players_here:
+            st.markdown("**👥 Player Status**")
+            for uname_p, info in players_here.items():
+                status = "✅ Submitted" if info.get('submitted') else "⏳ Playing"
+                st.write(f"{info.get('avatar','🎮')} **{uname_p}** — {status}")
+
+        st.markdown("---")
+
+        # Invite & challenge buttons
+        if not g.get('closed'):
+            st.info(f"Share this Game ID: {gid}")
+            render_copy_button(gid, gid)
+
+            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', []))
+                    )
+                    # Update session for new game
+                    st.session_state["current_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']}** "
+                f"(Game {r['game_id']}) — {r['score']} pts"
+            )
+
+# ----------------------------
+# Play Page
+# ----------------------------
+def play_page():
+    uname = st.session_state.get("username")
+    gid = st.session_state.get("current_game_id")
+    if not uname or not gid:
+        st.warning("No active game or username found. Please join or create a game first.")
+        return
+
+    games = unified_get("games") or {}
+    game = games.get(gid)
+    if not game:
+        st.error("Game not found.")
+        return
+
+    questions = st.session_state.get("game_questions") or game.get("questions", [])
+    if not questions:
+        st.error("No questions loaded.")
+        return
+
+    # Timer
+    if 'question_started_at' not in st.session_state or st.session_state['question_started_at'] is None:
+        st.session_state['question_started_at'] = time.time()
+
+    idx = st.session_state.get('current_index', 0)
+    if idx >= len(questions):
+        st.success("All done — submit!")
+        return
+
+    q = questions[idx]
+    st.subheader(f"Question {idx+1}/{len(questions)}")
+    st.write(q["question"])
+    elapsed = int(time.time() - st.session_state['question_started_at'])
+    time_limit = 15
+    st.markdown(f"**Time left:** {max(0, time_limit - elapsed)} seconds")
+
+    choice = st.radio("Choose an answer:", q["options"], key=f"choice_{idx}")
+
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("Next", key=f"next_{idx}"):
+            taken = time.time() - st.session_state['question_started_at']
+            answers = st.session_state.get('answers', [""]*len(questions))
+            times = st.session_state.get('answer_times', [None]*len(questions))
+            answers[idx] = choice
+            times[idx] = taken
+            st.session_state['answers'] = answers
+            st.session_state['answer_times'] = times
+            st.session_state['current_index'] = idx + 1
+            st.session_state['question_started_at'] = time.time()
+
+            # Update heartbeat
+            players = unified_get("players") or {}
+            if players.get(gid, {}).get(uname):
+                players[gid][uname]['last_heartbeat'] = now_iso()
+                unified_set("players", players)
+            heartbeat_unified(gid, uname)
+            st.experimental_rerun()
+
+    with col2:
+        if idx == len(questions) - 1:
+            if st.button("Submit All Answers", key=f"submit_{idx}"):
+                answers = st.session_state.get('answers', [""]*len(questions))
+                times = st.session_state.get('answer_times', [None]*len(questions))
+                answers[idx] = choice
+                times[idx] = time.time() - st.session_state.get('question_started_at', time.time())
+                st.session_state['answers'] = answers
+                st.session_state['answer_times'] = times
+
+                score, flags = compute_score(questions, answers, times)
+                now = now_iso()
+                avatar = "🎮"
+                row = {
+                    "name": uname,
+                    "score": score,
+                    "game_id": gid,
+                    "topics": ",".join(game.get('topics', [])),
+                    "timestamp": now,
+                    "avatar": avatar,
+                    "questions": " || ".join([q["question"] for q in questions]),
+                    "answers": " || ".join([str(a) for a in answers]),
+                    "correct_flags": " || ".join(flags),
+                    "percentage": round(score / len(questions) * 100, 2)
+                }
+                unified_push_leaderboard(row)
+
+                # Mark player submitted
+                players = unified_get("players") or {}
+                if players.get(gid) and players[gid].get(uname):
+                    players[gid][uname]['submitted'] = True
+                    players[gid][uname]['submitted_at'] = now
+                    players[gid][uname]['score'] = score
+                    players[gid][uname]['percentage'] = row['percentage']
+                    unified_set("players", players)
+
+                # Auto close game
+                if game.get('auto_close', True):
+                    games[gid]['closed'] = True
+                    games[gid]['closed_at'] = now
+                    unified_set("games", games)
+
+                st.success(f"Submitted! Score: {score}")
+                st.balloons()
+                st.session_state['last_score'] = score
+                st.session_state['last_game'] = gid
+                st.session_state['current_index'] = 0
+def home_pe():
     st.header("Home")
     st.write("Create games, join games, and view leaderboard.")
 
@@ -766,7 +958,7 @@ def home_page():
                      f"(Game {r['game_id']}) — {r['score']} pts")
 
 # ---------- PLAY PAGE ----------
-def play_page():
+def y_page():
     gid = st.session_state.get("current_game_id")
     uname = st.session_state.get("username")