Update app.py

app.py

@@ -51,7 +51,7 @@ PRAIRIE_LOCATIONS = {
     "Wyoming Spring Creek": (41.6666, -106.6666)
 }
 
-# Directories
+# Directories and Files
 for d in ["chat_logs", "audio_logs"]:
     os.makedirs(d, exist_ok=True)
 
@@ -59,6 +59,28 @@ CHAT_DIR = "chat_logs"
 AUDIO_DIR = "audio_logs"
 STATE_FILE = "user_state.txt"
 CHAT_FILE = os.path.join(CHAT_DIR, "quest_log.md")
+GAME_STATE_FILE = "game_state.json"
+
+# Cached Game State as "ML Model"
+@st.cache_data
+def load_game_state(_timestamp):
+    """Load or initialize the game state, treated as a cached 'model'."""
+    if os.path.exists(GAME_STATE_FILE):
+        with open(GAME_STATE_FILE, 'r') as f:
+            state = json.load(f)
+    else:
+        state = {"players": {}, "treasures": []}
+        with open(GAME_STATE_FILE, 'w') as f:
+            json.dump(state, f)
+    return state
+
+def update_game_state(state):
+    """Update the game state and persist to file."""
+    with open(GAME_STATE_FILE, 'w') as f:
+        json.dump(state, f)
+    # Clear cache to force reload with new timestamp
+    load_game_state.clear()
+    return load_game_state(time.time())
 
 # Helpers
 def format_timestamp(username=""):
@@ -162,7 +184,8 @@ def init_session_state():
         'speech_processed': False, 'players': {}, 'last_update': time.time(),
         'update_interval': 1, 'x_pos': 0, 'z_pos': 0, 'move_left': False,
         'move_right': False, 'move_up': False, 'move_down': False,
-        'prairie_players': {}, 'last_chat_update': 0
+        'prairie_players': {}, 'last_chat_update': 0,
+        'game_state_timestamp': time.time()
     }
     for k, v in defaults.items():
         if k not in st.session_state:
@@ -175,6 +198,17 @@ def init_session_state():
             st.session_state.username = random.choice(list(CHARACTERS.keys()))
             asyncio.run(save_chat_entry(st.session_state.username, "🗺️ Begins the Rocky Mountain Quest!", CHARACTERS[st.session_state.username]["voice"]))
         save_username(st.session_state.username)
+    # Initialize player in cached game state if new
+    game_state = load_game_state(st.session_state.game_state_timestamp)
+    if st.session_state.username not in game_state["players"]:
+        game_state["players"][st.session_state.username] = {
+            "x": random.uniform(-20, 20),
+            "z": random.uniform(-40, 40),
+            "color": CHARACTERS[st.session_state.username]["color"],
+            "score": 0,
+            "treasures": 0
+        }
+        update_game_state(game_state)
 
 init_session_state()
 
@@ -187,6 +221,7 @@ async def websocket_handler(websocket, path):
     st.session_state.active_connections[room_id][client_id] = websocket
     username = st.session_state.username
     
+    game_state = load_game_state(st.session_state.game_state_timestamp)
     if "prairie" in path:
         st.session_state.prairie_players[client_id] = {
             "username": username,
@@ -196,14 +231,16 @@ async def websocket_handler(websocket, path):
         }
         await broadcast_message(f"System|{username} joins the prairie!", room_id)
     else:
-        st.session_state.players[client_id] = {
-            "username": username,
-            "x": random.uniform(-20, 20),
-            "z": random.uniform(-50, 50),
-            "color": CHARACTERS[username]["color"],
-            "score": 0,
-            "treasures": 0
-        }
+        if username not in game_state["players"]:
+            game_state["players"][username] = {
+                "x": random.uniform(-20, 20),
+                "z": random.uniform(-40, 40),
+                "color": CHARACTERS[username]["color"],
+                "score": 0,
+                "treasures": 0
+            }
+            update_game_state(game_state)
+        st.session_state.players[client_id] = game_state["players"][username]
         await broadcast_message(f"System|{username} joins the quest!", room_id)
     
     try:
@@ -211,10 +248,30 @@ async def websocket_handler(websocket, path):
             if '|' in message:
                 sender, content = message.split('|', 1)
                 voice = CHARACTERS.get(sender, {"voice": "en-US-AriaNeural"})["voice"]
+                game_state = load_game_state(st.session_state.game_state_timestamp)
                 if content.startswith("MOVE:"):
                     _, x, z = content.split(":")
-                    st.session_state.players[client_id]["x"] = float(x)
-                    st.session_state.players[client_id]["z"] = float(z)
+                    x, z = float(x), float(z)
+                    st.session_state.players[client_id]["x"] = x
+                    st.session_state.players[client_id]["z"] = z
+                    game_state["players"][sender]["x"] = x
+                    game_state["players"][sender]["z"] = z
+                    update_game_state(game_state)
+                elif content.startswith("SCORE:"):
+                    score = int(content.split(":")[1])
+                    st.session_state.players[client_id]["score"] = score
+                    game_state["players"][sender]["score"] = score
+                    update_game_state(game_state)
+                elif content.startswith("TREASURE:"):
+                    treasures = int(content.split(":")[1])
+                    st.session_state.players[client_id]["treasures"] = treasures
+                    game_state["players"][sender]["treasures"] = treasures
+                    for i, t in enumerate(game_state["treasures"]):
+                        if st.session_state.players[client_id]["x"] - 2 < t["x"] < st.session_state.players[client_id]["x"] + 2 and \
+                           st.session_state.players[client_id]["z"] - 2 < t["z"] < st.session_state.players[client_id]["z"] + 2:
+                            game_state["treasures"].pop(i)
+                            break
+                    update_game_state(game_state)
                 elif content.startswith("PRAIRIE:"):
                     action, value = content.split(":", 1)
                     if action == "ANIMAL":
@@ -224,10 +281,6 @@ async def websocket_handler(websocket, path):
                         st.session_state.prairie_players[client_id]["location"] = target
                         action_msg = f"{sender} ({st.session_state.prairie_players[client_id]['animal']}) moves to {value}"
                         await save_chat_entry(sender, action_msg, voice)
-                elif content.startswith("SCORE:"):
-                    st.session_state.players[client_id]["score"] = int(content.split(":")[1])
-                elif content.startswith("TREASURE:"):
-                    st.session_state.players[client_id]["treasures"] = int(content.split(":")[1])
                 else:
                     await save_chat_entry(sender, content, voice)
                     await perform_arxiv_search(content, sender)
@@ -244,9 +297,10 @@ async def websocket_handler(websocket, path):
 async def periodic_update():
     while True:
         if st.session_state.active_connections.get("quest"):
+            game_state = load_game_state(st.session_state.game_state_timestamp)
             player_list = ", ".join([p["username"] for p in st.session_state.players.values()]) or "No adventurers yet!"
             message = f"📢 Quest Update: Active Adventurers - {player_list}"
-            player_data = json.dumps(list(st.session_state.players.values()))
+            player_data = json.dumps(list(game_state["players"].values()))
             await broadcast_message(f"MAP_UPDATE:{player_data}", "quest")
             
             prairie_list = ", ".join([f"{p['username']} ({p['animal']})" for p in st.session_state.prairie_players.values()]) or "No animals yet!"
@@ -254,9 +308,9 @@ async def periodic_update():
             prairie_data = json.dumps(list(st.session_state.prairie_players.values()))
             await broadcast_message(f"PRAIRIE_UPDATE:{prairie_data}", "quest")
             
-            # Multicast chat update every second
             chat_content = await load_chat()
             await broadcast_message(f"CHAT_UPDATE:{json.dumps(chat_content[-10:])}", "quest")
+            await broadcast_message(f"GAME_STATE:{json.dumps(game_state)}", "quest")
             await save_chat_entry("System", f"{message}\n{prairie_message}", "en-US-AriaNeural")
         await asyncio.sleep(st.session_state.update_interval)
 
@@ -351,6 +405,7 @@ rocky_map_html = f"""
         let players = {{}};
         const playerMeshes = {{}};
         let treasures = [];
+        const treasureMeshes = {{}};
         let xPos = 0, zPos = 0, moveLeft = false, moveRight = false, moveUp = false, moveDown = false, collect = false;
         let score = 0, treasureCount = 0;
 
@@ -363,22 +418,31 @@ rocky_map_html = f"""
         scene.add(playerMesh);
         players[playerName] = {{ mesh: playerMesh, score: 0, treasures: 0 }};
 
-        // Treasure spawning
-        function spawnTreasure() {{
-            if (treasures.length < 5) {{
-                const treasure = new THREE.Mesh(
-                    new THREE.SphereGeometry(1, 8, 8),
-                    new THREE.MeshPhongMaterial({{ color: 0xffff00 }})
-                );
-                treasure.position.set(
-                    Math.random() * 80 - 40,
-                    1,
-                    Math.random() * 80 - 40
-                );
-                treasure.castShadow = true;
-                treasures.push(treasure);
-                scene.add(treasure);
-            }}
+        // Treasure updating
+        function updateTreasures(treasureData) {{
+            treasureData.forEach(t => {{
+                if (!treasureMeshes[t.id]) {{
+                    const treasure = new THREE.Mesh(
+                        new THREE.SphereGeometry(1, 8, 8),
+                        new THREE.MeshPhongMaterial({{ color: 0xffff00 }})
+                    );
+                    treasure.position.set(t.x, 1, t.z);
+                    treasure.castShadow = true;
+                    treasureMeshes[t.id] = treasure;
+                    treasures.push(treasure);
+                    scene.add(treasure);
+                }} else {{
+                    treasureMeshes[t.id].position.set(t.x, 1, t.z);
+                }}
+            }});
+            // Remove treasures not in the data
+            Object.keys(treasureMeshes).forEach(id => {{
+                if (!treasureData.some(t => t.id === id)) {{
+                    scene.remove(treasureMeshes[id]);
+                    treasures = treasures.filter(t => t !== treasureMeshes[id]);
+                    delete treasureMeshes[id];
+                }}
+            }});
         }}
 
         // Controls
@@ -413,13 +477,14 @@ rocky_map_html = f"""
             if (collect) {{
                 for (let i = treasures.length - 1; i >= 0; i--) {{
                     if (playerMesh.position.distanceTo(treasures[i].position) < 2) {{
+                        const id = Object.keys(treasureMeshes).find(key => treasureMeshes[key] === treasures[i]);
                         scene.remove(treasures[i]);
                         treasures.splice(i, 1);
+                        delete treasureMeshes[id];
                         score += 50;
                         treasureCount += 1;
                         ws.send(`${{playerName}}|SCORE:${{score}}`);
                         ws.send(`${{playerName}}|TREASURE:${{treasureCount}}`);
-                        spawnTreasure();
                     }}
                 }}
             }}
@@ -428,13 +493,6 @@ rocky_map_html = f"""
             camera.lookAt(xPos, 0, zPos);
         }}
 
-        function updateTreasures(delta) {{
-            treasures.forEach(treasure => {{
-                treasure.rotation.y += delta;
-            }});
-            if (Math.random() < 0.01) spawnTreasure();
-        }}
-
         function updatePlayers(playerData) {{
             playerData.forEach(player => {{
                 if (!playerMeshes[player.username]) {{
@@ -446,6 +504,13 @@ rocky_map_html = f"""
                 const mesh = playerMeshes[player.username];
                 mesh.position.set(player.x, 1, player.z);
                 players[player.username] = {{ mesh: mesh, score: player.score, treasures: player.treasures }};
+                if (player.username === playerName) {{
+                    xPos = player.x;
+                    zPos = player.z;
+                    score = player.score;
+                    treasureCount = player.treasures;
+                    playerMesh.position.set(xPos, 1, zPos);
+                }}
             }});
             document.getElementById('players').textContent = `Players: ${{Object.keys(playerMeshes).length}}`;
             document.getElementById('score').textContent = `Score: ${{score}}`;
@@ -462,6 +527,10 @@ rocky_map_html = f"""
                 const chatBox = document.getElementById('chatBox');
                 chatBox.innerHTML = chatData.map(line => `<p>${{line}}</p>`).join('');
                 chatBox.scrollTop = chatBox.scrollHeight;
+            }} else if (data.startsWith('GAME_STATE:')) {{
+                const gameState = JSON.parse(data.split('GAME_STATE:')[1]);
+                updatePlayers(gameState.players);
+                updateTreasures(gameState.treasures);
             }} else if (!data.startsWith('PRAIRIE_UPDATE:')) {{
                 const [sender, message] = data.split('|');
                 const chatBox = document.getElementById('chatBox');
@@ -478,12 +547,10 @@ rocky_map_html = f"""
             lastTime = currentTime;
 
             updatePlayer(delta);
-            updateTreasures(delta);
+            treasures.forEach(t => t.rotation.y += delta);
             renderer.render(scene, camera);
         }}
 
-        // Initial spawn
-        for (let i = 0; i < 5; i++) spawnTreasure();
         animate();
     </script>
 </body>
@@ -500,6 +567,9 @@ def main():
     st.sidebar.write(f"🎵 Treasures: {st.session_state.treasures}")
     st.sidebar.write(f"👥 Players: {', '.join([p['username'] for p in st.session_state.players.values()]) or 'None'}")
     
+    # User-controlled refresh interval
+    st.session_state.update_interval = st.sidebar.slider("Refresh Interval (seconds)", 1, 10, 1, step=1)
+
     new_username = st.sidebar.selectbox("🧙‍♂️ Choose Your Hero", list(CHARACTERS.keys()), index=list(CHARACTERS.keys()).index(st.session_state.username))
     if new_username != st.session_state.username:
         asyncio.run(save_chat_entry(st.session_state.username, f"🔄 Transforms into {new_username}!", CHARACTERS[st.session_state.username]["voice"]))
@@ -561,6 +631,7 @@ def main():
     st.sidebar.markdown(f"⏳ Next Update in: {int(remaining)}s")
     if remaining <= 0:
         st.session_state.last_update = time.time()
+        st.session_state.game_state_timestamp = time.time()  # Update timestamp to refresh cache
         st.rerun()
 
     if not st.session_state.get('server_running', False):