Update app.py

app.py

@@ -17,6 +17,8 @@ import json
 import streamlit.components.v1 as components
 from gradio_client import Client
 from streamlit_marquee import streamlit_marquee
+import folium
+from streamlit_folium import folium_static
 
 # Patch asyncio for nesting
 nest_asyncio.apply()
@@ -43,6 +45,13 @@ CHARACTERS = {
 }
 FILE_EMOJIS = {"md": "📜", "mp3": "🎵"}
 
+# Prairie Simulator Locations
+PRAIRIE_LOCATIONS = {
+    "Deadwood, SD": (44.3769, -103.7298),
+    "Wind Cave National Park": (43.6047, -103.4798),
+    "Wyoming Spring Creek": (41.6666, -106.6666)
+}
+
 # Directories
 for d in ["chat_logs", "audio_logs"]:
     os.makedirs(d, exist_ok=True)
@@ -152,12 +161,12 @@ def init_session_state():
         'username': None, 'score': 0, 'treasures': 0, 'location': START_LOCATION,
         'speech_processed': False, 'players': {}, 'last_update': time.time(),
         'update_interval': 20, 'x_pos': 0, 'z_pos': 0, 'move_left': False,
-        'move_right': False, 'move_up': False, 'move_down': False
+        'move_right': False, 'move_up': False, 'move_down': False,
+        'prairie_players': {}  # New for prairie simulator
     }
     for k, v in defaults.items():
         if k not in st.session_state:
             st.session_state[k] = v
-    # Ensure username is initialized immediately
     if st.session_state.username is None:
         saved_username = load_username()
         if saved_username and saved_username in CHARACTERS:
@@ -167,7 +176,6 @@ def init_session_state():
             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)
 
-# Call init_session_state immediately to ensure username is set
 init_session_state()
 
 # ArXiv Integration
@@ -189,12 +197,23 @@ async def websocket_handler(websocket, path):
         st.session_state.active_connections[room_id] = {}
     st.session_state.active_connections[room_id][client_id] = websocket
     username = st.session_state.username
-    st.session_state.players[client_id] = {
-        "username": username,
-        "x": random.uniform(-20, 20),
-        "z": random.uniform(-50, 50),
-        "color": CHARACTERS[username]["color"]
-    }
+    
+    # Handle both Rocky Mountain Quest and Prairie Simulator players
+    if "prairie" in path:
+        st.session_state.prairie_players[client_id] = {
+            "username": username,
+            "animal": "prairie_dog",  # Default
+            "location": PRAIRIE_LOCATIONS["Deadwood, SD"],
+            "color": CHARACTERS[username]["color"]
+        }
+    else:
+        st.session_state.players[client_id] = {
+            "username": username,
+            "x": random.uniform(-20, 20),
+            "z": random.uniform(-50, 50),
+            "color": CHARACTERS[username]["color"]
+        }
+    
     await save_chat_entry(username, f"🗺️ Joins the quest at {START_LOCATION}!", CHARACTERS[username]["voice"])
     
     try:
@@ -206,6 +225,15 @@ async def websocket_handler(websocket, path):
                     _, x, z = content.split(":")
                     st.session_state.players[client_id]["x"] = float(x)
                     st.session_state.players[client_id]["z"] = float(z)
+                elif content.startswith("PRAIRIE:"):
+                    action, value = content.split(":", 1)
+                    if action == "ANIMAL":
+                        st.session_state.prairie_players[client_id]["animal"] = value
+                    elif action == "MOVE":
+                        target = PRAIRIE_LOCATIONS.get(value, PRAIRIE_LOCATIONS["Deadwood, SD"])
+                        st.session_state.prairie_players[client_id]["location"] = target
+                        action_msg = f"{username} ({st.session_state.prairie_players[client_id]['animal']}) moves to {value}"
+                        await save_chat_entry(username, action_msg, voice)
                 else:
                     await save_chat_entry(username, content, voice)
                     await perform_arxiv_search(content, username)
@@ -213,6 +241,8 @@ async def websocket_handler(websocket, path):
         await save_chat_entry(username, "🏃 Leaves the quest!", CHARACTERS[username]["voice"])
         if client_id in st.session_state.players:
             del st.session_state.players[client_id]
+        if client_id in st.session_state.prairie_players:
+            del st.session_state.prairie_players[client_id]
     finally:
         if room_id in st.session_state.active_connections and client_id in st.session_state.active_connections[room_id]:
             del st.session_state.active_connections[room_id][client_id]
@@ -220,12 +250,21 @@ async def websocket_handler(websocket, path):
 async def periodic_update():
     while True:
         if st.session_state.active_connections.get("quest"):
+            # Rocky Mountain Quest update
             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()))
             await broadcast_message(f"System|{message}", "quest")
             await broadcast_message(f"MAP_UPDATE:{player_data}", "quest")
-            await save_chat_entry("System", message, "en-US-AriaNeural")
+            
+            # Prairie Simulator update
+            prairie_list = ", ".join([f"{p['username']} ({p['animal']})" for p in st.session_state.prairie_players.values()]) or "No animals yet!"
+            prairie_message = f"🌾 Prairie Update: {prairie_list}"
+            prairie_data = json.dumps(list(st.session_state.prairie_players.values()))
+            await broadcast_message(f"System|{prairie_message}", "quest")
+            await broadcast_message(f"PRAIRIE_UPDATE:{prairie_data}", "quest")
+            
+            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():
@@ -240,8 +279,8 @@ def start_websocket_server():
     asyncio.set_event_loop(loop)
     loop.run_until_complete(run_websocket_server())
 
-# Game HTML with Map (Corrected JavaScript)
-html_code = f"""
+# Rocky Mountain Quest Map HTML
+rocky_map_html = f"""
 <!DOCTYPE html>
 <html lang="en">
 <head>
@@ -296,7 +335,7 @@ html_code = f"""
 
         let players = {{}};
         const playerMeshes = {{}};
-        let xPos = 0;  // Local JavaScript variables for position
+        let xPos = 0;
         let zPos = 0;
 
         function updatePlayers(playerData) {{
@@ -341,7 +380,7 @@ html_code = f"""
             if (data.startsWith('MAP_UPDATE:')) {{
                 const playerData = JSON.parse(data.split('MAP_UPDATE:')[1]);
                 updatePlayers(playerData);
-            }} else {{
+            }} else if (!data.startsWith('PRAIRIE_UPDATE:')) {{
                 const [sender, message] = data.split('|');
                 const chatBox = document.getElementById('chatBox');
                 chatBox.innerHTML += `<p>${{sender}}: ${{message}}</p>`;
@@ -359,10 +398,165 @@ html_code = f"""
 </html>
 """
 
+# Prairie Simulator HTML
+prairie_simulator_html = f"""
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Prairie Simulator</title>
+    <style>
+        body {{ margin: 0; font-family: Arial, sans-serif; background: #f0f0f0; }}
+        #simContainer {{ width: 100%; height: 600px; position: relative; }}
+        #map {{ width: 100%; height: 400px; }}
+        #controls {{ padding: 10px; background: #fff; border-radius: 5px; }}
+        #status {{ color: #333; padding: 5px; }}
+        #chatBox {{ height: 100px; overflow-y: auto; background: #fff; border: 1px solid #ccc; padding: 5px; }}
+    </style>
+</head>
+<body>
+    <div id="simContainer">
+        <div id="map"></div>
+        <div id="controls">
+            <label>Animal: </label>
+            <select id="animal" onchange="updateAnimal()">
+                <option value="prairie_dog">Prairie Dog</option>
+                <option value="deer">Deer</option>
+                <option value="sheep">Sheep</option>
+                <option value="groundhog">Groundhog</option>
+            </select>
+            <label>Move to: </label>
+            <select id="location">
+                <option value="Deadwood, SD">Deadwood, SD</option>
+                <option value="Wind Cave National Park">Wind Cave National Park</option>
+                <option value="Wyoming Spring Creek">Wyoming Spring Creek</option>
+            </select>
+            <button onclick="move()">Move</button>
+            <div id="status">Players: 0</div>
+            <div id="chatBox"></div>
+        </div>
+    </div>
+
+    <script src="https://unpkg.com/[email protected]/dist/leaflet.js"></script>
+    <link rel="stylesheet" href="https://unpkg.com/[email protected]/dist/leaflet.css" />
+    <script>
+        const playerName = "{st.session_state.username}";
+        let ws = new WebSocket('ws://localhost:8765/prairie');
+        const map = L.map('map').setView([44.0, -103.0], 7);
+        L.tileLayer('https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png', {{
+            attribution: '&copy; OpenStreetMap contributors'
+        }}).addTo(map);
+
+        const locations = {{
+            "Deadwood, SD": [44.3769, -103.7298],
+            "Wind Cave National Park": [43.6047, -103.4798],
+            "Wyoming Spring Creek": [41.6666, -106.6666]
+        }};
+        for (let loc in locations) {{
+            L.marker(locations[loc]).addTo(map).bindPopup(loc);
+        }}
+
+        let players = {{}};
+        const markers = {{}};
+
+        function updatePlayers(playerData) {{
+            playerData.forEach(player => {{
+                if (!markers[player.username]) {{
+                    markers[player.username] = L.circleMarker(player.location, {{
+                        color: `#${{player.color.toString(16).padStart(6, '0')}}`,
+                        radius: 5
+                    }}).addTo(map).bindPopup(`${{player.username}} (${{player.animal}})`);
+                }} else {{
+                    markers[player.username].setLatLng(player.location);
+                }}
+            }});
+            document.getElementById('status').textContent = `Players: ${{Object.keys(markers).length}}`;
+        }}
+
+        function updateAnimal() {{
+            const animal = document.getElementById('animal').value;
+            ws.send(`${{playerName}}|PRAIRIE:ANIMAL:${{animal}}`);
+        }}
+
+        function move() {{
+            const location = document.getElementById('location').value;
+            ws.send(`${{playerName}}|PRAIRIE:MOVE:${{location}}`);
+        }}
+
+        ws.onmessage = function(event) {{
+            const data = event.data;
+            if (data.startsWith('PRAIRIE_UPDATE:')) {{
+                const playerData = JSON.parse(data.split('PRAIRIE_UPDATE:')[1]);
+                updatePlayers(playerData);
+            }} else if (!data.startsWith('MAP_UPDATE:')) {{
+                const [sender, message] = data.split('|');
+                const chatBox = document.getElementById('chatBox');
+                chatBox.innerHTML += `<p>${{sender}}: ${{message}}</p>`;
+                chatBox.scrollTop = chatBox.scrollHeight;
+            }}
+        }};
+    </script>
+</body>
+</html>
+"""
+
 # Main Game Loop
 def main():
-    st.title(f"🎮 {GAME_NAME}")
-    st.subheader(f"🌄 {st.session_state.username}’s Adventure - Score: {st.session_state.score} 🏆")
+    # Sidebar Titles and HUD
+    st.sidebar.title(f"🎮 {GAME_NAME}")
+    st.sidebar.subheader(f"🌄 {st.session_state.username}’s Adventure - Score: {st.session_state.score} 🏆")
+    st.sidebar.write(f"📜 {CHARACTERS[st.session_state.username]['desc']}")
+    st.sidebar.write(f"📍 Location: {st.session_state.location}")
+    st.sidebar.write(f"🏅 Score: {st.session_state.score}")
+    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'}")
+    
+    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"]))
+        st.session_state.username = new_username
+        save_username(st.session_state.username)
+        st.rerun()
+
+    # Two-column layout
+    left_col, right_col = st.columns([2, 1])
+
+    # Left Column: Rocky Mountain Quest
+    with left_col:
+        # Render Rocky Mountain Map
+        components.html(rocky_map_html, width=800, height=600)
+
+        # Chat Interface
+        chat_content = asyncio.run(load_chat())
+        st.text_area("📜 Quest Log", "\n".join(chat_content[-10:]), height=200, disabled=True)
+
+        message = st.text_input(f"🗨️ {st.session_state.username} says:", placeholder="Speak or type to chat! 🌲")
+        if st.button("🌟 Send & Chat 🎤"):
+            if message:
+                voice = CHARACTERS[st.session_state.username]["voice"]
+                md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, message, voice))
+                if audio_file:
+                    play_and_download_audio(audio_file)
+                st.success(f"🌄 +10 points! New Score: {st.session_state.score}")
+
+        # Voice Input Component
+        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:
+            val_stripped = val.strip().replace('\n', ' ')
+            if val_stripped:
+                voice = CHARACTERS.get(st.session_state.username, {"voice": "en-US-AriaNeural"})["voice"]
+                st.session_state['speech_processed'] = True
+                md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, val_stripped, voice))
+                if audio_file:
+                    play_and_download_audio(audio_file)
+                st.rerun()
+
+    # Right Column: Prairie Simulator
+    with right_col:
+        st.subheader("🌾 Prairie Simulator")
+        components.html(prairie_simulator_html, width=600, height=600)
 
     # Countdown Timer
     elapsed = time.time() - st.session_state.last_update
@@ -372,49 +566,6 @@ def main():
         st.session_state.last_update = time.time()
         st.rerun()
 
-    # Voice Input Component
-    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:
-        val_stripped = val.strip().replace('\n', ' ')
-        if val_stripped:
-            voice = CHARACTERS.get(st.session_state.username, {"voice": "en-US-AriaNeural"})["voice"]
-            st.session_state['speech_processed'] = True
-            md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, val_stripped, voice))
-            if audio_file:
-                play_and_download_audio(audio_file)
-            st.rerun()
-
-    # Render Map
-    components.html(html_code, width=800, height=600)
-
-    # Chat Interface
-    chat_content = asyncio.run(load_chat())
-    st.text_area("📜 Quest Log", "\n".join(chat_content[-10:]), height=200, disabled=True)
-
-    message = st.text_input(f"🗨️ {st.session_state.username} says:", placeholder="Speak or type to chat! 🌲")
-    if st.button("🌟 Send & Chat 🎤"):
-        if message:
-            voice = CHARACTERS[st.session_state.username]["voice"]
-            md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, message, voice))
-            if audio_file:
-                play_and_download_audio(audio_file)
-            st.success(f"🌄 +10 points! New Score: {st.session_state.score}")
-
-    # Sidebar: Game HUD
-    st.sidebar.subheader("🎮 Adventurer’s HUD")
-    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"]))
-        st.session_state.username = new_username
-        save_username(st.session_state.username)
-        st.rerun()
-    st.sidebar.write(f"📜 {CHARACTERS[st.session_state.username]['desc']}")
-    st.sidebar.write(f"📍 Location: {st.session_state.location}")
-    st.sidebar.write(f"🏅 Score: {st.session_state.score}")
-    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'}")
-
     if not st.session_state.get('server_running', False):
         st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)
         st.session_state.server_task.start()