Update app.py

app.py

@@ -10,8 +10,11 @@ import glob
 import base64
 import edge_tts
 import nest_asyncio
-import re  # For regular expressions in clean_text_for_tts
-import threading  # For running WebSocket server in a separate thread
+import re
+import threading
+import time
+import json
+import streamlit.components.v1 as components
 from gradio_client import Client
 from streamlit_marquee import streamlit_marquee
 
@@ -29,14 +32,14 @@ st.set_page_config(
 GAME_NAME = "Rocky Mountain Quest 🏔️🎮"
 START_LOCATION = "Trailhead Camp ⛺"
 CHARACTERS = {
-    "Trailblazer Tim 🌄": {"voice": "en-US-GuyNeural", "desc": "Fearless hiker seeking epic trails!"},
-    "Meme Queen Mia 😂": {"voice": "en-US-JennyNeural", "desc": "Spreads laughs with wild memes!"},
-    "Elk Whisperer Eve 🦌": {"voice": "en-GB-SoniaNeural", "desc": "Talks to wildlife, loves nature!"},
-    "Tech Titan Tara 💾": {"voice": "en-AU-NatashaNeural", "desc": "Codes her way through the Rockies!"},
-    "Ski Guru Sam ⛷️": {"voice": "en-CA-ClaraNeural", "desc": "Shreds slopes, lives for snow!"},
-    "Cosmic Camper Cal 🌠": {"voice": "en-US-AriaNeural", "desc": "Stargazes and tells epic tales!"},
-    "Rasta Ranger Rick 🍃": {"voice": "en-GB-RyanNeural", "desc": "Chills with nature’s vibes!"},
-    "Boulder Bro Ben 🪨": {"voice": "en-AU-WilliamNeural", "desc": "Climbs rocks, bro-style!"}
+    "Trailblazer Tim 🌄": {"voice": "en-US-GuyNeural", "desc": "Fearless hiker seeking epic trails!", "color": 0x00ff00},
+    "Meme Queen Mia 😂": {"voice": "en-US-JennyNeural", "desc": "Spreads laughs with wild memes!", "color": 0xff00ff},
+    "Elk Whisperer Eve 🦌": {"voice": "en-GB-SoniaNeural", "desc": "Talks to wildlife, loves nature!", "color": 0x0000ff},
+    "Tech Titan Tara 💾": {"voice": "en-AU-NatashaNeural", "desc": "Codes her way through the Rockies!", "color": 0xffff00},
+    "Ski Guru Sam ⛷️": {"voice": "en-CA-ClaraNeural", "desc": "Shreds slopes, lives for snow!", "color": 0xffa500},
+    "Cosmic Camper Cal 🌠": {"voice": "en-US-AriaNeural", "desc": "Stargazes and tells epic tales!", "color": 0x800080},
+    "Rasta Ranger Rick 🍃": {"voice": "en-GB-RyanNeural", "desc": "Chills with nature’s vibes!", "color": 0x00ffff},
+    "Boulder Bro Ben 🪨": {"voice": "en-AU-WilliamNeural", "desc": "Climbs rocks, bro-style!", "color": 0xff4500}
 }
 FILE_EMOJIS = {"md": "📜", "mp3": "🎵"}
 
@@ -55,10 +58,9 @@ def init_session_state():
         'server_running': False, 'server_task': None, 'active_connections': {},
         'chat_history': [], 'audio_cache': {}, 'last_transcript': "",
         'username': None, 'score': 0, 'treasures': 0, 'location': START_LOCATION,
-        'marquee_settings': {
-            "background": "#2E8B57", "color": "#FFFFFF", "font-size": "16px",
-            "animationDuration": "15s", "width": "100%", "lineHeight": "40px"
-        }
+        '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
     }
     for k, v in defaults.items():
         if k not in st.session_state:
@@ -131,8 +133,8 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
     await broadcast_message(f"{username}|{message}", "quest")
     st.session_state.chat_history.append(entry)
     st.session_state.last_transcript = message
-    st.session_state.score += 10  # Points for participation
-    st.session_state.treasures += 1  # Audio treasure collected
+    st.session_state.score += 10
+    st.session_state.treasures += 1
     return md_file, audio_file
 
 async def load_chat():
@@ -154,7 +156,7 @@ async def perform_arxiv_search(query, username):
     md_file, audio_file = await save_chat_entry(username, result, voice, True)
     return md_file, audio_file
 
-# WebSocket for Multiplayer
+# WebSocket for Multiplayer with Map Updates
 async def websocket_handler(websocket, path):
     client_id = str(uuid.uuid4())
     room_id = "quest"
@@ -162,16 +164,30 @@ 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.get('username', random.choice(list(CHARACTERS.keys())))
+    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:
         async for message in websocket:
             if '|' in message:
                 username, content = message.split('|', 1)
                 voice = CHARACTERS.get(username, {"voice": "en-US-AriaNeural"})["voice"]
-                await save_chat_entry(username, content, voice)
-                await perform_arxiv_search(content, username)  # ArXiv response for every chat
+                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)
+                else:
+                    await save_chat_entry(username, content, voice)
+                    await perform_arxiv_search(content, username)
     except websockets.ConnectionClosed:
         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]
     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]
@@ -188,10 +204,22 @@ 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]
 
+async def periodic_update():
+    while True:
+        if st.session_state.active_connections.get("quest"):
+            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")
+        await asyncio.sleep(st.session_state.update_interval)
+
 async def run_websocket_server():
     if not st.session_state.get('server_running', False):
         server = await websockets.serve(websocket_handler, '0.0.0.0', 8765)
         st.session_state['server_running'] = True
+        asyncio.create_task(periodic_update())
         await server.wait_closed()
 
 def start_websocket_server():
@@ -199,19 +227,135 @@ def start_websocket_server():
     asyncio.set_event_loop(loop)
     loop.run_until_complete(run_websocket_server())
 
-# Game Quest (Mad Libs)
-def generate_quest_story(username, inputs):
-    locations = ["Peak Summit 🏞️", "Elk Valley 🌲", "Meme Cave 🕳️", "Tech Outpost 💻"]
-    st.session_state.location = random.choice(locations)
-    story = f"""
-    🌄 **{username}’s Quest at {st.session_state.location}:**
-    🗺️ {username} discovers {inputs['quantity']} {inputs['plural_noun']}!
-    🌟 They’re {inputs['adjective']} and {inputs['action']} everywhere.
-    🎯 With a {inputs['tool']}, {username} faces a {inputs['creature']} 
-    ⚡ that {inputs['event']}—shouting “{inputs['shout']}!”
-    🎉 Victory earns {username} {st.session_state.treasures} audio treasures!
-    """
-    return story.strip()
+# Game HTML with Map
+html_code = 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 Map</title>
+    <style>
+        body {{ margin: 0; overflow: hidden; font-family: Arial, sans-serif; background: #000; }}
+        #gameContainer {{ width: 800px; height: 600px; position: relative; }}
+        canvas {{ width: 100%; height: 100%; display: block; }}
+        #chatBox {{ 
+            position: absolute; bottom: 10px; left: 10px; width: 300px; height: 150px; 
+            background: rgba(0, 0, 0, 0.7); color: white; padding: 10px; 
+            border-radius: 5px; overflow-y: auto; 
+        }}
+        #status {{ 
+            position: absolute; top: 10px; left: 10px; color: white; 
+            background: rgba(0, 0, 0, 0.5); padding: 5px; border-radius: 5px; 
+        }}
+    </style>
+</head>
+<body>
+    <div id="gameContainer">
+        <div id="status">Players: 1</div>
+        <div id="chatBox"></div>
+    </div>
+
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/three.js/r128/three.min.js"></script>
+    <script>
+        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);
+        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);
+
+        const ambientLight = new THREE.AmbientLight(0xffffff, 0.5);
+        scene.add(ambientLight);
+        const sunLight = new THREE.DirectionalLight(0xffddaa, 1);
+        sunLight.position.set(50, 50, 50);
+        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;
+        scene.add(ground);
+
+        let players = {{}};
+        const playerMeshes = {{}};
+
+        function updatePlayers(playerData) {{
+            playerData.forEach(player => {{
+                if (!playerMeshes[player.username]) {{
+                    const geometry = new THREE.BoxGeometry(2, 2, 2);
+                    const material = new THREE.MeshPhongMaterial({{ color: player.color }});
+                    const mesh = new THREE.Mesh(geometry, material);
+                    scene.add(mesh);
+                    playerMeshes[player.username] = mesh;
+                }}
+                const mesh = playerMeshes[player.username];
+                mesh.position.set(player.x, 1, player.z);
+            }});
+            document.getElementById('status').textContent = `Players: ${{Object.keys(playerMeshes).length}}`;
+        }}
+
+        document.addEventListener('keydown', (event) => {{
+            const speed = 2;
+            switch (event.code) {{
+                case 'ArrowLeft': case 'KeyA': 
+                    st.session_state.move_left = true; 
+                    st.session_state.x_pos -= speed; 
+                    ws.send(`{playerName}|MOVE:${{st.session_state.x_pos}}:${{st.session_state.z_pos}}`);
+                    break;
+                case 'ArrowRight': case 'KeyD': 
+                    st.session_state.move_right = true; 
+                    st.session_state.x_pos += speed; 
+                    ws.send(`{playerName}|MOVE:${{st.session_state.x_pos}}:${{st.session_state.z_pos}}`);
+                    break;
+                case 'ArrowUp': case 'KeyW': 
+                    st.session_state.move_up = true; 
+                    st.session_state.z_pos -= speed; 
+                    ws.send(`{playerName}|MOVE:${{st.session_state.x_pos}}:${{st.session_state.z_pos}}`);
+                    break;
+                case 'ArrowDown': case 'KeyS': 
+                    st.session_state.move_down = true; 
+                    st.session_state.z_pos += speed; 
+                    ws.send(`{playerName}|MOVE:${{st.session_state.x_pos}}:${{st.session_state.z_pos}}`);
+                    break;
+            }}
+        }});
+
+        document.addEventListener('keyup', (event) => {{
+            switch (event.code) {{
+                case 'ArrowLeft': case 'KeyA': st.session_state.move_left = false; break;
+                case 'ArrowRight': case 'KeyD': st.session_state.move_right = false; break;
+                case 'ArrowUp': case 'KeyW': st.session_state.move_up = false; break;
+                case 'ArrowDown': case 'KeyS': st.session_state.move_down = false; break;
+            }}
+        }});
+
+        ws.onmessage = function(event) {{
+            const data = event.data;
+            if (data.startsWith('MAP_UPDATE:')) {{
+                const playerData = JSON.parse(data.split('MAP_UPDATE:')[1]);
+                updatePlayers(playerData);
+            }} else {{
+                const [sender, message] = data.split('|');
+                const chatBox = document.getElementById('chatBox');
+                chatBox.innerHTML += `<p>${{sender}}: ${{message}}</p>`;
+                chatBox.scrollTop = chatBox.scrollHeight;
+            }}
+        }};
+
+        function animate() {{
+            requestAnimationFrame(animate);
+            renderer.render(scene, camera);
+        }}
+        animate();
+    </script>
+</body>
+</html>
+"""
 
 # Main Game Loop
 def main():
@@ -226,63 +370,44 @@ def main():
 
     st.title(f"🎮 {GAME_NAME}")
     st.subheader(f"🌄 {st.session_state.username}’s Adventure - Score: {st.session_state.score} 🏆")
-    chat_text = " ".join([line.split(": ")[-1] for line in asyncio.run(load_chat()) if ": " in line][-5:])
-    streamlit_marquee(content=f"🏔️ {st.session_state.location} | 🎙️ {st.session_state.username} | 💬 {chat_text}", 
-                      **st.session_state['marquee_settings'], key="quest_marquee")
-
-    tab_main = st.radio("🎲 Quest Actions:", ["🗣️ Explore & Chat", "🎵 Treasure Vault", "🗺️ Quest Challenge"], horizontal=True)
-
-    if tab_main == "🗣️ Explore & Chat":
-        st.subheader(f"🗣️ Explore {st.session_state.location} 🎙️")
-        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="Explore the wilds! 🌲")
-        if st.button("🌟 Send & Explore 🎤"):
-            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}")
-
-    elif tab_main == "🎵 Treasure Vault":
-        st.subheader("🎵 Audio Treasure Vault 🏆")
-        mp3_files = sorted(glob.glob("*.mp3"), key=os.path.getmtime, reverse=True)
-        if mp3_files:
-            st.write(f"🏅 Treasures Collected: {st.session_state.treasures}")
-            for i, mp3 in enumerate(mp3_files[:10]):
-                with st.expander(f"🎵 Treasure #{i+1}: {os.path.basename(mp3)}"):
-                    play_and_download_audio(mp3)
-        else:
-            st.write("🔍 No treasures yet—explore or complete quests to collect audio loot! 🎤")
-
-    elif tab_main == "🗺️ Quest Challenge":
-        st.subheader("🗺️ Rocky Mountain Quest Challenge 🌟")
-        st.write("Fill in the blanks to embark on a wild adventure!")
-        inputs = {}
-        col1, col2 = st.columns(2)
-        with col1:
-            inputs['quantity'] = st.text_input("🔢 How Many?", "3", key="quantity")
-            inputs['plural_noun'] = st.text_input("🌄 Things?", "elk", key="plural_noun")
-            inputs['adjective'] = st.text_input("✨ Describe Them?", "wild", key="adjective")
-            inputs['action'] = st.text_input("🏃 What They Do?", "running", key="action")
-        with col2:
-            inputs['tool'] = st.text_input("🛠️ Your Tool?", "map", key="tool")
-            inputs['creature'] = st.text_input("🦇 Encounter?", "bear", key="creature")
-            inputs['event'] = st.text_input("⚡ What Happens?", "roars", key="event")
-            inputs['shout'] = st.text_input("🗣️ Your Cry?", "Yeehaw!", key="shout")
-
-        if st.button("🎉 Start Quest! 🎤"):
-            story = generate_quest_story(st.session_state.username, inputs)
-            st.markdown(f"### 🌄 {st.session_state.username}’s Quest Log")
-            st.write(story)
+
+    # Countdown Timer
+    elapsed = time.time() - st.session_state.last_update
+    remaining = max(0, st.session_state.update_interval - elapsed)
+    st.sidebar.markdown(f"⏳ Next Update in: {int(remaining)}s")
+    if remaining <= 0:
+        st.session_state.last_update = time.time()
+        st.rerun()
+
+    # Voice Input Component
+    mycomponent = components.html(open("speech.html").read(), height=400)  # Assuming speech.html is in the same directory
+    val = st.session_state.get('speech_input', '')
+    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.session_state.speech_input = ''
+            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, story, voice, True))
+            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.session_state.score += 50  # Bonus for quest completion
-            st.success(f"🏆 Quest Complete! +50 points! New Score: {st.session_state.score}")
+            st.success(f"🌄 +10 points! New Score: {st.session_state.score}")
 
     # Sidebar: Game HUD
     st.sidebar.subheader("🎮 Adventurer’s HUD")
@@ -296,6 +421,7 @@ def main():
     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)