Update app.py

app.py

@@ -99,8 +99,13 @@ UNICODE_FONTS = [
     ("Regional Indicator", lambda x: "".join(chr(ord(c) - 0x41 + 0x1F1E6) if 'A' <= c <= 'Z' else c for c in x)),
 ]
 
-server_running = False
-server_task = None
+# Global state - keeping tabs! 🌍📋
+if 'server_running' not in st.session_state:
+    st.session_state.server_running = False
+if 'server_task' not in st.session_state:
+    st.session_state.server_task = None
+if 'active_connections' not in st.session_state:
+    st.session_state.active_connections = {}
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix():
@@ -146,7 +151,7 @@ async def load_chat():
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "📜🚀 - Chat loader - history unleashed!")
     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"))
+        await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').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
@@ -263,8 +268,6 @@ async def get_audio_html(audio_path, width="100%"):
     audio_url = f"data:audio/mpeg;base64,{base64.b64encode(await asyncio.to_thread(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 = {}
-
 # Websocket handler - chat links up! 🌐🔗
 async def websocket_handler(websocket, path):
     username = st.session_state.get('username', 'System 🌟')
@@ -272,7 +275,7 @@ async def websocket_handler(websocket, path):
     try:
         client_id = str(uuid.uuid4())
         room_id = "chat"
-        active_connections.setdefault(room_id, {})[client_id] = websocket
+        st.session_state.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):
@@ -286,31 +289,30 @@ async def websocket_handler(websocket, path):
     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]
+        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]
 
 # Message broadcaster - words fly far! 📢✈️
 async def broadcast_message(message, room_id):
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "📢✈️ - Message broadcaster - words fly far!")
-    if room_id in active_connections:
+    if room_id in st.session_state.active_connections:
         disconnected = []
-        for client_id, ws in active_connections[room_id].items():
+        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:
-            del active_connections[room_id][client_id]
+            del st.session_state.active_connections[room_id][client_id]
 
 # Server starter - web spins up! 🖥️🌀
 async def run_websocket_server():
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "🖥️🌀 - Server starter - web spins up!")
-    global server_running, server_task
-    if not server_running:
+    if not st.session_state.server_running:
         server = await websockets.serve(websocket_handler, '0.0.0.0', 8765)
-        server_running = True
+        st.session_state.server_running = True
         await server.wait_closed()
 
 # Voice processor - speech to text! 🎤📝
@@ -322,216 +324,222 @@ async def process_voice_input(audio_bytes):
         await save_chat_entry(username, text)
 
 # Interface builder - UI takes shape! 🎨🖌️
-async def create_streamlit_interface():
-    username = st.session_state.get('username', 'System 🌟')
-    await asyncio.to_thread(log_action, username, "🎨🖌️ - Interface builder - UI takes shape!")
+def create_streamlit_interface():
+    # Run async tasks within an event loop
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+
+    async def async_interface():
+        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("""
+            <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(f"🤖🧠MMO {st.session_state.username}📝🔬")
+        st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy quoting! 🎉")
+
+        # Start websocket server if not running
+        if not st.session_state.server_task:
+            st.session_state.server_task = loop.create_task(run_websocket_server())
+
+        # Voice Input
+        audio_bytes = audio_recorder()
+        if audio_bytes:
+            await process_voice_input(audio_bytes)
+            st.rerun()
 
-    if 'username' not in st.session_state:
+        # Chat Section
+        st.subheader(f"{START_ROOM} Chat 💬")
         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("""
-        <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(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)
+        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():
+                    await asyncio.to_thread(log_action, st.session_state.username, "📢💬 - Quote processor - echoes resound!")
+                    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:
-                            await save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                            markdown_response += f"\n- **Image**: ![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():
-                await asyncio.to_thread(log_action, st.session_state.username, "📢💬 - Quote processor - echoes resound!")
-                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**: ![Pasted Image]({filename})"
-                        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
+                    await save_chat_entry(st.session_state.username, markdown_response)
+                loop.run_until_complete(process_quote())
+                del st.session_state.quote_line
+                st.session_state.message_text = ''
+                st.rerun()
+
+        # Username changer
+        new_username = st.selectbox("Change Name", [""] + list(FUN_USERNAMES.keys()), index=0)
+        if new_username and new_username != st.session_state.username:
+            loop.run_until_complete(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():
+            loop.run_until_complete(save_chat_entry(st.session_state.username, message))
+            if st.session_state.pasted_image_data:
+                filename = loop.run_until_complete(save_pasted_image(st.session_state.pasted_image_data))
+                if filename:
+                    loop.run_until_complete(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()
 
-    # Username changer
-    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
+        # Paste Target
+        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
+        )
+
+        # Media Section
+        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)
+            await asyncio.to_thread(lambda: open(file_path, 'wb').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 = loop.run_until_complete(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:
+                    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(loop.run_until_complete(get_audio_html(media_file)), unsafe_allow_html=True)
+                        elif media_file.endswith('.mp4'):
+                            st.markdown(loop.run_until_complete(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.message_text
+                                loop.run_until_complete(save_vote(MEDIA_VOTES_FILE, media_file, await generate_user_hash(), st.session_state.username, comment))
+                                if st.session_state.pasted_image_data:
+                                    filename = loop.run_until_complete(save_pasted_image(st.session_state.pasted_image_data))
+                                    if filename:
+                                        loop.run_until_complete(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 col2:
+                            if st.button("🗑️", key=f"media_delete_{idx}"):
+                                await asyncio.to_thread(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)
+            loop.run_until_complete(asyncio.sleep(1))
         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()
+        # Sidebar History
+        st.sidebar.subheader("Chat History 📜")
+        with open(HISTORY_FILE, 'r') as f:
+            history_content = f.read()
+        st.sidebar.markdown(history_content)
 
-    # Paste Target
-    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
-    )
-
-    # Media Section
-    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)
-        await asyncio.to_thread(lambda: open(file_path, 'wb').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 = await 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:
-                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(await get_audio_html(media_file), unsafe_allow_html=True)
-                    elif media_file.endswith('.mp4'):
-                        st.markdown(await 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.message_text
-                            await save_vote(MEDIA_VOTES_FILE, media_file, 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 col2:
-                        if st.button("🗑️", key=f"media_delete_{idx}"):
-                            await asyncio.to_thread(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)
-        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)
-
-# Main execution
-NODE_NAME, port = get_node_name()
-server_task = None
-
-# Start the websocket server
-if server_task is None:
-    server_task = asyncio.create_task(run_websocket_server())
-
-# Run the interface
-asyncio.run(create_streamlit_interface())
+    # Run the async interface
+    loop.run_until_complete(async_interface())
+
+# Main execution - let’s roll! 🎲🚀
+def main():
+    NODE_NAME, port = get_node_name()
+    create_streamlit_interface()
+
+if __name__ == "__main__":
+    main()