Create app.py

app.py

@@ -0,0 +1,394 @@
+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
+
+# 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
+FUN_USERNAMES = [
+    "CosmicJester 🌌", "PixelPanda 🐼", "QuantumQuack 🦆", "StellarSquirrel 🐿️",
+    "GizmoGuru ⚙️", "NebulaNinja 🌠", "ByteBuster 💾", "GalacticGopher 🌍",
+    "RocketRaccoon 🚀", "EchoElf 🧝", "PhantomFox 🦊", "WittyWizard 🧙",
+    "LunarLlama 🌙", "SolarSloth ☀️", "AstroAlpaca 🦙", "CyberCoyote 🐺",
+    "MysticMoose 🦌", "GlitchGnome 🧚", "VortexViper 🐍", "ChronoChimp 🐒"
+]
+
+# Directories and files
+CHAT_DIR = "chat_logs"
+VOTE_DIR = "vote_logs"
+STATE_FILE = "user_state.txt"
+os.makedirs(CHAT_DIR, exist_ok=True)
+os.makedirs(VOTE_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(VOTE_DIR, "vote_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)),
+    # ... (other font styles remain the same)
+]
+
+server_running = False
+server_task = None
+
+def get_node_name():
+    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 save_chat_entry(username, message):
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    entry = f"[{timestamp}] {username}: {message}"
+    with open(CHAT_FILE, 'a') as f:
+        f.write(f"{entry}\n")
+
+def load_chat():
+    if not os.path.exists(CHAT_FILE):
+        with open(CHAT_FILE, 'w') as f:
+            f.write(f"# {START_ROOM} Chat\n\nWelcome to the cosmic hub - start chatting! 🎤\n")
+    with open(CHAT_FILE, 'r') as f:
+        content = f.read()
+    lines = content.strip().split('\n')
+    return "\n".join(f"{i+1}. {line}" for i, line in enumerate(lines) if line.strip())
+
+def get_user_list(chat_content):
+    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))
+
+def has_joined_before(client_id, chat_content):
+    return any(f"Client-{client_id} has joined" in line for line in chat_content.split('\n'))
+
+def get_message_suggestions(chat_content, prefix):
+    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]
+
+def load_quotes(source="famous"):
+    famous_quotes = [
+        "The true sign of intelligence is not knowledge but imagination. – Albert Einstein",
+        # ... (other quotes remain the same)
+    ]
+    custom_quotes = [
+        "Every age unfolds a new lesson. Life's chapters evolve, each teaching us anew.",
+        # ... (other custom quotes remain the same)
+    ]
+    return famous_quotes if source == "famous" else custom_quotes
+
+def save_vote(file, item, user_hash, username, comment=""):
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    entry = f"[{timestamp}] {user_hash} voted for {item}"
+    with open(file, 'a') as f:
+        f.write(f"{entry}\n")
+    with open(HISTORY_FILE, "a") as f:
+        f.write(f"- {timestamp} - User {user_hash} voted for {item}\n")
+    # Add a new chat entry for the upvote
+    chat_message = f"{username} upvoted: \"{item}\""
+    if comment:
+        chat_message += f" - {comment}"
+    save_chat_entry(username, chat_message)
+
+def load_votes(file):
+    if not os.path.exists(file):
+        with open(file, 'w') as f:
+            f.write("# Vote Tally\n\nNo votes yet - get clicking! 🖱️\n")
+    with open(file, 'r') as f:
+        lines = f.read().strip().split('\n')[2:]  # Skip header
+    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
+
+def generate_user_hash():
+    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
+
+def save_pasted_image(image_data):
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    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))
+    img.save(filepath, "PNG")
+    return filename
+
+def get_video_html(video_path, width="100%"):
+    video_url = f"data:video/mp4;base64,{base64.b64encode(open(video_path, 'rb').read()).decode()}"
+    return f'<video width="{width}" controls autoplay muted loop><source src="{video_url}" type="video/mp4">Your browser does not support the video tag.</video>'
+
+def get_audio_html(audio_path, width="100%"):
+    audio_url = f"data:audio/mpeg;base64,{base64.b64encode(open(audio_path, 'rb').read()).decode()}"
+    return f'<audio controls style="width: {width};"><source src="{audio_url}" type="audio/mpeg">Your browser does not support the audio element.</audio>'
+
+active_connections = {}
+
+async def websocket_handler(websocket, path):
+    try:
+        client_id = str(uuid.uuid4())
+        room_id = "chat"
+        active_connections.setdefault(room_id, {})[client_id] = websocket
+        chat_content = load_chat()
+        username = st.session_state.get('username', random.choice(FUN_USERNAMES))
+        if not has_joined_before(client_id, chat_content):
+            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
+                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):
+    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():
+    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()
+
+def create_streamlit_interface():
+    # Dynamic title based on username
+    if 'username' in st.session_state:
+        dynamic_title = f"🤖🧠MMO {st.session_state.username}📝🔬"
+    else:
+        dynamic_title = "🤖🧠MMO Chat Brain📝🔬"
+
+    # Base CSS
+    st.markdown("""
+        <style>
+        .chat-box {font-family: monospace; background: #1e1e1e; color: #d4d4d4; padding: 10px; border-radius: 5px; height: 300px; overflow-y: auto;}
+        .timer {font-size: 24px; color: #ffcc00; text-align: center; animation: pulse 1s infinite;}
+        @keyframes pulse {0% {transform: scale(1);} 50% {transform: scale(1.1);} 100% {transform: scale(1);}}
+        #paste-target {border: 2px dashed #ccc; padding: 20px; text-align: center; cursor: pointer;}
+        </style>
+    """, unsafe_allow_html=True)
+
+    st.title(dynamic_title)
+    st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy! 🎉")
+
+    # Initialize client ID and username
+    if 'client_id' not in st.session_state:
+        st.session_state.client_id = str(uuid.uuid4())
+    if 'username' not in st.session_state:
+        available_names = [name for name in FUN_USERNAMES if not any(f"{name} has joined" in line for line in load_chat().split('\n'))]
+        st.session_state.username = random.choice(available_names) if available_names else random.choice(FUN_USERNAMES)
+
+    # Session state initialization
+    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_index' not in st.session_state:
+        quotes = load_quotes("famous")
+        st.session_state.quote_index = random.randint(0, max(0, len(quotes) - 1)) if quotes else 0
+    if 'quote_source' not in st.session_state:
+        st.session_state.quote_source = "famous"
+    if 'pasted_image_data' not in st.session_state:
+        st.session_state.pasted_image_data = None
+
+    # JavaScript component for image paste
+    paste_component = components.html(
+        """
+        <div id="paste-target">Paste an image here (Ctrl+V)</div>
+        <script>
+        const pasteTarget = document.getElementById('paste-target');
+        pasteTarget.addEventListener('paste', (event) => {
+            const items = (event.clipboardData || window.clipboardData).items;
+            for (let i = 0; i < items.length; i++) {
+                if (items[i].type.indexOf('image') !== -1) {
+                    const blob = items[i].getAsFile();
+                    const reader = new FileReader();
+                    reader.onload = (e) => {
+                        window.parent.postMessage({
+                            type: 'streamlit:setComponentValue',
+                            value: e.target.result
+                        }, '*');
+                        pasteTarget.innerHTML = '<p>Image pasted! Processing...</p>';
+                    };
+                    reader.readAsDataURL(blob);
+                }
+            }
+            event.preventDefault();
+        });
+        </script>
+        """,
+        height=100
+    )
+
+    # Handle pasted image data
+    pasted_image = st.session_state.get('pasted_image_data')
+    if pasted_image:
+        filename = save_pasted_image(pasted_image)
+        if filename:
+            save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+            st.session_state.pasted_image_data = None  # Clear after processing
+            st.rerun()
+
+    # Chat section
+    st.subheader(f"{START_ROOM} Chat 💬")
+    chat_content = load_chat()
+    chat_votes = load_votes(QUOTE_VOTES_FILE)
+    for i, line in enumerate(chat_content.split('\n')):
+        if line.strip() and ': ' in line:
+            col1, col2 = st.columns([5, 1])
+            with col1:
+                st.markdown(line)
+            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.get('message_input', '')
+                    save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, generate_user_hash(), st.session_state.username, comment)
+                    if pasted_image:
+                        filename = save_pasted_image(pasted_image)
+                        if filename:
+                            save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                            st.session_state.pasted_image_data = None
+                    st.session_state.message_input = ''  # Clear input after vote
+                    st.rerun()
+
+    # Username change dropdown
+    new_username = st.selectbox("Change Name", [""] + FUN_USERNAMES, index=0)
+    if new_username and new_username != st.session_state.username:
+        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")
+    if st.button("Send 🚀", key="send_button") and message.strip():
+        save_chat_entry(st.session_state.username, message)
+        st.session_state.message_input = ''  # Clear input after send
+        if st.session_state.get('pasted_image_data'):
+            filename = save_pasted_image(st.session_state.pasted_image_data)
+            if filename:
+                save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                st.session_state.pasted_image_data = None
+        st.rerun()
+
+    # Media section with upload and delete
+    st.subheader("Media Gallery 🎨🎶🎥")
+    uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp3', 'mp4'])
+    if uploaded_file:
+        file_path = os.path.join('./', uploaded_file.name)
+        with open(file_path, 'wb') as f:
+            f.write(uploaded_file.getbuffer())
+        st.success(f"Uploaded {uploaded_file.name}")
+
+    media_files = glob.glob("./*.png") + glob.glob("./*.jpg") + glob.glob("./*.mp3") + glob.glob("./*.mp4")
+    if media_files:
+        cols = st.columns(3)
+        media_votes = load_votes(MEDIA_VOTES_FILE)
+        for idx, media_file in enumerate(media_files):
+            vote_count = media_votes.get(media_file, 0)
+            if vote_count > 0:  # Only show if upvoted
+                with cols[idx % 3]:
+                    if media_file.endswith(('.png', '.jpg')):
+                        st.image(media_file, use_container_width=True)
+                    elif media_file.endswith('.mp3'):
+                        st.markdown(get_audio_html(media_file), unsafe_allow_html=True)
+                    elif media_file.endswith('.mp4'):
+                        st.markdown(get_video_html(media_file), unsafe_allow_html=True)
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        if st.button(f"👍 {vote_count}", key=f"media_vote_{idx}"):
+                            comment = st.session_state.get('message_input', '')
+                            save_vote(MEDIA_VOTES_FILE, media_file, generate_user_hash(), st.session_state.username, comment)
+                            if st.session_state.get('pasted_image_data'):
+                                filename = save_pasted_image(st.session_state.pasted_image_data)
+                                if filename:
+                                    save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                                    st.session_state.pasted_image_data = None
+                            st.session_state.message_input = ''  # Clear input after vote
+                            st.rerun()
+                    with col2:
+                        if st.button("🗑️", key=f"media_delete_{idx}"):
+                            os.remove(media_file)
+                            st.rerun()
+
+    # Refresh timer
+    st.subheader("Refresh ⏳")
+    refresh_rate = st.slider("Refresh Rate", 1, 300, st.session_state.refresh_rate)
+    st.session_state.refresh_rate = refresh_rate
+    timer_placeholder = st.empty()
+    for i in range(st.session_state.refresh_rate, -1, -1):
+        font_name, font_func = random.choice(UNICODE_FONTS)
+        countdown_str = "".join(UNICODE_DIGITS[int(d)] for d in str(i)) if i < 10 else font_func(str(i))
+        timer_placeholder.markdown(f"<p class='timer'>⏳ {font_func('Refresh in:')} {countdown_str}</p>", unsafe_allow_html=True)
+        time.sleep(1)
+    st.rerun()
+
+    # Sidebar vote stats
+    st.sidebar.subheader("Vote Counts")
+    chat_votes = load_votes(QUOTE_VOTES_FILE)
+    media_votes = load_votes(MEDIA_VOTES_FILE)
+    for item, count in {**chat_votes, **media_votes}.items():
+        if count > 0:  # Only show items with upvotes
+            st.sidebar.write(f"{item}: {count} votes")
+
+async def main():
+    global NODE_NAME, server_task
+    NODE_NAME, port = get_node_name()
+    if server_task is None:
+        server_task = asyncio.create_task(run_websocket_server())
+    create_streamlit_interface()
+
+if __name__ == "__main__":
+    asyncio.run(main())