ColoradoMadLibs / backup1.app.py
awacke1's picture
Update backup1.app.py
e485ea3 verified
raw
history blame
25.8 kB
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()