Update app.py (#96)

- Update app.py (98479d5f7bfaebb8e72dcd9496b39f64d8e3659c)

app.py

@@ -921,83 +921,84 @@ def create_game(host=None, topics=[], num_questions=5, auto_close=True, ai_topic
 # PLAY PAGE
 # -------------------------
 def play_page():
+    import time
+    import streamlit as st
+
     gid = st.session_state.get("active_game_id")
     uname = st.session_state.get("username")
+    
     if not gid or not uname:
         st.error("No active game or username found. Please join or create a game first.")
         return
 
-    # Fetch game and questions
+    # Get game and questions
     games = unified_get("games") or {}
     game = games.get(gid)
     if not game:
         st.error("Game not found.")
         return
-    if game.get("closed"):
-        st.warning("This game has already ended.")
+    if game.get('closed'):
+        st.warning("This game is closed.")
         return
 
     questions = game.get("questions", [])
     if not questions:
-        st.info("No questions loaded.")
+        st.info("No questions loaded for this game.")
         return
 
-    # Initialize session state
-    if "current_index" not in st.session_state:
-        st.session_state["current_index"] = 0
-    if "answers" not in st.session_state:
-        st.session_state["answers"] = [None] * len(questions)
-    if "answer_times" not in st.session_state:
-        st.session_state["answer_times"] = [0] * len(questions)
-    if "question_started_at" not in st.session_state:
-        st.session_state["question_started_at"] = time.time()
+    # Initialize session state for answers and times
+    if 'answers' not in st.session_state:
+        st.session_state['answers'] = [""] * len(questions)
+    if 'answer_times' not in st.session_state:
+        st.session_state['answer_times'] = [0] * len(questions)
+    if 'current_index' not in st.session_state:
+        st.session_state['current_index'] = 0
+    if 'question_started_at' not in st.session_state:
+        st.session_state['question_started_at'] = time.time()
 
-    idx = st.session_state["current_index"]
+    idx = st.session_state['current_index']
 
-    # ---------------- End of questions ----------------
+    # All done
     if idx >= len(questions):
         st.success("All done — submit your answers!")
         return
 
-    # ---------------- Question UI ----------------
+    # ---------------- QUESTION UI ----------------
     q = questions[idx]
-    st.subheader(f"Question {idx+1}/{len(questions)}")
+    st.subheader(f"Question {idx + 1}/{len(questions)}")
     st.write(q["question"])
-    elapsed = int(time.time() - (st.session_state.get('question_started_at') or time.time()))
+
+    # Safe elapsed calculation
+    start_time = st.session_state.get('question_started_at') or time.time()
+    elapsed = int(time.time() - start_time)
     time_limit = 15
     st.markdown(f"**Time left:** {max(0, time_limit - elapsed)} seconds")
 
-    # ---------------- Stable Radio ----------------
+    # Stable radio buttons
     choice = st.radio(
         "Choose an answer:",
         q["options"],
-        index=st.session_state['answers'][idx] if st.session_state['answers'][idx] in q["options"] else 0,
         key=f"choice_{gid}_{idx}"
     )
 
     col1, col2 = st.columns(2)
 
-    # ---------------- Next Button ----------------
+    # ---------------- NEXT ----------------
     with col1:
         if st.button("Next", key=f"next_{gid}_{idx}"):
-            # Record answer and time
             st.session_state['answers'][idx] = choice
-            start_time = st.session_state.get('question_started_at') or time.time()
             st.session_state['answer_times'][idx] = time.time() - start_time
-
-           # st.session_state['answer_times'][idx] = time.time() - st.session_state['question_started_at']
-            # Move to next question
             st.session_state['current_index'] = idx + 1
             st.session_state['question_started_at'] = time.time()
             st.rerun()
 
-    # ---------------- Submit Button ----------------
+    # ---------------- SUBMIT ----------------
     with col2:
         if idx == len(questions) - 1:
             if st.button("Submit All Answers", key=f"submit_{gid}_{idx}"):
                 # Record last question
                 st.session_state['answers'][idx] = choice
-                st.session_state['answer_times'][idx] = time.time() - (st.session_state.get('question_started_at') or time.time())
+                st.session_state['answer_times'][idx] = time.time() - start_time
 
                 answers = st.session_state['answers']
                 times = st.session_state['answer_times']
@@ -1006,7 +1007,7 @@ def play_page():
                 score, flags = compute_score(questions, answers, times)
                 percentage = int(score / (len(questions) * 15) * 100)
 
-                # Update players dict
+                # ---------------- Update players dict ----------------
                 players = unified_get("players") or {}
                 players.setdefault(gid, {})
                 players[gid][uname] = {
@@ -1019,7 +1020,7 @@ def play_page():
                 }
                 unified_set("players", players)
 
-                # Push to leaderboard
+                # ---------------- Push leaderboard ----------------
                 row = {
                     "name": uname,
                     "avatar": st.session_state.get("avatar", "🎮"),
@@ -1049,9 +1050,6 @@ def play_page():
                 st.session_state['answers'] = []
                 st.session_state['answer_times'] = []
                 st.rerun()
-
-
-        
 # Join game
 def join_game_page():
     st.header("Join Game")