Update app.py (#59)

- Update app.py (07cd3a879936e06995c80340c1c20c4c021aaaa3)

app.py

@@ -505,6 +505,51 @@ def compute_winners(game_id):
 
     return df.head(3).to_dict(orient="records")
 
+# ----------------- Create Game Page -----------------
+def create_game_page():
+    st.header("Create Game")
+
+    host = st.text_input(
+        "Host name",
+        value=st.session_state.get("username", "")
+    )
+
+    # 🔹 Normal topics (static)
+    topics = st.multiselect(
+        "Topics",
+        list(questions_db.keys())
+    )
+
+    # 🔥 NEW: AI topic input
+    ai_topic = st.text_input(
+        "AI Topic (optional)",
+        placeholder="Eg: IPL 2024, Space, Python"
+    )
+
+    num_questions = st.number_input("Number of Questions", min_value=1, max_value=20, value=5)
+    auto_close = st.checkbox("Auto-close game after submission?", value=True)
+
+    if st.button("Create Game"):
+        if not host:
+            st.warning("Please enter your name.")
+            return
+        if not topics and not ai_topic:
+            st.warning("Please select at least one topic or enter an AI topic.")
+            return
+
+        # Pass AI topic to create_game
+        gid = create_game(topics=topics, num_questions=num_questions, auto_close=auto_close, ai_topic=ai_topic)
+
+        st.success(f"Game created! Game ID: {gid}")
+        st.session_state['game_id'] = gid
+        st.session_state['username'] = host
+
+        # Optional: generate AI questions immediately if AI topic is provided
+        if ai_topic:
+            st.info(f"AI questions will be generated for: {ai_topic}")
+            # Here you can call your AI question generator function if you have one
+
+
 def join_game(game_id, username, avatar):
     games = unified_get("games") or {}
     if game_id not in games:
@@ -685,12 +730,21 @@ def home_page():
 def create_game(topics=None, num_questions=5, auto_close=True, ai_topic=None):
     """
     Creates a game and returns a unique game ID.
+    Stores questions to avoid KeyError on join/play.
     """
     topics = topics or []
 
     # Generate a unique game ID
     gid = f"GAME{int(time.time())}{random.randint(100,999)}"
 
+    # Generate questions from selected topics (random)
+    questions = []
+    for t in topics:
+        qlist = questions_db.get(t, [])
+        if qlist:
+            questions.extend(random.sample(qlist, min(len(qlist), num_questions)))
+    questions = questions[:num_questions]
+
     # Save game data in session_state
     if "games" not in st.session_state:
         st.session_state["games"] = {}
@@ -700,69 +754,12 @@ def create_game(topics=None, num_questions=5, auto_close=True, ai_topic=None):
         "auto_close": auto_close,
         "ai_topic": ai_topic,
         "players": [],
-        "closed": False
+        "closed": False,
+        "questions": questions   # important to store questions
     }
 
     return gid
 
-# 🔹 Function to render the Create Game page
-def create_game_page():
-    st.header("Create Game")
-
-    # Host input
-    host = st.text_input(
-        "Host name",
-        value=st.session_state.get("username", "")
-    )
-
-    # Normal topics (static)
-    topics = st.multiselect(
-        "Topics",
-        list(questions_db.keys())
-    )
-
-    # AI topic input
-    ai_topic = st.text_input(
-        "AI Topic (optional)",
-        placeholder="Eg: IPL 2024, Python basics, Technology trends"
-    )
-
-    # Number of questions
-    num_q = st.selectbox(
-        "Number of questions",
-        [5, 10, 15, 20],
-        index=0
-    )
-
-    # Auto-close option
-    auto_close = st.checkbox(
-        "Auto-close when someone submits",
-        value=True
-    )
-
-    # Create Game button
-    if st.button("Create Game"):
-        # Input validation
-        if not host or (not topics and not ai_topic):
-            st.error("Enter host and choose topics or AI topic.")
-            return
-
-        # Call create_game correctly
-        gid = create_game(
-            topics=topics,
-            num_questions=num_q,
-            auto_close=auto_close,
-            ai_topic=ai_topic
-        )
-
-        # Save session state info
-        st.session_state['game_id'] = gid
-        st.session_state['username'] = host
-        st.session_state['avatar'] = st.session_state.get('avatar', '🎮')
-        st.session_state['is_host'] = True
-
-        st.success(f"Created game: {gid}")
-
 # Join game
 def join_game_page():
     st.header("Join Game")
@@ -790,6 +787,9 @@ def join_game_page():
                 st.rerun()
 
 # Play page
+# ----------------- Create Game -----------------
+
+# ----------------- Play Page -----------------
 def play_page():
     st.header("Play")
     gid = st.session_state.get('game_id')
@@ -815,31 +815,33 @@ def play_page():
         st.error("Game closed.")
         return
 
-    questions = st.session_state.get('game_questions', game.get('questions', []))
+    # Ensure questions exist
+    questions = st.session_state.get('game_questions') or game.get('questions', [])
     if not questions:
         st.error("No questions loaded.")
         return
 
+    # Ensure timer exists
+    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[0])
 
-    # Ensure timer always exists
-    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()
-
-    choice = st.radio("Choose an answer:", q[1], key=f"choice_{idx}")
-
     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[1], key=f"choice_{idx}")
+
     col1, col2 = st.columns(2)
-    
+
     with col1:
         if st.button("Next"):
             taken = time.time() - st.session_state['question_started_at']
@@ -854,8 +856,8 @@ def play_page():
             st.session_state['answer_times'] = times
             st.session_state['current_index'] = idx + 1
 
-            # ❌ DO NOT set to None
-            # st.session_state['question_started_at'] = None
+            # Do NOT reset question_started_at to None
+            st.session_state['question_started_at'] = time.time()  # reset for next question
 
             players = unified_get("players") or {}
             if players.get(gid, {}).get(uname):
@@ -871,7 +873,7 @@ def play_page():
                 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['question_started_at'] if st.session_state.get('question_started_at') else None
+                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)
@@ -893,7 +895,7 @@ def play_page():
                     players[gid][uname]['submitted'] = True
                     players[gid][uname]['submitted_at'] = now
                     unified_set("players", players)
-                if game.get('auto_close_on_submit', True):
+                if game.get('auto_close', True):
                     games[gid]['closed'] = True
                     games[gid]['closed_at'] = now
                     unified_set("games", games)
@@ -903,20 +905,7 @@ def play_page():
                 st.session_state['last_game'] = gid
                 st.session_state['current_index'] = 0
                 st.rerun()
-
-    st.markdown("---")
-    st.subheader("Game Chat")
-    msgs = unified_get("messages") or {}
-    game_msgs = msgs.get(gid, []) if isinstance(msgs, dict) else msgs
-    for m in game_msgs[-100:]:
-        st.write(f"**{m.get('user')}** [{m.get('ts')}]: {m.get('text')}")
-    new_msg = st.text_input("Send message to game", key=f"chat_{gid}")
-    if st.button("Send", key=f"send_{gid}"):
-        if new_msg and new_msg.strip():
-            unified_push_message(gid, {"user": uname, "text": new_msg.strip(), "ts": now_iso()})
-            heartbeat_unified(gid, uname)
-            st.rerun()
-
+            
 # Friends page
 def friends_page():
     st.header("Friends")