Update app.py (#41)

- Update app.py (c83a736ca9a6b12af60ebaff79a08fe177fb7a3b)

app.py

@@ -1,6 +1,7 @@
 # app.py
 """
-Fully functional AI Quiz Game — Single Player Only with Chat, Leaderboard & History
+AI Quiz Game — Online & Offline
+Multiplayer, Chat, Friends, Leaderboard
 """
 
 import os, uuid, time, json, random
@@ -17,62 +18,57 @@ PLAYERS_FILE = os.path.join(DATA_DIR,"players.json")
 MESSAGES_FILE = os.path.join(DATA_DIR,"messages.json")
 SESSIONS_FILE = os.path.join(DATA_DIR,"sessions.json")
 LEADERBOARD_FILE = os.path.join(DATA_DIR,"leaderboard.csv")
+FRIENDS_FILE = os.path.join(DATA_DIR,"friends.json")
+INBOX_FILE = os.path.join(DATA_DIR,"inbox.json")
 
-# ---------------- Questions ----------------
+HEARTBEAT_THRESHOLD_SECONDS=40
+
+# ---------------- Questions DB ----------------
 questions_db = {
     "Geography":[("Capital of France?",["Paris","London","Berlin","Madrid"],"Paris"),
                  ("Largest country by area?",["Canada","USA","Russia","China"],"Russia")],
     "Math":[("5*12?",["50","60","55","70"],"60"),
-            ("sqrt(64)?",["6","7","8","9"],"8")]
+            ("sqrt(64)?",["6","7","8","9"],"8")],
+    "Science":[("Water formula?",["H2O","CO2","O2","NaCl"],"H2O"),
+               ("Planet closest to Sun?",["Venus","Earth","Mercury","Mars"],"Mercury")]
 }
 
 # ---------------- Helpers ----------------
-def load_json(path, default):
-    if os.path.exists(path):
-        try:
-            with open(path,"r",encoding="utf-8") as f:
-                return json.load(f)
-        except:
-            return default
-    return default
-def save_json(path,data):
-    with open(path,"w",encoding="utf-8") as f:
-        json.dump(data,f,indent=2,ensure_ascii=False)
-
 def now_iso(): return datetime.utcnow().isoformat()
 def parse_iso(s):
     try: return datetime.fromisoformat(s)
     except: return None
 
-# ---------------- Initialize ----------------
-for f in [GAMES_FILE,PLAYERS_FILE,MESSAGES_FILE,SESSIONS_FILE]:
-    if not os.path.exists(f): save_json(f,{})
-
-if not os.path.exists(LEADERBOARD_FILE):
-    pd.DataFrame(columns=['name','score','game_id','topics','timestamp','avatar']).to_csv(LEADERBOARD_FILE,index=False)
-
-if "uid" not in st.session_state: st.session_state['uid']=str(uuid.uuid4())
-if "session_id" not in st.session_state: st.session_state['session_id']=str(uuid.uuid4())
+def load_json(path, default={}):
+    if os.path.exists(path):
+        try:
+            with open(path,"r",encoding="utf-8") as f: return json.load(f)
+        except: return default
+    return default
 
-# ---------------- Load Data ----------------
-def get_data(file): return load_json(file,{})
-def set_data(file,data): save_json(file,data)
+def save_json(path,data):
+    with open(path,"w",encoding="utf-8") as f:
+        json.dump(data,f,indent=2,ensure_ascii=False)
 
 def unified_get(collection):
-    if collection=="games": return get_data(GAMES_FILE)
-    if collection=="players": return get_data(PLAYERS_FILE)
-    if collection=="messages": return get_data(MESSAGES_FILE)
-    if collection=="sessions": return get_data(SESSIONS_FILE)
+    if collection=="games": return load_json(GAMES_FILE,{})
+    if collection=="players": return load_json(PLAYERS_FILE,{})
+    if collection=="messages": return load_json(MESSAGES_FILE,{})
+    if collection=="sessions": return load_json(SESSIONS_FILE,{})
+    if collection=="friends": return load_json(FRIENDS_FILE,{})
+    if collection=="inbox": return load_json(INBOX_FILE,{})
     if collection=="leaderboard":
         try: return pd.read_csv(LEADERBOARD_FILE).to_dict(orient="records")
         except: return []
     return {}
 
 def unified_set(collection,data):
-    if collection=="games": set_data(GAMES_FILE,data)
-    if collection=="players": set_data(PLAYERS_FILE,data)
-    if collection=="messages": set_data(MESSAGES_FILE,data)
-    if collection=="sessions": set_data(SESSIONS_FILE,data)
+    if collection=="games": save_json(GAMES_FILE,data)
+    if collection=="players": save_json(PLAYERS_FILE,data)
+    if collection=="messages": save_json(MESSAGES_FILE,data)
+    if collection=="sessions": save_json(SESSIONS_FILE,data)
+    if collection=="friends": save_json(FRIENDS_FILE,data)
+    if collection=="inbox": save_json(INBOX_FILE,data)
     if collection=="leaderboard":
         pd.DataFrame(data).to_csv(LEADERBOARD_FILE,index=False)
 
@@ -85,12 +81,11 @@ def unified_push_message(game_id,msg_obj):
     unified_set("messages",all_msgs)
 
 def unified_push_leaderboard(row):
-    df = pd.read_csv(LEADERBOARD_FILE)
+    df = pd.DataFrame(unified_get("leaderboard"))
     df = pd.concat([df,pd.DataFrame([row])],ignore_index=True)
-    df.to_csv(LEADERBOARD_FILE,index=False)
+    unified_set("leaderboard",df.to_dict(orient="records"))
 
 # ---------------- Session ----------------
-HEARTBEAT_THRESHOLD_SECONDS=40
 def claim_session(game_id,username):
     sessions=unified_get("sessions") or {}
     game_sessions=sessions.get(game_id,{})
@@ -157,15 +152,48 @@ def compute_score(questions,answers,times):
             total+=base+bonus
     return total
 
+# ---------------- Friends ----------------
+def get_friends(): return unified_get("friends") or {}
+def save_friends(f): unified_set("friends",f)
+def get_inbox(): return unified_get("inbox") or {}
+def save_inbox(i): unified_set("inbox",i)
+def send_friend_request(frm,to):
+    inbox=get_inbox()
+    inbox.setdefault(to,[]).append({"type":"friend_request","from":frm,"ts":now_iso()})
+    save_inbox(inbox)
+def accept_friend_request(user,frm):
+    friends=get_friends()
+    friends.setdefault(user,[]); friends.setdefault(frm,[])
+    if frm not in friends[user]: friends[user].append(frm)
+    if user not in friends[frm]: friends[frm].append(user)
+    save_friends(friends)
+    inbox=get_inbox()
+    inbox[user]=[e for e in inbox.get(user,[]) if not (e.get("type")=="friend_request" and e.get("from")==frm)]
+    save_inbox(inbox)
+
+# ---------------- Initialize ----------------
+for f in [GAMES_FILE,PLAYERS_FILE,MESSAGES_FILE,SESSIONS_FILE,FRIENDS_FILE,INBOX_FILE]:
+    if not os.path.exists(f): save_json(f,{})
+
+if not os.path.exists(LEADERBOARD_FILE):
+    pd.DataFrame(columns=['name','score','game_id','topics','timestamp','avatar']).to_csv(LEADERBOARD_FILE,index=False)
+
+if "uid" not in st.session_state: st.session_state['uid']=str(uuid.uuid4())
+if "session_id" not in st.session_state: st.session_state['session_id']=str(uuid.uuid4())
+
 # ---------------- UI ----------------
-st.title("AI Quiz Game — Single Player Only")
+st.title("🎯 AI Quiz Game — Online & Offline Multiplayer")
+
+# Mode selection
+mode = st.sidebar.selectbox("Select Mode",["Online","Offline"],0)
 
+# Player info
 if "username" not in st.session_state:
     st.session_state['username']=st.text_input("Enter your name","Player1")
 if "avatar" not in st.session_state:
     st.session_state['avatar']=st.selectbox("Choose avatar",["🎮","🐱","🐶","🦄","👽","🎩"])
 
-page=st.sidebar.selectbox("Page",["Home","Create Game","Join Game","Play","Leaderboard"],0)
+page=st.sidebar.selectbox("Page",["Home","Create Game","Join Game","Play","Friends","Inbox","Leaderboard"],0)
 
 # ---------------- Home ----------------
 if page=="Home":
@@ -185,7 +213,6 @@ elif page=="Create Game":
     if st.button("Create Game"):
         gid=create_game(st.session_state['username'],topics,num_q)
         st.success(f"Game created! Game ID: {gid}")
-        st.experimental_rerun()
 
 # ---------------- Join Game ----------------
 elif page=="Join Game":
@@ -194,12 +221,9 @@ elif page=="Join Game":
         ok,msg=join_game(gid,st.session_state['username'],st.session_state['avatar'])
         if ok: st.success(f"Joined Game {gid} successfully!")
         else: st.error(msg)
-        st.experimental_rerun()
 
 # ---------------- Play ----------------
 elif page=="Play":
-    st.markdown("---")
-    st.write("Offline mode stores data in ./data/")
     gid = st.text_input("Enter Game ID to Play")
     if gid:
         games = unified_get("games")
@@ -210,6 +234,7 @@ elif page=="Play":
             game = games[gid]
             username = st.session_state['username']
             avatar = st.session_state['avatar']
+
             # Join if not joined
             if gid not in players or username not in players[gid]:
                 ok,msg=join_game(gid,username,avatar)
@@ -217,9 +242,15 @@ elif page=="Play":
             # Claim session
             ok,msg=claim_session(gid,username)
             if not ok: st.warning(msg)
-            
+
             st.subheader(f"Game {gid} — Topics: {', '.join(game['topics'])}")
-            
+
+            # Offline graphics difference
+            if mode=="Offline":
+                st.markdown("### 🎮 Offline Mode — Local Play")
+            else:
+                st.markdown("### 🌐 Online Mode — Interactive Multiplayer")
+
             # Initialize session variables
             if 'play_index' not in st.session_state or st.session_state.get('current_gid')!=gid:
                 st.session_state['play_index']=0
@@ -227,19 +258,19 @@ elif page=="Play":
                 st.session_state['times']=[]
                 st.session_state['start_time']=time.time()
                 st.session_state['current_gid']=gid
-            
+
             questions = game['questions']
             qidx = st.session_state['play_index']
-            
+
             if qidx < len(questions):
                 q = questions[qidx]
                 st.write(f"Question {qidx+1}: {q[0]}")
                 ans = st.radio("Select Answer", q[1], key=f"ans{qidx}")
-                
+
                 # Timer
                 elapsed = int(time.time() - st.session_state['start_time'])
                 st.write(f"Time elapsed: {elapsed} sec")
-                
+
                 if st.button("Submit Answer"):
                     st.session_state['answers'].append(ans)
                     st.session_state['times'].append(elapsed)
@@ -247,50 +278,84 @@ elif page=="Play":
                     st.session_state['start_time']=time.time()
                     st.experimental_rerun()
             else:
-                # Game finished
                 total_score = compute_score(questions, st.session_state['answers'], st.session_state['times'])
                 st.success(f"Game Finished! Your score: {total_score}")
-                
+
                 # Update leaderboard
                 row={"name":username,"score":total_score,"game_id":gid,"topics":",".join(game['topics']),
                      "timestamp":now_iso(),"avatar":avatar}
                 unified_push_leaderboard(row)
-                
+
                 # Mark submitted
                 players[gid][username]['submitted']=True
                 unified_set("players",players)
-                
+
                 st.write("Correct Answers:")
                 for i,q in enumerate(questions):
                     st.write(f"Q{i+1}: {q[0]} — Correct: {q[2]} — Your Answer: {st.session_state['answers'][i]}")
-                
+
                 # Reset session variables
                 del st.session_state['play_index']
                 del st.session_state['answers']
                 del st.session_state['times']
                 del st.session_state['current_gid']
                 del st.session_state['start_time']
-            
-            # ---------------- Chat ----------------
-            st.markdown("---")
-            st.subheader("Game Chat")
-            msg_text = st.text_input("Type a message","",key="chat_input")
-            if st.button("Send Message"):
-                if msg_text.strip():
-                    unified_push_message(gid,{"from":username,"avatar":avatar,"msg":msg_text,"ts":now_iso()})
+
+            # ---------------- Chat (Only Online) ----------------
+            if mode=="Online":
+                st.markdown("---")
+                st.subheader("Game Chat")
+                msg_text = st.text_input("Type a message","",key="chat_input")
+                if st.button("Send Message"):
+                    if msg_text.strip():
+                        unified_push_message(gid,{"from":username,"avatar":avatar,"msg":msg_text,"ts":now_iso()})
+                        st.experimental_rerun()
+
+                msgs = unified_get("messages").get(gid,[])
+                for m in msgs[-20:]:
+                    st.write(f"{m['avatar']} **{m['from']}**: {m['msg']}")
+
+# ---------------- Friends ----------------
+elif page=="Friends":
+    if mode=="Online":
+        friends=get_friends()
+        inbox=get_inbox()
+        st.subheader("Friends")
+        myfriends=friends.get(st.session_state['username'],[])
+        st.write(myfriends)
+        newfriend=st.text_input("Add Friend")
+        if st.button("Send Friend Request"):
+            if newfriend: send_friend_request(st.session_state['username'],newfriend)
+        st.subheader("Inbox")
+        user_inbox=inbox.get(st.session_state['username'],[])
+        for msg in user_inbox:
+            if msg['type']=="friend_request":
+                st.write(f"Friend request from {msg['from']}")
+                if st.button(f"Accept {msg['from']}"):
+                    accept_friend_request(st.session_state['username'],msg['from'])
                     st.experimental_rerun()
-            
-            msgs = unified_get("messages").get(gid,[])
-            for m in msgs[-20:]:
-                st.write(f"{m['avatar']} **{m['from']}**: {m['msg']}")
+    else:
+        st.info("Friends feature is only available in Online Mode.")
+
+# ---------------- Inbox ----------------
+elif page=="Inbox":
+    if mode=="Online":
+        inbox=get_inbox()
+        user_inbox=inbox.get(st.session_state['username'],[])
+        st.subheader("Inbox Messages")
+        for msg in user_inbox:
+            st.write(msg)
+    else:
+        st.info("Inbox is only available in Online Mode.")
 
 # ---------------- Leaderboard ----------------
 elif page=="Leaderboard":
-    st.markdown("---")
-    st.write("Offline mode stores data in ./data/")
-    df=pd.read_csv(LEADERBOARD_FILE)
+    df=pd.DataFrame(unified_get("leaderboard"))
     st.subheader("Global Leaderboard")
     if not df.empty:
         df['display']=df['avatar']+" "+df['name']+" - "+df['score'].astype(str)
         st.write(df.sort_values("score",ascending=False)[['display']].head(20))
     else: st.write("No scores yet.")
+
+st.markdown("---")
+st.write("Data stored in ./data/. Online mode is interactive; Offline mode is local only.")