Update app.py

app.py

@@ -17,7 +17,7 @@ import streamlit.components.v1 as components
 from gradio_client import Client
 from streamlit_marquee import streamlit_marquee
 import folium
-from streamlit_folium import st_folium  # Use st_folium instead of folium_static
+from streamlit_folium import st_folium  # Using st_folium instead of folium_static
 import glob
 import pytz
 from collections import defaultdict
@@ -117,7 +117,89 @@ def load_username():
     return None
 
 # -----------------------------
-# Audio and Chat Processing Functions
+# Game State Functions
+# -----------------------------
+@st.cache_resource
+def load_game_state(_timestamp):
+    if os.path.exists(GAME_STATE_FILE):
+        with open(GAME_STATE_FILE, 'r') as f:
+            state = json.load(f)
+    else:
+        state = {
+            "players": {},
+            "treasures": [
+                {"id": str(uuid.uuid4()), "x": random.uniform(-40, 40), "z": random.uniform(-40, 40)}
+                for _ in range(5)
+            ],
+            "world_objects": [],
+            "history": []
+        }
+        with open(GAME_STATE_FILE, 'w') as f:
+            json.dump(state, f)
+    return state
+
+def update_game_state(state):
+    state["timestamp"] = os.path.getmtime(GAME_STATE_FILE) if os.path.exists(GAME_STATE_FILE) else time.time()
+    with open(GAME_STATE_FILE, 'w') as f:
+        json.dump(state, f)
+    load_game_state.clear()
+    return load_game_state(time.time())
+
+def reset_game_state():
+    state = {
+        "players": {},
+        "treasures": [
+            {"id": str(uuid.uuid4()), "x": random.uniform(-40, 40), "z": random.uniform(-40, 40)}
+            for _ in range(5)
+        ],
+        "world_objects": [],
+        "history": []
+    }
+    return update_game_state(state)
+
+# -----------------------------
+# PlayerAgent Class
+# -----------------------------
+class PlayerAgent:
+    def __init__(self, username, char_data):
+        self.username = username
+        self.x = random.uniform(-20, 20)
+        self.z = random.uniform(-40, 40)
+        self.color = char_data["color"]
+        self.shape = char_data["shape"]
+        self.score = 0
+        self.treasures = 0
+        self.last_active = time.time()
+        self.voice = char_data["voice"]
+
+    def to_dict(self):
+        return {
+            "username": self.username,
+            "x": self.x,
+            "z": self.z,
+            "color": self.color,
+            "shape": self.shape,
+            "score": self.score,
+            "treasures": self.treasures,
+            "last_active": self.last_active
+        }
+
+    def update_from_message(self, message):
+        if '|' in message:
+            _, content = message.split('|', 1)
+            if content.startswith("MOVE:"):
+                _, x, z = content.split(":")
+                self.x, self.z = float(x), float(z)
+                self.last_active = time.time()
+            elif content.startswith("SCORE:"):
+                self.score = int(content.split(":")[1])
+                self.last_active = time.time()
+            elif content.startswith("TREASURE:"):
+                self.treasures = int(content.split(":")[1])
+                self.last_active = time.time()
+
+# -----------------------------
+# Audio and Chat Processing
 # -----------------------------
 async def async_edge_tts_generate(text, voice, username):
     cache_key = f"{text[:100]}_{voice}"
@@ -149,24 +231,6 @@ async def generate_chat_audio(username, message, voice):
     audio_file = await async_edge_tts_generate(message, voice, username)
     return audio_file
 
-# -----------------------------
-# Broadcast Message (defined before usage)
-# -----------------------------
-async def broadcast_message(message, room_id):
-    if room_id in st.session_state.active_connections:
-        disconnected = []
-        for client_id, ws in st.session_state.active_connections[room_id].items():
-            try:
-                await ws.send(message)
-            except websockets.ConnectionClosed:
-                disconnected.append(client_id)
-        for client_id in disconnected:
-            if client_id in st.session_state.active_connections[room_id]:
-                del st.session_state.active_connections[room_id][client_id]
-
-# -----------------------------
-# Chat Entry Saving Function
-# -----------------------------
 async def save_chat_entry(username, message, voice, is_markdown=False):
     if not message.strip() or message == st.session_state.get('last_transcript', ''):
         return None, None
@@ -203,6 +267,21 @@ async def load_chat():
         content = f.read().strip()
     return content.split('\n')
 
+# -----------------------------
+# Broadcast Message Function
+# -----------------------------
+async def broadcast_message(message, room_id):
+    if room_id in st.session_state.active_connections:
+        disconnected = []
+        for client_id, ws in st.session_state.active_connections[room_id].items():
+            try:
+                await ws.send(message)
+            except websockets.ConnectionClosed:
+                disconnected.append(client_id)
+        for client_id in disconnected:
+            if client_id in st.session_state.active_connections[room_id]:
+                del st.session_state.active_connections[room_id][client_id]
+
 # -----------------------------
 # Session State Initialization
 # -----------------------------
@@ -383,7 +462,7 @@ def display_file_history_in_sidebar():
 
 def log_performance_metrics():
     st.sidebar.markdown("### ⏱️ Performance Metrics")
-    metrics = st.session_state['operation_timings']
+    metrics = st.session_state.get('operation_timings', {})
     if metrics:
         total_time = sum(metrics.values())
         st.sidebar.write(f"**Total Processing Time:** {total_time:.2f}s")
@@ -394,7 +473,6 @@ def log_performance_metrics():
 # -----------------------------
 # Enhanced 3D World HTML
 # -----------------------------
-# Precompute player attributes for correct interpolation.
 player_color = next(c["color"] for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
 player_shape = next(c["shape"] for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
 
@@ -565,7 +643,6 @@ rocky_map_html = f"""
         }}
         const mesh = playerMeshes[player.username];
         mesh.position.set(player.x, 1, player.z);
-        players[player.username] = {{ score: player.score, treasures: player.treasures }};
         if(player.username === playerName) {{
           xPos = player.x;
           zPos = player.z;
@@ -677,7 +754,7 @@ def main():
             st.session_state.last_activity = time.time()
             st.success(f"🌄 +10 points! New Score: ${st.session_state.score}")
 
-    # Display Quest Log as a Code Block (with Python formatting)
+    # Display Quest Log as a Python code block
     chat_content = asyncio.run(load_chat())
     st.code("\n".join(chat_content[-10:]), language="python")