File size: 25,769 Bytes
4a491eb
d67ede3
 
 
 
 
 
 
 
 
 
2099bf7
 
 
 
 
d67ede3
 
239b4d8
 
4a491eb
d67ede3
 
 
 
 
 
e485ea3
d67ede3
 
 
2ecb5bf
e485ea3
2ecb5bf
 
2099bf7
 
 
 
 
 
 
 
d67ede3
2ecb5bf
d67ede3
239b4d8
 
 
 
 
 
 
d67ede3
2ecb5bf
d67ede3
 
 
 
 
2ecb5bf
d67ede3
44237bb
2ecb5bf
 
 
 
d67ede3
 
 
 
2ecb5bf
d67ede3
 
 
 
 
 
 
 
 
 
 
 
2ecb5bf
 
d67ede3
 
 
 
 
 
 
 
 
 
 
a6329f2
d67ede3
 
 
 
 
e485ea3
d67ede3
 
 
 
 
 
 
 
 
 
 
4a491eb
f672794
a6329f2
 
 
 
 
 
 
 
 
 
 
 
 
d67ede3
e485ea3
d67ede3
2ecb5bf
 
d67ede3
 
 
 
2ecb5bf
d67ede3
 
2099bf7
 
e485ea3
d67ede3
4a491eb
d67ede3
 
 
2ecb5bf
d67ede3
 
2ecb5bf
d67ede3
f672794
44237bb
 
 
 
 
 
e485ea3
239b4d8
e485ea3
44237bb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e485ea3
d67ede3
 
2ecb5bf
d67ede3
 
 
44237bb
239b4d8
 
 
 
642d274
239b4d8
 
 
e485ea3
239b4d8
 
 
 
 
e485ea3
 
 
239b4d8
e485ea3
2099bf7
d67ede3
 
 
e485ea3
 
2099bf7
 
 
 
239b4d8
 
 
 
 
e485ea3
06b383c
e485ea3
 
 
 
 
 
2099bf7
e485ea3
 
d67ede3
e485ea3
2099bf7
 
239b4d8
 
d67ede3
 
 
 
2099bf7
 
 
 
 
 
 
239b4d8
7f324ee
239b4d8
 
 
 
e485ea3
 
 
239b4d8
2099bf7
 
d67ede3
 
 
 
2099bf7
d67ede3
 
 
 
 
 
 
e485ea3
 
 
 
 
 
 
 
 
 
 
 
239b4d8
2099bf7
 
 
 
 
e485ea3
2099bf7
 
 
 
e485ea3
 
 
 
2099bf7
e485ea3
2099bf7
 
 
e485ea3
 
2099bf7
 
 
 
 
 
e485ea3
 
 
 
 
2099bf7
e485ea3
2099bf7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e485ea3
2099bf7
 
 
 
 
 
e485ea3
2099bf7
 
 
 
e485ea3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2099bf7
 
 
 
e485ea3
 
2099bf7
 
 
 
 
e485ea3
2099bf7
e485ea3
 
 
2099bf7
 
 
 
 
 
 
e485ea3
 
 
 
 
239b4d8
2099bf7
 
 
 
 
 
 
e485ea3
2099bf7
 
e485ea3
 
 
 
 
 
2099bf7
 
e485ea3
 
 
2099bf7
 
 
 
 
d67ede3
2ecb5bf
d67ede3
239b4d8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e485ea3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2099bf7
 
 
 
 
 
 
 
2ecb5bf
 
 
d67ede3
2ecb5bf
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
import streamlit as st
import asyncio
import websockets
import uuid
from datetime import datetime
import os
import random
import hashlib
import base64
import edge_tts
import nest_asyncio
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
import folium
from streamlit_folium import folium_static

# Patch asyncio for nesting
nest_asyncio.apply()

# Page Config
st.set_page_config(
    layout="wide",
    page_title="Rocky Mountain Quest 3D ๐Ÿ”๏ธ๐ŸŽฎ",
    page_icon="๐ŸฆŒ"
)

# Game Config
GAME_NAME = "Rocky Mountain Quest 3D ๐Ÿ”๏ธ๐ŸŽฎ"
START_LOCATION = "Trailhead Camp โ›บ"
CHARACTERS = {
    "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": "๐ŸŽต"}

# 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)

CHAT_DIR = "chat_logs"
AUDIO_DIR = "audio_logs"
STATE_FILE = "user_state.txt"
CHAT_FILE = os.path.join(CHAT_DIR, "quest_log.md")

# Helpers
def format_timestamp(username=""):
    now = datetime.now().strftime("%Y%m%d_%H%M%S")
    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)
    hash_val = hashlib.md5(prompt.encode()).hexdigest()[:8]
    return f"{timestamp}-{hash_val}.{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

# Audio Processing
async def async_edge_tts_generate(text, voice, username):
    cache_key = f"{text[:100]}_{voice}"
    if cache_key in st.session_state['audio_cache']:
        return st.session_state['audio_cache'][cache_key]
    text = clean_text_for_tts(text)
    filename = f"{AUDIO_DIR}/{format_timestamp(username)}-{hashlib.md5(text.encode()).hexdigest()[:8]}.mp3"
    communicate = edge_tts.Communicate(text, voice)
    await communicate.save(filename)
    if os.path.exists(filename) and os.path.getsize(filename) > 0:
        st.session_state['audio_cache'][cache_key] = filename
        return filename
    return None

def play_and_download_audio(file_path):
    if file_path and os.path.exists(file_path):
        st.audio(file_path)
        st.markdown(get_download_link(file_path), unsafe_allow_html=True)

# WebSocket Broadcast
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 and Quest Log
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
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    entry = f"[{timestamp}] {username}: {message}" if not is_markdown else f"[{timestamp}] {username}:\n```markdown\n{message}\n```"
    md_file = create_file(entry, username, "md")
    with open(CHAT_FILE, 'a') as f:
        f.write(f"{entry}\n")
    audio_file = await async_edge_tts_generate(message, voice, username)
    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
    st.session_state.treasures += 1
    st.session_state.last_chat_update = time.time()
    return md_file, audio_file

async def load_chat():
    if not os.path.exists(CHAT_FILE):
        with open(CHAT_FILE, 'a') as f:
            f.write(f"# {GAME_NAME} Log\n\nThe adventure begins at {START_LOCATION}! ๐Ÿ”๏ธ\n")
    with open(CHAT_FILE, 'r') as f:
        content = f.read().strip()
    return content.split('\n')

# Session State Init
def init_session_state():
    defaults = {
        'server_running': False, 'server_task': None, 'active_connections': {},
        'chat_history': [], 'audio_cache': {}, 'last_transcript': "",
        'username': None, 'score': 0, 'treasures': 0, 'location': START_LOCATION,
        '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
    }
    for k, v in defaults.items():
        if k not in st.session_state:
            st.session_state[k] = v
    if st.session_state.username is None:
        saved_username = load_username()
        if saved_username and saved_username in CHARACTERS:
            st.session_state.username = saved_username
        else:
            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)

init_session_state()

# WebSocket for Multiplayer
async def websocket_handler(websocket, path):
    client_id = str(uuid.uuid4())
    room_id = "quest"
    if room_id not in st.session_state.active_connections:
        st.session_state.active_connections[room_id] = {}
    st.session_state.active_connections[room_id][client_id] = websocket
    username = st.session_state.username
    
    if "prairie" in path:
        st.session_state.prairie_players[client_id] = {
            "username": username,
            "animal": "prairie_dog",
            "location": PRAIRIE_LOCATIONS["Deadwood, SD"],
            "color": CHARACTERS[username]["color"]
        }
        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
        }
        await broadcast_message(f"System|{username} joins the quest!", room_id)
    
    try:
        async for message in websocket:
            if '|' in message:
                sender, content = message.split('|', 1)
                voice = CHARACTERS.get(sender, {"voice": "en-US-AriaNeural"})["voice"]
                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)
                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"{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)
    except websockets.ConnectionClosed:
        await broadcast_message(f"System|{username} leaves the quest!", room_id)
        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]

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"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!"
            prairie_message = f"๐ŸŒพ Prairie Update: {prairie_list}"
            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 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():
    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():
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    loop.run_until_complete(run_websocket_server())

# ArXiv Integration
async def perform_arxiv_search(query, username):
    gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
    refs = gradio_client.predict(
        query, 5, "Semantic Search", "mistralai/Mixtral-8x7B-Instruct-v0.1", api_name="/update_with_rag_md"
    )[0]
    result = f"๐Ÿ“š Ancient Rocky Knowledge:\n{refs}"
    voice = CHARACTERS[username]["voice"]
    md_file, audio_file = await save_chat_entry(username, result, voice, True)
    return md_file, audio_file

# Enhanced 3D Game HTML
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>
    <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; }}
        #ui {{ 
            position: absolute; top: 10px; left: 10px; color: white; 
            background: rgba(0, 0, 0, 0.5); padding: 5px; border-radius: 5px; 
        }}
        #chatBox {{ 
            position: absolute; bottom: 60px; left: 10px; width: 300px; height: 150px; 
            background: rgba(0, 0, 0, 0.7); color: white; padding: 10px; 
            border-radius: 5px; overflow-y: auto; 
        }}
        #controls {{ 
            position: absolute; bottom: 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="ui">
            <div id="players">Players: 1</div>
            <div id="score">Score: 0</div>
            <div id="treasures">Treasures: 0</div>
        </div>
        <div id="chatBox"></div>
        <div id="controls">WASD/Arrows to move, Space to collect treasure</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);
        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.receiveShadow = true;
        scene.add(ground);

        let players = {{}};
        const playerMeshes = {{}};
        let treasures = [];
        let xPos = 0, zPos = 0, moveLeft = false, moveRight = false, moveUp = false, moveDown = false, collect = false;
        let score = 0, treasureCount = 0;

        // Player initialization
        const playerGeometry = new THREE.BoxGeometry(2, 2, 2);
        const playerMaterial = new THREE.MeshPhongMaterial({{ color: {CHARACTERS[st.session_state.username]["color"]} }});
        const playerMesh = new THREE.Mesh(playerGeometry, playerMaterial);
        playerMesh.position.set(xPos, 1, zPos);
        playerMesh.castShadow = true;
        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);
            }}
        }}

        // Controls
        document.addEventListener('keydown', (event) => {{
            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;
            }}
        }});
        document.addEventListener('keyup', (event) => {{
            switch (event.code) {{
                case 'ArrowLeft': case 'KeyA': moveLeft = false; break;
                case 'ArrowRight': case 'KeyD': moveRight = false; break;
                case 'ArrowUp': case 'KeyW': moveUp = false; break;
                case 'ArrowDown': case 'KeyS': moveDown = false; break;
                case 'Space': collect = false; break;
            }}
        }});

        function updatePlayer(delta) {{
            const speed = 20;
            if (moveLeft && xPos > -40) xPos -= speed * delta;
            if (moveRight && xPos < 40) xPos += speed * delta;
            if (moveUp && zPos > -40) zPos -= speed * delta;
            if (moveDown && zPos < 40) zPos += speed * delta;
            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) {{
                        scene.remove(treasures[i]);
                        treasures.splice(i, 1);
                        score += 50;
                        treasureCount += 1;
                        ws.send(`${{playerName}}|SCORE:${{score}}`);
                        ws.send(`${{playerName}}|TREASURE:${{treasureCount}}`);
                        spawnTreasure();
                    }}
                }}
            }}

            camera.position.set(xPos, 50, zPos + 50);
            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]) {{
                    const mesh = new THREE.Mesh(playerGeometry, new THREE.MeshPhongMaterial({{ color: player.color }}));
                    mesh.castShadow = true;
                    scene.add(mesh);
                    playerMeshes[player.username] = mesh;
                }}
                const mesh = playerMeshes[player.username];
                mesh.position.set(player.x, 1, player.z);
                players[player.username] = {{ mesh: mesh, score: player.score, treasures: player.treasures }};
            }});
            document.getElementById('players').textContent = `Players: ${{Object.keys(playerMeshes).length}}`;
            document.getElementById('score').textContent = `Score: ${{score}}`;
            document.getElementById('treasures').textContent = `Treasures: ${{treasureCount}}`;
        }}

        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 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('PRAIRIE_UPDATE:')) {{
                const [sender, message] = data.split('|');
                const chatBox = document.getElementById('chatBox');
                chatBox.innerHTML += `<p>${{sender}}: ${{message}}</p>`;
                chatBox.scrollTop = chatBox.scrollHeight;
            }}
        }};

        let lastTime = performance.now();
        function animate() {{
            requestAnimationFrame(animate);
            const currentTime = performance.now();
            const delta = (currentTime - lastTime) / 1000;
            lastTime = currentTime;

            updatePlayer(delta);
            updateTreasures(delta);
            renderer.render(scene, camera);
        }}

        // Initial spawn
        for (let i = 0; i < 5; i++) spawnTreasure();
        animate();
    </script>
</body>
</html>
"""

# Main Game Loop
def main():
    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()

    left_col, right_col = st.columns([2, 1])

    with left_col:
        components.html(rocky_map_html, width=800, height=600)
        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}")
        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()

    with right_col:
        st.subheader("๐ŸŒพ Prairie Map")
        prairie_map = folium.Map(location=[44.0, -103.0], zoom_start=8, tiles="CartoDB Positron")
        for loc, (lat, lon) in PRAIRIE_LOCATIONS.items():
            folium.Marker([lat, lon], popup=loc).add_to(prairie_map)
        for client_id, player in st.session_state.prairie_players.items():
            folium.CircleMarker(
                location=player['location'],
                radius=5,
                color=f"#{player['color']:06x}",
                fill=True,
                fill_opacity=0.7,
                popup=f"{player['username']} ({player['animal']})"
            ).add_to(prairie_map)
        folium_static(prairie_map, width=600, height=400)
        
        animal = st.selectbox("Choose Animal", ["prairie_dog", "deer", "sheep", "groundhog"])
        location = st.selectbox("Move to", list(PRAIRIE_LOCATIONS.keys()))
        if st.button("Move"):
            asyncio.run(broadcast_message(f"{st.session_state.username}|PRAIRIE:ANIMAL:{animal}", "quest"))
            asyncio.run(broadcast_message(f"{st.session_state.username}|PRAIRIE:MOVE:{location}", "quest"))
            st.rerun()

    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()

    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()

if __name__ == "__main__":
    main()