import streamlit as st import asyncio import websockets import uuid import argparse from datetime import datetime import os import random import time import hashlib from PIL import Image import glob from urllib.parse import quote import base64 import io import streamlit.components.v1 as components import edge_tts from audio_recorder_streamlit import audio_recorder import nest_asyncio # Apply patch to allow nested asyncio.run() calls nest_asyncio.apply() # Initial App Configuration (static) icons = '๐ค๐ง ๐ฌ๐' START_ROOM = "Sector ๐" # Set page config once at the top st.set_page_config( page_title="๐ค๐ง MMO Chat Brain๐๐ฌ", # Initial static title page_icon=icons, layout="wide", initial_sidebar_state="auto" ) # Fun usernames with emojis and their paired voices FUN_USERNAMES = { "CosmicJester ๐": "en-US-AriaNeural", "PixelPanda ๐ผ": "en-US-JennyNeural", "QuantumQuack ๐ฆ": "en-GB-SoniaNeural", "StellarSquirrel ๐ฟ๏ธ": "en-AU-NatashaNeural", "GizmoGuru โ๏ธ": "en-CA-ClaraNeural", "NebulaNinja ๐ ": "en-US-GuyNeural", "ByteBuster ๐พ": "en-GB-RyanNeural", "GalacticGopher ๐": "en-AU-WilliamNeural", "RocketRaccoon ๐": "en-CA-LiamNeural", "EchoElf ๐ง": "en-US-AriaNeural", "PhantomFox ๐ฆ": "en-US-JennyNeural", "WittyWizard ๐ง": "en-GB-SoniaNeural", "LunarLlama ๐": "en-AU-NatashaNeural", "SolarSloth โ๏ธ": "en-CA-ClaraNeural", "AstroAlpaca ๐ฆ": "en-US-GuyNeural", "CyberCoyote ๐บ": "en-GB-RyanNeural", "MysticMoose ๐ฆ": "en-AU-WilliamNeural", "GlitchGnome ๐ง": "en-CA-LiamNeural", "VortexViper ๐": "en-US-AriaNeural", "ChronoChimp ๐": "en-US-JennyNeural" } # Directories and files CHAT_DIR = "chat_logs" VOTE_DIR = "vote_logs" STATE_FILE = "user_state.txt" AUDIO_DIR = "audio_logs" HISTORY_DIR = "history_logs" os.makedirs(CHAT_DIR, exist_ok=True) os.makedirs(VOTE_DIR, exist_ok=True) os.makedirs(AUDIO_DIR, exist_ok=True) os.makedirs(HISTORY_DIR, exist_ok=True) CHAT_FILE = os.path.join(CHAT_DIR, "global_chat.md") QUOTE_VOTES_FILE = os.path.join(VOTE_DIR, "quote_votes.md") MEDIA_VOTES_FILE = os.path.join(VOTE_DIR, "media_votes.md") HISTORY_FILE = os.path.join(HISTORY_DIR, "chat_history.md") # Unicode digits and fonts UNICODE_DIGITS = {i: f"{i}\uFE0Fโฃ" for i in range(10)} UNICODE_FONTS = [ ("Normal", lambda x: x), ("Bold", lambda x: "".join(chr(ord(c) + 0x1D400 - 0x41) if 'A' <= c <= 'Z' else chr(ord(c) + 0x1D41A - 0x61) if 'a' <= c <= 'z' else c for c in x)), ] server_running = False server_task = None # Helper Functions def format_timestamp_prefix(): """๐ - Time Stamp Champ - Marks the clock, no flop!""" return datetime.now().strftime("%Y%m%d_%H%M%S") def get_node_name(): """๐ - Naming Node with Code - Spins a name, oh so bold!""" action = "๐ - Naming Node with Code - Spins a name, oh so bold!" username = st.session_state.get('username', 'System ๐') log_action(username, action) parser = argparse.ArgumentParser(description='Start a chat node with a specific name') parser.add_argument('--node-name', type=str, default=None) parser.add_argument('--port', type=int, default=8501) args = parser.parse_args() return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port def log_action(username, action): """๐ - Log Jog Blog - Tracks the deed, no greed!""" if 'action_log' not in st.session_state: st.session_state.action_log = {} user_log = st.session_state.action_log.setdefault(username, {}) current_time = time.time() user_log = {k: v for k, v in user_log.items() if current_time - v < 10} st.session_state.action_log[username] = user_log if action not in user_log: with open(HISTORY_FILE, 'a') as f: f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: {action}\n") user_log[action] = current_time async def save_chat_entry(username, message): """๐ - Chat Snap Trap - Logs your yap, no cap! โ๏ธ๐ง """ action = "๐ - Chat Snap Trap - Logs your yap, no cap! โ๏ธ๐ง " await asyncio.to_thread(log_action, username, action) timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") entry = f"[{timestamp}] {username}: {message}" await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').write(f"{entry}\n")) # Generate and save audio for the message voice = FUN_USERNAMES.get(username, "en-US-AriaNeural") audio_file = await async_edge_tts_generate(message, voice) if audio_file: with open(HISTORY_FILE, 'a') as f: f.write(f"[{timestamp}] {username}: Audio generated - {audio_file}\n") async def load_chat(): """๐ - Chat Fetch Quest - Grabs the log, no jest!""" action = "๐ - Chat Fetch Quest - Grabs the log, no jest!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if not os.path.exists(CHAT_FILE): await asyncio.to_thread(lambda: open(CHAT_FILE, 'w').write(f"# {START_ROOM} Chat\n\nWelcome to the cosmic hub - start chatting! ๐ค\n")) with open(CHAT_FILE, 'r') as f: content = await asyncio.to_thread(f.read) return content async def get_user_list(chat_content): """๐ฅ - Crew Clue Brew - Spots whoโs who in the crew!""" action = "๐ฅ - Crew Clue Brew - Spots whoโs who in the crew!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) users = set() for line in chat_content.split('\n'): if line.strip() and ': ' in line: user = line.split(': ')[1].split(' ')[0] users.add(user) return sorted(list(users)) async def has_joined_before(client_id, chat_content): """๐ช - Join Check Trek - Sees whoโs back, no wreck!""" action = "๐ช - Join Check Trek - Sees whoโs back, no wreck!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) return any(f"Client-{client_id} has joined" in line for line in chat_content.split('\n')) async def get_message_suggestions(chat_content, prefix): """๐ - Suggest Jest Chest - Finds old quips, the best!""" action = "๐ - Suggest Jest Chest - Finds old quips, the best!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) lines = chat_content.split('\n') messages = [line.split(': ', 1)[1] for line in lines if ': ' in line and line.strip()] return [msg for msg in messages if msg.lower().startswith(prefix.lower())][:5] async def save_vote(file, item, user_hash, username, comment=""): """๐ - Vote Note Float - Cheers rise, we gloat!""" action = "๐ - Vote Note Float - Cheers rise, we gloat!" await asyncio.to_thread(log_action, username, action) timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") entry = f"[{timestamp}] {user_hash} voted for {item}" await asyncio.to_thread(lambda: open(file, 'a').write(f"{entry}\n")) await asyncio.to_thread(lambda: open(HISTORY_FILE, "a").write(f"- {timestamp} - User {user_hash} voted for {item}\n")) chat_message = f"{username} upvoted: \"{item}\"" if comment: chat_message += f" - {comment}" await save_chat_entry(username, chat_message) async def load_votes(file): """๐ - Tally Rally Call - Counts the cheers, no stall!""" action = "๐ - Tally Rally Call - Counts the cheers, no stall!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if not os.path.exists(file): await asyncio.to_thread(lambda: open(file, 'w').write("# Vote Tally\n\nNo votes yet - get clicking! ๐ฑ๏ธ\n")) with open(file, 'r') as f: lines = await asyncio.to_thread(f.read.strip().split, '\n')[2:] votes = {} user_votes = set() for line in lines: if line.strip() and 'voted for' in line: user_hash = line.split('] ')[1].split(' voted for ')[0] item = line.split('voted for ')[1] vote_key = f"{user_hash}-{item}" if vote_key not in user_votes: votes[item] = votes.get(item, 0) + 1 user_votes.add(vote_key) return votes async def generate_user_hash(): """๐ - Hash Dash Bash - Crafts a code, so brash!""" action = "๐ - Hash Dash Bash - Crafts a code, so brash!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if 'user_hash' not in st.session_state: st.session_state.user_hash = hashlib.md5(str(random.getrandbits(128)).encode()).hexdigest()[:8] return st.session_state.user_hash async def async_edge_tts_generate(text, voice, rate=0, pitch=0, file_format="mp3"): """๐ถ - Tune Moon Boom - Drops a beat, so groom!""" action = "๐ถ - Tune Moon Boom - Drops a beat, so groom!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) timestamp = format_timestamp_prefix() filename = os.path.join(AUDIO_DIR, f"audio_{timestamp}_{random.randint(1000, 9999)}.mp3") communicate = edge_tts.Communicate(text, voice, rate=f"{rate:+d}%", pitch=f"{pitch:+d}Hz") await communicate.save(filename) return filename if os.path.exists(filename) else None def play_and_download_audio(file_path): """๐ - Sound Pound Ground - Plays it loud, all around!""" if file_path and os.path.exists(file_path): st.audio(file_path) with open(file_path, "rb") as f: b64 = base64.b64encode(f.read()).decode() dl_link = f'๐ต Download {os.path.basename(file_path)}' st.markdown(dl_link, unsafe_allow_html=True) async def save_pasted_image(image_data): """๐ธ - Snap Cap Trap - Saves your pic, no flap!""" action = "๐ธ - Snap Cap Trap - Saves your pic, no flap!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) timestamp = format_timestamp_prefix() filename = f"paste_{timestamp}.png" filepath = os.path.join('./', filename) if ',' in image_data: image_data = image_data.split(',')[1] img_bytes = base64.b64decode(image_data) img = Image.open(io.BytesIO(img_bytes)) await asyncio.to_thread(img.save, filepath, "PNG") return filename async def get_video_html(video_path, width="100%"): """๐ฅ - Reel Deal Steal - Plays your flick, so real!""" action = "๐ฅ - Reel Deal Steal - Plays your flick, so real!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) video_url = f"data:video/mp4;base64,{base64.b64encode(await asyncio.to_thread(open, video_path, 'rb').read()).decode()}" return f'' async def get_audio_html(audio_path, width="100%"): """๐ถ - Tune Moon Boom - Drops a beat, so groom!""" action = "๐ถ - Tune Moon Boom - Drops a beat, so groom!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) audio_url = f"data:audio/mpeg;base64,{base64.b64encode(await asyncio.to_thread(open, audio_path, 'rb').read()).decode()}" return f'' active_connections = {} async def websocket_handler(websocket, path): """๐ - Web Sock Jock - Links the chat, no block!""" action = "๐ - Web Sock Jock - Links the chat, no block!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) try: client_id = str(uuid.uuid4()) room_id = "chat" active_connections.setdefault(room_id, {})[client_id] = websocket chat_content = await load_chat() username = st.session_state.get('username', random.choice(list(FUN_USERNAMES.keys()))) if not await has_joined_before(client_id, chat_content): await save_chat_entry(f"Client-{client_id}", f"{username} has joined {START_ROOM}!") async for message in websocket: parts = message.split('|', 1) if len(parts) == 2: username, content = parts await save_chat_entry(username, content) await broadcast_message(f"{username}|{content}", room_id) except websockets.ConnectionClosed: pass finally: if room_id in active_connections and client_id in active_connections[room_id]: del active_connections[room_id][client_id] async def broadcast_message(message, room_id): """๐ข - Shout Out Bout - Blasts the word, no doubt!""" action = "๐ข - Shout Out Bout - Blasts the word, no doubt!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if room_id in active_connections: disconnected = [] for client_id, ws in active_connections[room_id].items(): try: await ws.send(message) except websockets.ConnectionClosed: disconnected.append(client_id) for client_id in disconnected: del active_connections[room_id][client_id] async def run_websocket_server(): """๐ฅ๏ธ - Server Ferver Verve - Spins the web, with nerve!""" action = "๐ฅ๏ธ - Server Ferver Verve - Spins the web, with nerve!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) global server_running, server_task if not server_running: server = await websockets.serve(websocket_handler, '0.0.0.0', 8765) server_running = True await server.wait_closed() async def process_voice_input(audio_bytes): """๐ค - Voice Choice Rejoice - Speaks your mind, oh boy!""" action = "๐ค - Voice Choice Rejoice - Speaks your mind, oh boy!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if audio_bytes: # Simple placeholder for speech-to-text (requires additional library like `speech_recognition`) # For now, we'll simulate text input from audio text = "Voice input simulation" # Replace with actual speech-to-text logic await save_chat_entry(username, text) async def create_streamlit_interface(): """๐จ - UI Brew Crew - Builds the view, so new!""" action = "๐จ - UI Brew Crew - Builds the view, so new!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) if 'username' not in st.session_state: chat_content = await load_chat() available_names = [name for name in FUN_USERNAMES if not any(f"{name} has joined" in line for line in chat_content.split('\n'))] st.session_state.username = random.choice(available_names) if available_names else random.choice(list(FUN_USERNAMES.keys())) if 'refresh_rate' not in st.session_state: st.session_state.refresh_rate = 5 if 'timer_start' not in st.session_state: st.session_state.timer_start = time.time() if 'quote_line' not in st.session_state: st.session_state.quote_line = None if 'pasted_image_data' not in st.session_state: st.session_state.pasted_image_data = None if 'message_text' not in st.session_state: st.session_state.message_text = "" if 'audio_cache' not in st.session_state: st.session_state.audio_cache = {} if 'chat_history' not in st.session_state: st.session_state.chat_history = [] st.markdown(""" """, unsafe_allow_html=True) st.title(f"๐ค๐ง MMO {st.session_state.username}๐๐ฌ") st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy quoting! ๐") # Voice Input audio_bytes = audio_recorder() if audio_bytes: await process_voice_input(audio_bytes) st.rerun() # Chat Section st.subheader(f"{START_ROOM} Chat ๐ฌ") chat_content = await load_chat() chat_lines = chat_content.split('\n') chat_votes = await load_votes(QUOTE_VOTES_FILE) for i, line in enumerate(chat_lines): if line.strip() and ': ' in line: col1, col2, col3 = st.columns([4, 1, 1]) with col1: st.markdown(line) username = line.split(': ')[1].split(' ')[0] audio_file = None cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}" if cache_key in st.session_state.audio_cache: audio_file = st.session_state.audio_cache[cache_key] else: audio_file = await async_edge_tts_generate(line.split(': ', 1)[1], FUN_USERNAMES.get(username, "en-US-AriaNeural")) st.session_state.audio_cache[cache_key] = audio_file if audio_file: play_and_download_audio(audio_file) with col2: vote_count = chat_votes.get(line.split('. ')[1] if '. ' in line else line, 0) if st.button(f"๐ {vote_count}", key=f"chat_vote_{i}"): comment = st.session_state.message_text await save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, await generate_user_hash(), st.session_state.username, comment) if st.session_state.pasted_image_data: filename = await save_pasted_image(st.session_state.pasted_image_data) if filename: await save_chat_entry(st.session_state.username, f"Pasted image: {filename}") st.session_state.pasted_image_data = None st.session_state.message_text = '' st.rerun() with col3: if st.button("๐ข Quote", key=f"quote_{i}"): st.session_state.quote_line = line st.rerun() # Quoting Section if 'quote_line' in st.session_state: st.markdown(f"### Quoting: {st.session_state.quote_line}") quote_response = st.text_area("Add your response", key="quote_response") if st.button("Send Quote ๐", key="send_quote"): async def process_quote(): """๐ข - Quote Float Boat - Echoes chat, we gloat!""" action = "๐ข - Quote Float Boat - Echoes chat, we gloat!" await asyncio.to_thread(log_action, st.session_state.username, action) markdown_response = f"### Quote Response\n- **Original**: {st.session_state.quote_line}\n- **{st.session_state.username} Replies**: {quote_response}" if st.session_state.pasted_image_data: filename = await save_pasted_image(st.session_state.pasted_image_data) if filename: markdown_response += f"\n- **Image**: " st.session_state.pasted_image_data = None await save_chat_entry(st.session_state.username, markdown_response) await process_quote() del st.session_state.quote_line st.session_state.message_text = '' st.rerun() # Username Change Dropdown new_username = st.selectbox("Change Name", [""] + list(FUN_USERNAMES.keys()), index=0) if new_username and new_username != st.session_state.username: await save_chat_entry("System ๐", f"{st.session_state.username} changed name to {new_username}") st.session_state.username = new_username st.rerun() # Message Input message = st.text_input(f"Message as {st.session_state.username}", key="message_input", value=st.session_state.message_text, on_change=lambda: st.session_state.update(message_text=st.session_state.message_input)) if st.button("Send ๐", key="send_button") and message.strip(): await save_chat_entry(st.session_state.username, message) if st.session_state.pasted_image_data: filename = await save_pasted_image(st.session_state.pasted_image_data) if filename: await save_chat_entry(st.session_state.username, f"Pasted image: {filename}") st.session_state.pasted_image_data = None st.session_state.message_text = '' st.rerun() # Paste Target paste_component = components.html( """
โณ {font_func('Refresh in:')} {countdown_str}
", unsafe_allow_html=True) await asyncio.sleep(1) st.rerun() # Sidebar History st.sidebar.subheader("Chat History ๐") with open(HISTORY_FILE, 'r') as f: history_content = f.read() st.sidebar.markdown(history_content) async def main(): """๐ฎ - Game Fame Claim - Starts the fun, no shame!""" action = "๐ฎ - Game Fame Claim - Starts the fun, no shame!" username = st.session_state.get('username', 'System ๐') await asyncio.to_thread(log_action, username, action) global NODE_NAME, server_task NODE_NAME, port = await get_node_name() if server_task is None: server_task = asyncio.create_task(run_websocket_server()) await create_streamlit_interface() if __name__ == "__main__": asyncio.run(main())