Update app.py

app.py

@@ -58,15 +58,15 @@ UNICODE_FONTS = [
 server_running = False
 server_task = None
 
-def get_node_name():
+async 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)
+    await asyncio.to_thread(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()
+    args = await asyncio.to_thread(parser.parse_args)
     return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port
 
 def log_action(username, action):
@@ -75,42 +75,38 @@ def log_action(username, action):
         st.session_state.action_log = {}
     user_log = st.session_state.action_log.setdefault(username, {})
     current_time = time.time()
-    # Clean up old entries (older than 10 seconds)
     user_log = {k: v for k, v in user_log.items() if current_time - v < 10}
     st.session_state.action_log[username] = user_log
-    # Log if action hasn't been seen recently
     if action not in user_log:
         with open(CHAT_FILE, 'a') as f:
             f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: {action}\n")
         user_log[action] = current_time
 
-def save_chat_entry(username, message):
+async def save_chat_entry(username, message):
     """📝 - Chat Snap Trap - Logs your yap, no cap! ✍️🧠"""
-    action = "📝 - Chat Snap Trap - Logs your yap, no cap! ✍️🧠"  # Double rhyme bonus!
-    log_action(username, action)
+    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}"
-    with open(CHAT_FILE, 'a') as f:
-        f.write(f"{entry}\n")
+    await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').write(f"{entry}\n"))
 
-def load_chat():
+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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     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")
+        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 = f.read()
+        content = await asyncio.to_thread(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):
+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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     users = set()
     for line in chat_content.split('\n'):
         if line.strip() and ': ' in line:
@@ -118,27 +114,27 @@ def get_user_list(chat_content):
             users.add(user)
     return sorted(list(users))
 
-def has_joined_before(client_id, chat_content):
+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 🌟')
-    log_action(username, action)
+    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'))
 
-def get_message_suggestions(chat_content, prefix):
+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 🌟')
-    log_action(username, action)
+    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]
 
-def load_quotes(source="famous"):
+async def load_quotes(source="famous"):
     """📜 - Quote Tote Note - Wise words float, we gloat!"""
     action = "📜 - Quote Tote Note - Wise words float, we gloat!"
     username = st.session_state.get('username', 'System 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     famous_quotes = [
         "The true sign of intelligence is not knowledge but imagination. – Albert Einstein",
     ]
@@ -147,31 +143,28 @@ def load_quotes(source="famous"):
     ]
     return famous_quotes if source == "famous" else custom_quotes
 
-def save_vote(file, item, user_hash, username, comment=""):
+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!"
-    log_action(username, action)
+    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}"
-    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")
+    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}"
-    save_chat_entry(username, chat_message)
+    await save_chat_entry(username, chat_message)
 
-def load_votes(file):
+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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     if not os.path.exists(file):
-        with open(file, 'w') as f:
-            f.write("# Vote Tally\n\nNo votes yet - get clicking! 🖱️\n")
+        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 = f.read().strip().split('\n')[2:]  # Skip header
+        lines = await asyncio.to_thread(f.read.strip().split, '\n')[2:]
     votes = {}
     user_votes = set()
     for line in lines:
@@ -184,20 +177,20 @@ def load_votes(file):
                 user_votes.add(vote_key)
     return votes
 
-def generate_user_hash():
+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 🌟')
-    log_action(username, action)
+    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
 
-def save_pasted_image(image_data):
+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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     filename = f"paste_{timestamp}.png"
     filepath = os.path.join('./', filename)
@@ -205,23 +198,23 @@ def save_pasted_image(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")
+    await asyncio.to_thread(img.save, filepath, "PNG")
     return filename
 
-def get_video_html(video_path, width="100%"):
+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 🌟')
-    log_action(username, action)
-    video_url = f"data:video/mp4;base64,{base64.b64encode(open(video_path, 'rb').read()).decode()}"
+    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'<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%"):
+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 🌟')
-    log_action(username, action)
-    audio_url = f"data:audio/mpeg;base64,{base64.b64encode(open(audio_path, 'rb').read()).decode()}"
+    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'<audio controls style="width: {width};"><source src="{audio_url}" type="audio/mpeg">Your browser does not support the audio element.</audio>'
 
 active_connections = {}
@@ -230,20 +223,20 @@ 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 🌟')
-    log_action(username, action)
+    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 = load_chat()
+        chat_content = await 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}!")
+        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
-                save_chat_entry(username, content)
+                await save_chat_entry(username, content)
                 await broadcast_message(f"{username}|{content}", room_id)
     except websockets.ConnectionClosed:
         pass
@@ -255,7 +248,7 @@ 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 🌟')
-    log_action(username, action)
+    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():
@@ -270,18 +263,18 @@ 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 🌟')
-    log_action(username, action)
+    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()
 
-def create_streamlit_interface():
+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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
 
     # Dynamic title based on username
     if 'username' in st.session_state:
@@ -300,13 +293,13 @@ def create_streamlit_interface():
     """, unsafe_allow_html=True)
 
     st.title(dynamic_title)
-    st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy! 🎉")
+    st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy quoting! 🎉")
 
     # 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'))]
+        available_names = [name for name in FUN_USERNAMES if not any(f"{name} has joined" in line for line in await load_chat().split('\n'))]
         st.session_state.username = random.choice(available_names) if available_names else random.choice(FUN_USERNAMES)
 
     # Session state initialization
@@ -315,7 +308,7 @@ def create_streamlit_interface():
     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")
+        quotes = await 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"
@@ -356,49 +349,75 @@ def create_streamlit_interface():
     # Handle pasted image data
     pasted_image = st.session_state.get('pasted_image_data')
     if pasted_image:
-        filename = save_pasted_image(pasted_image)
+        filename = await save_pasted_image(pasted_image)
         if filename:
-            save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+            await save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
             st.session_state.pasted_image_data = None
             st.rerun()
 
-    # Chat section
+    # Chat section with quoting
     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')):
+    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 = st.columns([5, 1])
+            col1, col2, col3 = st.columns([4, 1, 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.message_text
-                    save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, generate_user_hash(), st.session_state.username, comment)
+                    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 = save_pasted_image(st.session_state.pasted_image_data)
+                        filename = await save_pasted_image(st.session_state.pasted_image_data)
                         if filename:
-                            save_chat_entry(st.session_state.username, f"Pasted image: {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**: ![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
+            st.session_state.message_text = ''
+            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}")
+        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():
-        save_chat_entry(st.session_state.username, message)
+        await save_chat_entry(st.session_state.username, message)
         if st.session_state.pasted_image_data:
-            filename = save_pasted_image(st.session_state.pasted_image_data)
+            filename = await save_pasted_image(st.session_state.pasted_image_data)
             if filename:
-                save_chat_entry(st.session_state.username, f"Pasted image: {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()
@@ -408,14 +427,13 @@ def create_streamlit_interface():
     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())
+        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 = load_votes(MEDIA_VOTES_FILE)
+        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:
@@ -423,24 +441,24 @@ def create_streamlit_interface():
                     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)
+                        st.markdown(await 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)
+                        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
-                            save_vote(MEDIA_VOTES_FILE, media_file, generate_user_hash(), st.session_state.username, comment)
+                            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 = save_pasted_image(st.session_state.pasted_image_data)
+                                filename = await save_pasted_image(st.session_state.pasted_image_data)
                                 if filename:
-                                    save_chat_entry(st.session_state.username, f"Pasted image: {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}"):
-                            os.remove(media_file)
+                            await asyncio.to_thread(os.remove, media_file)
                             st.rerun()
 
     # Refresh timer
@@ -452,13 +470,13 @@ def create_streamlit_interface():
         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)
+        await asyncio.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)
+    chat_votes = await load_votes(QUOTE_VOTES_FILE)
+    media_votes = await load_votes(MEDIA_VOTES_FILE)
     for item, count in {**chat_votes, **media_votes}.items():
         if count > 0:
             st.sidebar.write(f"{item}: {count} votes")
@@ -467,12 +485,12 @@ 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 🌟')
-    log_action(username, action)
+    await asyncio.to_thread(log_action, username, action)
     global NODE_NAME, server_task
-    NODE_NAME, port = get_node_name()
+    NODE_NAME, port = await get_node_name()
     if server_task is None:
         server_task = asyncio.create_task(run_websocket_server())
-    create_streamlit_interface()
+    await create_streamlit_interface()
 
 if __name__ == "__main__":
     asyncio.run(main())