Update app.py

app.py

@@ -35,10 +35,11 @@ FUN_USERNAMES = [
     "MysticMoose 🦌", "GlitchGnome 🧚", "VortexViper 🐍", "ChronoChimp 🐒"
 ]
 
-# Directories for chat and votes - the secret vaults where tales and tallies are stashed! 🗄️🔒
+# Directories for chat, votes, and media - the secret vaults where treasures are stashed! 🗄️🔒
 CHAT_DIR = "chat_logs"
 VOTE_DIR = "vote_logs"
 MEDIA_DIR = "media"
+STATE_FILE = "user_state.txt"
 os.makedirs(CHAT_DIR, exist_ok=True)
 os.makedirs(VOTE_DIR, exist_ok=True)
 os.makedirs(MEDIA_DIR, exist_ok=True)
@@ -69,6 +70,10 @@ UNICODE_FONTS = [
     ("Circle", lambda x: "".join(chr(ord(c) + 0x24B6 - 0x41) if 'A' <= c <= 'Z' else chr(ord(c) + 0x24D0 - 0x61) if 'a' <= c <= 'z' else c for c in x))
 ]
 
+# Global WebSocket server flag
+server_running = False
+server_task = None
+
 # Node name - the app’s codename generator, sneaky and slick! 🕵️‍♂️💾
 def get_node_name():
     """🎲 Spins the wheel of fate to name our node - a random alias or user pick! 🏷️"""
@@ -194,6 +199,25 @@ def generate_user_hash():
         st.session_state['user_hash'] = hash_object.hexdigest()[:8]
     return st.session_state['user_hash']
 
+# Username persistence - save and load across refreshes! 💾🔄
+def save_username(username):
+    """💾 Stashes your cosmic alias before the refresh void claims it! 🌌"""
+    try:
+        with open(STATE_FILE, 'w') as f:
+            f.write(username)
+    except Exception as e:
+        print(f"Failed to save username: {e}")
+
+def load_username():
+    """🔄 Resurrects your cosmic alias from the refresh abyss! 🌠"""
+    if os.path.exists(STATE_FILE):
+        try:
+            with open(STATE_FILE, 'r') as f:
+                return f.read().strip()
+        except Exception as e:
+            print(f"Failed to load username: {e}")
+    return None
+
 # Media HTML generators - IVA’s multimedia magic! 🎥🎶🖼️
 def get_video_html(video_path, width="100%"):
     """🎬 Rolls out the red carpet for videos - autoplay and loop like a star! 🌟"""
@@ -261,12 +285,15 @@ async def broadcast_message(message, room_id):
         for client_id in disconnected:
             del active_connections[room_id][client_id]
 
-# WebSocket starter - the DJ spinning up the Sector tunes! 🎧🔥
-async def start_websocket_server(host='0.0.0.0', port=8765):
-    """🌐 Cranks up the WebSocket jukebox, ready to rock the Sector scene! 🎸"""
-    server = await websockets.serve(websocket_handler, host, port)
-    print(f"WebSocket server jamming on ws://{host}:{port}")
-    return server
+# WebSocket server runner - the DJ spinning up the Sector tunes once! 🎧🔥
+async def run_websocket_server():
+    """🌐 Cranks up the WebSocket jukebox once, keeping Sector rocking! 🎸"""
+    global server_running, server_task
+    if not server_running:
+        server = await websockets.serve(websocket_handler, '0.0.0.0', 8765)
+        print(f"WebSocket server jamming on ws://0.0.0.0:8765")
+        server_running = True
+        await server.wait_closed()
 
 # Chat interface maker - the stage builder for our Sector extravaganza! 🎭🏟️
 def create_streamlit_interface(initial_username):
@@ -301,10 +328,16 @@ def create_streamlit_interface(initial_username):
     st.title(f"{Site_Name}")
     st.markdown(f"Welcome to {START_ROOM} - chat, vote, and enjoy IVA’s media magic! 🎉")
 
-    # Session state initialization
+    # Load or set username
+    saved_username = load_username()
+    if saved_username and saved_username in FUN_USERNAMES:
+        initial_username = saved_username
     if 'username' not in st.session_state:
         st.session_state.username = initial_username
         save_chat_entry("System 🌟", f"{initial_username} has joined {START_ROOM}!")
+    save_username(st.session_state.username)  # Persist before any refresh
+
+    # Session state for refresh rate and quote index
     if 'refresh_rate' not in st.session_state:
         st.session_state.refresh_rate = 5
     if 'timer_start' not in st.session_state:
@@ -329,6 +362,7 @@ def create_streamlit_interface(initial_username):
                     user_hash = generate_user_hash()
                     save_vote(QUOTE_VOTES_FILE, line, user_hash)
                     st.session_state.timer_start = time.time()
+                    save_username(st.session_state.username)
                     st.rerun()
 
     user_list = get_user_list(chat_content)
@@ -337,11 +371,13 @@ def create_streamlit_interface(initial_username):
         save_chat_entry("System 🌟", f"{st.session_state.username} switched to {new_username} in {START_ROOM}!")
         st.session_state.username = new_username
         st.session_state.timer_start = time.time()
+        save_username(st.session_state.username)
 
     message = st.text_input("Message", placeholder="Type your epic line here! ✍️")
     if st.button("Send 🚀") and message.strip():
         save_chat_entry(st.session_state.username, message)
         st.session_state.timer_start = time.time()
+        save_username(st.session_state.username)
         st.rerun()
 
     # Quote section
@@ -358,6 +394,7 @@ def create_streamlit_interface(initial_username):
                 user_hash = generate_user_hash()
                 save_vote(QUOTE_VOTES_FILE, quote, user_hash)
                 st.session_state.timer_start = time.time()
+                save_username(st.session_state.username)
                 st.rerun()
         if time.time() - st.session_state.timer_start > 10:  # 10s quote refresh
             st.session_state.quote_index = (st.session_state.quote_index + 1) % len(quotes)
@@ -365,6 +402,7 @@ def create_streamlit_interface(initial_username):
             quotes = load_quotes(st.session_state.quote_source)
             st.session_state.quote_index = st.session_state.quote_index % len(quotes) if quotes else 0
             st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
             st.rerun()
     else:
         st.markdown("No quotes available - check back later! 📭")
@@ -391,6 +429,7 @@ def create_streamlit_interface(initial_username):
                     user_hash = generate_user_hash()
                     save_vote(IMAGE_VOTES_FILE, media_file, user_hash)
                     st.session_state.timer_start = time.time()
+                    save_username(st.session_state.username)
                     st.rerun()
     else:
         st.error("No media files (images, audio, video) found in 'media' directory!")
@@ -401,20 +440,24 @@ def create_streamlit_interface(initial_username):
     if refresh_rate != st.session_state.refresh_rate:
         st.session_state.refresh_rate = refresh_rate
         st.session_state.timer_start = time.time()
+        save_username(st.session_state.username)
 
     col1, col2, col3 = st.columns(3)
     with col1:
         if st.button("🐇 Small (1s)"):
             st.session_state.refresh_rate = 1
             st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
     with col2:
         if st.button("🐢 Medium (5s)"):
             st.session_state.refresh_rate = 5
             st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
     with col3:
         if st.button("🐘 Large (5m)"):
             st.session_state.refresh_rate = 300
             st.session_state.timer_start = time.time()
+            save_username(st.session_state.username)
 
     # Pulsing countdown timer with emoji digits and random Unicode font
     timer_placeholder = st.empty()
@@ -427,6 +470,7 @@ def create_streamlit_interface(initial_username):
         time.sleep(1)  # Pulse every second
     st.session_state.timer_start = time.time()
     st.session_state.last_refresh = time.time()
+    save_username(st.session_state.username)
     st.rerun()
 
     # Sidebar vote stats
@@ -441,9 +485,12 @@ def create_streamlit_interface(initial_username):
 # Main event - the ringmaster kicking off the Sector circus! 🎪🤡
 async def main():
     """🎤 Drops the mic and starts the Sector party - it’s showtime, folks! 🎉"""
-    global NODE_NAME
+    global NODE_NAME, server_task
     NODE_NAME, port = get_node_name()
-    await start_websocket_server()
+
+    # Start WebSocket server only once as a background task
+    if server_task is None:
+        server_task = asyncio.create_task(run_websocket_server())
 
     query_params = st.query_params if hasattr(st, 'query_params') else {}
     initial_username = query_params.get("username", random.choice(FUN_USERNAMES)) if query_params else random.choice(FUN_USERNAMES)