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  # Updated: use st_folium instead of folium_static
+from streamlit_folium import st_folium  # Use st_folium instead of folium_static
 import glob
 import pytz
 from collections import defaultdict
@@ -55,11 +55,11 @@ EDGE_TTS_VOICES = [
 ]
 FILE_EMOJIS = {"md": "📜", "mp3": "🎵"}
 WORLD_OBJECTS = [
-    {"type": "🌿 Plants", "emoji": "🌱", "color": 0x228B22, "shape": "cylinder", "count": 10},
-    {"type": "🐦 Animal Flocks", "emoji": "🕊️", "color": 0x87CEEB, "shape": "sphere", "count": 5},
-    {"type": "🏔️ Mountains", "emoji": "🏔️", "color": 0x808080, "shape": "cone", "count": 3},
-    {"type": "🌳 Trees", "emoji": "🌲", "color": 0x006400, "shape": "cylinder", "count": 8},
-    {"type": "🐾 Animals", "emoji": "🐻", "color": 0x8B4513, "shape": "cube", "count": 6}
+    {"type": "🌿 Plants", "emoji": "🌱🌿🌾", "color": 0x228B22, "shape": "cylinder", "count": 10},
+    {"type": "🐦 Animal Flocks", "emoji": "🕊️🐦🐤", "color": 0x87CEEB, "shape": "sphere", "count": 5},
+    {"type": "🏔️ Mountains", "emoji": "🏔️⛰️", "color": 0x808080, "shape": "cone", "count": 3},
+    {"type": "🌳 Trees", "emoji": "🌲🌳🌴", "color": 0x006400, "shape": "cylinder", "count": 8},
+    {"type": "🐾 Animals", "emoji": "🐻🐺🦌", "color": 0x8B4513, "shape": "cube", "count": 6}
 ]
 PRAIRIE_LOCATIONS = {
     "Deadwood, SD": (44.3769, -103.7298),
@@ -70,7 +70,6 @@ PRAIRIE_LOCATIONS = {
 # Directories and Files
 for d in ["chat_logs", "audio_logs"]:
     os.makedirs(d, exist_ok=True)
-
 CHAT_DIR = "chat_logs"
 AUDIO_DIR = "audio_logs"
 STATE_FILE = "user_state.txt"
@@ -80,7 +79,6 @@ GAME_STATE_FILE = "game_state.json"
 # Cached Game State as "Resource"
 @st.cache_resource
 def load_game_state(_timestamp):
-    """Load or initialize the game state, treated as a cached resource."""
     if os.path.exists(GAME_STATE_FILE):
         with open(GAME_STATE_FILE, 'r') as f:
             state = json.load(f)
@@ -99,7 +97,6 @@ def load_game_state(_timestamp):
     return state
 
 def update_game_state(state):
-    """Update the game state and persist to file."""
     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)
@@ -107,7 +104,6 @@ def update_game_state(state):
     return load_game_state(time.time())
 
 def reset_game_state():
-    """Reset the game state to initial conditions."""
     state = {
         "players": {},
         "treasures": [
@@ -119,44 +115,11 @@ def reset_game_state():
     }
     return update_game_state(state)
 
-# Timestamp Formatter (Pattern: mmddYYYY-HHMM-AM/PM)
-def format_timestamp(username=""):
-    now = datetime.now(pytz.timezone('US/Central')).strftime("%m%d%Y-%I%M-%p")
-    return f"{now}-by-{username}"
-
-def clean_text_for_tts(text):
-    return re.sub(r'[#*!\[\]]+', '', ' '.join(text.split()))[:200] or "No text"
-
-def generate_filename(prompt, username, file_type="md"):
-    timestamp = format_timestamp(username)
-    cleaned_prompt = clean_text_for_tts(prompt)[:50].replace(" ", "_")
-    return f"{cleaned_prompt}_{timestamp}.{file_type}"
-
-def create_file(prompt, username, file_type="md"):
-    filename = generate_filename(prompt, username, file_type)
-    with open(filename, 'w', encoding='utf-8') as f:
-        f.write(prompt)
-    return filename
-
-def get_download_link(file, file_type="mp3"):
-    with open(file, "rb") as f:
-        b64 = base64.b64encode(f.read()).decode()
-    mime_types = {"mp3": "audio/mpeg", "md": "text/markdown"}
-    return f'<a href="data:{mime_types.get(file_type, "application/octet-stream")};base64,{b64}" download="{os.path.basename(file)}">{FILE_EMOJIS.get(file_type, "📥")} {os.path.basename(file)}</a>'
-
-def save_username(username):
-    with open(STATE_FILE, 'w') as f:
-        f.write(username)
-
-def load_username():
-    if os.path.exists(STATE_FILE):
-        with open(STATE_FILE, 'r') as f:
-            return f.read().strip()
-    return None
+# Precompute player's attributes for the 3D scene.
+player_color = None
+player_shape = None
 
-# -----------------------------------------------------------
-# Define PlayerAgent class (used in session state initialization)
-# -----------------------------------------------------------
+# Define PlayerAgent class
 class PlayerAgent:
     def __init__(self, username, char_data):
         self.username = username
@@ -195,11 +158,8 @@ class PlayerAgent:
                 self.treasures = int(content.split(":")[1])
                 self.last_active = time.time()
 
-# -----------------------------------------------------------
-# New Helper Functions: Chat Logging & TTS Generation
-# -----------------------------------------------------------
+# Helpers for chat logging and TTS generation
 def log_chat_history(username, message, is_markdown=False):
-    """Log the chat entry to a markdown file."""
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     if is_markdown:
         entry = f"[{timestamp}] {username}:\n```markdown\n{message}\n```"
@@ -208,16 +168,12 @@ def log_chat_history(username, message, is_markdown=False):
     md_file = create_file(entry, username, "md")
     with open(CHAT_FILE, 'a') as f:
         f.write(f"{entry}\n")
-    return md_file
+    return entry
 
 async def generate_chat_audio(username, message, voice):
-    """Generate TTS audio file for the chat message."""
     audio_file = await async_edge_tts_generate(message, voice, username)
     return audio_file
 
-# -----------------------------------------------------------
-# Audio Processing Function
-# -----------------------------------------------------------
 async def async_edge_tts_generate(text, voice, username):
     cache_key = f"{text[:100]}_{voice}"
     if cache_key in st.session_state['audio_cache']:
@@ -236,9 +192,6 @@ def play_and_download_audio(file_path):
         st.audio(file_path, start_time=0)
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)
 
-# -----------------------------------------------------------
-# WebSocket Broadcast Function
-# -----------------------------------------------------------
 async def broadcast_message(message, room_id):
     if room_id in st.session_state.active_connections:
         disconnected = []
@@ -251,13 +204,10 @@ async def broadcast_message(message, room_id):
             if client_id in st.session_state.active_connections[room_id]:
                 del st.session_state.active_connections[room_id][client_id]
 
-# -----------------------------------------------------------
-# Wrapper Function to Save Chat Entry (Using the Two Helpers Above)
-# -----------------------------------------------------------
 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
-    md_file = log_chat_history(username, message, is_markdown)
+    entry = log_chat_history(username, message, is_markdown)
     audio_file = await generate_chat_audio(username, message, voice)
     await broadcast_message(f"{username}|{message}", "quest")
     st.session_state.chat_history.append({"username": username, "message": message, "audio": audio_file})
@@ -267,7 +217,7 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
         game_state["players"][username]["score"] += 10
         game_state["players"][username]["treasures"] += 1
         game_state["players"][username]["last_active"] = time.time()
-    game_state["history"].append(md_file)
+    game_state["history"].append(entry)
     if message.lower() in ["plants", "animal flocks", "mountains", "trees", "animals"]:
         obj_type = next(o for o in WORLD_OBJECTS if message.lower() in o["type"].lower())
         for _ in range(obj_type["count"]):
@@ -280,7 +230,7 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
                 "shape": obj_type["shape"]
             })
     update_game_state(game_state)
-    return md_file, audio_file
+    return entry, audio_file
 
 async def load_chat():
     if not os.path.exists(CHAT_FILE):
@@ -290,9 +240,6 @@ async def load_chat():
         content = f.read().strip()
     return content.split('\n')
 
-# -----------------------------------------------------------
-# Session State Initialization
-# -----------------------------------------------------------
 def init_session_state():
     defaults = {
         'server_running': False, 'server_task': None, 'active_connections': {},
@@ -314,6 +261,11 @@ def init_session_state():
         st.session_state.username = char["name"]
         st.session_state.tts_voice = char["voice"]
         asyncio.run(save_chat_entry(st.session_state.username, f"🗺️ Welcome to the Rocky Mountain Quest, {st.session_state.username}!", char["voice"]))
+    global player_color, player_shape
+    if player_color is None or player_shape is None:
+        char = next(c for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
+        player_color = char["color"]
+        player_shape = char["shape"]
     game_state = load_game_state(st.session_state.game_state_timestamp)
     if st.session_state.username not in game_state["players"]:
         char = next(c for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
@@ -323,12 +275,7 @@ def init_session_state():
 
 init_session_state()
 
-# -----------------------------------------------------------
 # WebSocket Handler and Related Functions
-# -----------------------------------------------------------
-async def save_chat_entry_wrapper(username, message, voice, is_markdown=False):
-    return await save_chat_entry(username, message, voice, is_markdown)
-
 async def websocket_handler(websocket, path):
     client_id = str(uuid.uuid4())
     room_id = "quest"
@@ -372,9 +319,10 @@ async def websocket_handler(websocket, path):
                         target = PRAIRIE_LOCATIONS.get(value, PRAIRIE_LOCATIONS["Deadwood, SD"])
                         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_wrapper(sender, action_msg, voice)
+                        await save_chat_entry(sender, action_msg, voice)
                 else:
-                    await save_chat_entry_wrapper(sender, content, voice)
+                    await save_chat_entry(sender, content, voice)
+                    await perform_arxiv_search(content, sender)
                 update_game_state(game_state)
     except websockets.ConnectionClosed:
         await broadcast_message(f"System|{username} leaves the quest!", room_id)
@@ -409,7 +357,7 @@ async def periodic_update():
             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_wrapper("System", f"{message}\n{prairie_message}", "en-US-AriaNeural")
+            await save_chat_entry("System", f"{message}\n{prairie_message}", "en-US-AriaNeural")
         await asyncio.sleep(st.session_state.update_interval)
 
 async def run_websocket_server():
@@ -424,9 +372,6 @@ def start_websocket_server():
     asyncio.set_event_loop(loop)
     loop.run_until_complete(run_websocket_server())
 
-# -----------------------------------------------------------
-# ArXiv Integration (Unchanged)
-# -----------------------------------------------------------
 async def perform_arxiv_search(query, username):
     gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     refs = gradio_client.predict(
@@ -434,12 +379,9 @@ async def perform_arxiv_search(query, username):
     )[0]
     result = f"📚 Ancient Rocky Knowledge:\n{refs}"
     voice = next(c["voice"] for c in EDGE_TTS_VOICES if c["name"] == username)
-    md_file, audio_file = await save_chat_entry_wrapper(username, result, voice, True)
+    md_file, audio_file = await save_chat_entry(username, result, voice, True)
     return md_file, audio_file
 
-# -----------------------------------------------------------
-# Sidebar Functions
-# -----------------------------------------------------------
 def update_marquee_settings_ui():
     st.sidebar.markdown("### 🎯 Marquee Settings")
     cols = st.sidebar.columns(2)
@@ -478,20 +420,17 @@ def log_performance_metrics():
             percentage = (duration / total_time) * 100
             st.sidebar.write(f"**{operation}:** {duration:.2f}s ({percentage:.1f}%)")
 
-# -----------------------------------------------------------
-# Enhanced 3D Game HTML (With dynamic insertion of username and fluid movement)
-# -----------------------------------------------------------
-# Precompute player color and shape for use in the HTML.
-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)
-
+# ---------------------------
+# Enhanced 3D Game HTML
+# ---------------------------
+# Use precomputed player_color and player_shape in the JS code.
 rocky_map_html = f"""
 <!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Rocky Mountain Quest 3D</title>
+    <title>{GAME_NAME}</title>
     <style>
         body {{ margin: 0; overflow: hidden; font-family: Arial, sans-serif; background: #000; }}
         #gameContainer {{ width: 800px; height: 600px; position: relative; }}
@@ -528,7 +467,7 @@ rocky_map_html = f"""
         <div class="leaderboard" id="leaderboard">Leaderboard</div>
         <div id="chatBox"></div>
         <div class="controls">
-            <p>Controls: Use WASD or Arrow keys for movement. Space to collect treasure.</p>
+            <p>Controls: WASD/Arrows to move, Space to collect treasure</p>
             <p>Chat to add world features!</p>
         </div>
     </div>
@@ -538,33 +477,33 @@ rocky_map_html = f"""
         const playerName = "{st.session_state.username}";
         let ws = new WebSocket('ws://localhost:8765');
         const scene = new THREE.Scene();
-        const camera = new THREE.PerspectiveCamera(75, 800 / 600, 0.1, 1000);
+        const camera = new THREE.PerspectiveCamera(75, 800/600, 0.1, 1000);
         camera.position.set(0, 50, 50);
         camera.lookAt(0, 0, 0);
+
         const renderer = new THREE.WebGLRenderer({{ antialias: true }});
         renderer.setSize(800, 600);
         document.getElementById('gameContainer').appendChild(renderer.domElement);
 
-        // Lighting & Ground
         const ambientLight = new THREE.AmbientLight(0xffffff, 0.5);
         scene.add(ambientLight);
         const sunLight = new THREE.DirectionalLight(0xffddaa, 1);
         sunLight.position.set(50, 50, 50);
         sunLight.castShadow = true;
         scene.add(sunLight);
+
         const groundGeometry = new THREE.PlaneGeometry(100, 100);
         const groundMaterial = new THREE.MeshStandardMaterial({{ color: 0x228B22 }});
         const ground = new THREE.Mesh(groundGeometry, groundMaterial);
-        ground.rotation.x = -Math.PI / 2;
+        ground.rotation.x = -Math.PI/2;
         ground.receiveShadow = true;
         scene.add(ground);
 
-        // Fluid Movement Variables
+        // Fluid movement variables (you can adjust acceleration/friction as needed)
         let velocity = new THREE.Vector2(0, 0);
         const acceleration = 50;
         const friction = 5;
 
-        // Player Setup
         let xPos = 0, zPos = 0;
         const shapeGeometries = {{
             "sphere": new THREE.SphereGeometry(1, 16, 16),
@@ -582,22 +521,23 @@ rocky_map_html = f"""
         playerMesh.position.set(xPos, 1, zPos);
         playerMesh.castShadow = true;
         scene.add(playerMesh);
+
         let score = 0, treasureCount = 0;
-        let lastActive = performance.now() / 1000;
+        let lastActive = performance.now()/1000;
         let moveLeft = false, moveRight = false, moveUp = false, moveDown = false, collect = false;
 
         document.addEventListener('keydown', (event) => {{
-            switch (event.code) {{
+            switch(event.code) {{
                 case 'ArrowLeft': case 'KeyA': moveLeft = true; break;
                 case 'ArrowRight': case 'KeyD': moveRight = true; break;
                 case 'ArrowUp': case 'KeyW': moveUp = true; break;
                 case 'ArrowDown': case 'KeyS': moveDown = true; break;
                 case 'Space': collect = true; break;
             }}
-            lastActive = performance.now() / 1000;
+            lastActive = performance.now()/1000;
         }});
         document.addEventListener('keyup', (event) => {{
-            switch (event.code) {{
+            switch(event.code) {{
                 case 'ArrowLeft': case 'KeyA': moveLeft = false; break;
                 case 'ArrowRight': case 'KeyD': moveRight = false; break;
                 case 'ArrowUp': case 'KeyW': moveUp = false; break;
@@ -607,10 +547,10 @@ rocky_map_html = f"""
         }});
 
         function updatePlayer(delta) {{
-            if (moveLeft) velocity.x -= acceleration * delta;
-            if (moveRight) velocity.x += acceleration * delta;
-            if (moveUp) velocity.y -= acceleration * delta;
-            if (moveDown) velocity.y += acceleration * delta;
+            if(moveLeft) velocity.x -= acceleration * delta;
+            if(moveRight) velocity.x += acceleration * delta;
+            if(moveUp) velocity.y -= acceleration * delta;
+            if(moveDown) velocity.y += acceleration * delta;
             velocity.x -= velocity.x * friction * delta;
             velocity.y -= velocity.y * friction * delta;
             xPos += velocity.x * delta;
@@ -619,9 +559,9 @@ rocky_map_html = f"""
             zPos = Math.max(-40, Math.min(40, zPos));
             playerMesh.position.set(xPos, 1, zPos);
             ws.send(`${{playerName}}|MOVE:${{xPos}}:${{zPos}}`);
-            if (collect) {{
-                for (let i = treasures.length - 1; i >= 0; i--) {{
-                    if (playerMesh.position.distanceTo(treasures[i].position) < 2) {{
+            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);
@@ -642,7 +582,7 @@ rocky_map_html = f"""
 
         function updatePlayers(playerData) {{
             playerData.forEach(player => {{
-                if (!playerMeshes[player.username]) {{
+                if(!playerMeshes[player.username]) {{
                     const geometry = shapeGeometries[player.shape] || new THREE.BoxGeometry(2, 2, 2);
                     const material = new THREE.MeshPhongMaterial({{ color: player.color }});
                     const mesh = new THREE.Mesh(geometry, material);
@@ -653,7 +593,7 @@ 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) {{
+                if(player.username === playerName) {{
                     xPos = player.x;
                     zPos = player.z;
                     score = player.score;
@@ -670,7 +610,7 @@ rocky_map_html = f"""
 
         function updateTreasures(treasureData) {{
             treasureData.forEach(t => {{
-                if (!treasureMeshes[t.id]) {{
+                if(!treasureMeshes[t.id]) {{
                     const treasure = new THREE.Mesh(
                         new THREE.SphereGeometry(1, 8, 8),
                         new THREE.MeshPhongMaterial({{ color: 0xffff00 }})
@@ -685,7 +625,7 @@ rocky_map_html = f"""
                 }}
             }});
             Object.keys(treasureMeshes).forEach(id => {{
-                if (!treasureData.some(t => t.id === id)) {{
+                if(!treasureData.some(t => t.id === id)) {{
                     scene.remove(treasureMeshes[id]);
                     treasures = treasures.filter(t => t !== treasureMeshes[id]);
                     delete treasureMeshes[id];
@@ -695,7 +635,7 @@ rocky_map_html = f"""
 
         function updateWorldObjects(objectData) {{
             objectData.forEach(obj => {{
-                if (!worldObjectMeshes[obj.type + obj.x + obj.z]) {{
+                if(!worldObjectMeshes[obj.type + obj.x + obj.z]) {{
                     const geometry = shapeGeometries[obj.shape] || new THREE.BoxGeometry(2, 2, 2);
                     const material = new THREE.MeshPhongMaterial({{ color: obj.color }});
                     const objMesh = new THREE.Mesh(geometry, material);
@@ -710,20 +650,20 @@ rocky_map_html = f"""
 
         ws.onmessage = function(event) {{
             const data = event.data;
-            if (data.startsWith('MAP_UPDATE:')) {{
+            if(data.startsWith('MAP_UPDATE:')) {{
                 const playerData = JSON.parse(data.split('MAP_UPDATE:')[1]);
                 updatePlayers(playerData);
-            }} else if (data.startsWith('CHAT_UPDATE:')) {{
+            }} else if(data.startsWith('CHAT_UPDATE:')) {{
                 const chatData = JSON.parse(data.split('CHAT_UPDATE:')[1]);
                 const chatBox = document.getElementById('chatBox');
                 chatBox.innerHTML = chatData.map(line => `<p>${{line}}</p>`).join('');
                 chatBox.scrollTop = chatBox.scrollHeight;
-            }} else if (data.startsWith('GAME_STATE:')) {{
+            }} else if(data.startsWith('GAME_STATE:')) {{
                 const gameState = JSON.parse(data.split('GAME_STATE:')[1]);
                 updatePlayers(gameState.players);
                 updateTreasures(gameState.treasures);
                 updateWorldObjects(gameState.world_objects);
-            }} else if (!data.startsWith('PRAIRIE_UPDATE:')) {{
+            }} else if(!data.startsWith('PRAIRIE_UPDATE:')) {{
                 const [sender, message] = data.split('|');
                 const chatBox = document.getElementById('chatBox');
                 chatBox.innerHTML += `<p>${{sender}}: ${{message}}</p>`;
@@ -748,22 +688,23 @@ rocky_map_html = f"""
 </html>
 """
 
-# -----------------------------------------------------------
-# Main Game Loop
-# -----------------------------------------------------------
 def main():
     st.markdown(f"<h2 style='text-align: center;'>Welcome, {st.session_state.username}!</h2>", unsafe_allow_html=True)
-    # Chat Input Area
+    # Display Chat Input above the Quest Log
     message = st.text_input(f"🗨️ Chat as {st.session_state.username}:", placeholder="Type to chat or add world features! 🌲", key="chat_input")
     if st.button("🌟 Send"):
         if message:
             voice = next(c["voice"] for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
-            md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, message, voice))
+            entry, audio_file = asyncio.run(save_chat_entry(st.session_state.username, message, voice))
             if audio_file:
                 play_and_download_audio(audio_file)
             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 and line numbers
+    chat_content = asyncio.run(load_chat())
+    st.code("\n".join(chat_content[-10:]), language="python")
+
     update_marquee_settings_ui()
     st.sidebar.title(f"🎮 {GAME_NAME}")
     st.sidebar.subheader(f"🌄 {st.session_state.username}’s Adventure - Score: ${st.session_state.score} 🏆")
@@ -789,10 +730,6 @@ def main():
 
     left_col, right_col = st.columns([2, 1])
     with left_col:
-        # Display the Quest Log as a code block with Python syntax and line numbers (if supported)
-        chat_content = asyncio.run(load_chat())
-        st.code("\n".join(chat_content[-10:]), language="python")
-        # Then display the 3D world
         components.html(rocky_map_html, width=800, height=600)
         uploaded_file = st.file_uploader("Upload a File 📤", type=["txt", "md", "mp3"])
         if uploaded_file:
@@ -805,7 +742,6 @@ def main():
             update_game_state(game_state)
             st.session_state.last_activity = time.time()
             st.success(f"File uploaded! +20 points! New Score: ${st.session_state.score}")
-            # Speech component integration remains as before.
             mycomponent = components.declare_component("speech_component", path="./speech_component")
             val = mycomponent(my_input_value="", key=f"speech_{st.session_state.get('speech_processed', False)}")
             if val and val != st.session_state.last_transcript:
@@ -813,7 +749,7 @@ def main():
                 if val_stripped:
                     voice = next(c["voice"] for c in EDGE_TTS_VOICES if c["name"] == st.session_state.username)
                     st.session_state['speech_processed'] = True
-                    md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, val_stripped, voice))
+                    entry, audio_file = asyncio.run(save_chat_entry(st.session_state.username, val_stripped, voice))
                     if audio_file:
                         play_and_download_audio(audio_file)
                     st.session_state.last_activity = time.time()
@@ -821,11 +757,9 @@ def main():
     with right_col:
         st.subheader("🌾 Prairie Map")
         prairie_map = folium.Map(location=[44.0, -103.0], zoom_start=8, tiles="CartoDB Positron")
-        # Add base locations
         for loc, (lat, lon) in PRAIRIE_LOCATIONS.items():
             folium.Marker([lat, lon], popup=loc).add_to(prairie_map)
         game_state = load_game_state(st.session_state.game_state_timestamp)
-        # Mark players
         for username, player in game_state["players"].items():
             folium.CircleMarker(
                 location=[44.0 + player["x"] * 0.01, -103.0 + player["z"] * 0.01],
@@ -835,7 +769,6 @@ def main():
                 fill_opacity=0.7,
                 popup=f"{username} (Score: ${player['score']}, Treasures: {player['treasures']})"
             ).add_to(prairie_map)
-        # Mark prairie players (animals)
         for client_id, player in st.session_state.prairie_players.items():
             folium.CircleMarker(
                 location=player['location'],
@@ -845,7 +778,6 @@ def main():
                 fill_opacity=0.7,
                 popup=f"{player['username']} ({player['animal']})"
             ).add_to(prairie_map)
-        # Mark world objects (plants, trees, etc.) with emojis and pins
         for obj in game_state["world_objects"]:
             lat = 44.0 + obj["x"] * 0.01
             lon = -103.0 + obj["z"] * 0.01