Update app.py

app.py

@@ -17,8 +17,6 @@ import streamlit.components.v1 as components
 import edge_tts
 from audio_recorder_streamlit import audio_recorder
 import nest_asyncio
-import re
-from streamlit_paste_button import paste_image_button  # Using this for image pasting
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -35,7 +33,7 @@ st.set_page_config(
     initial_sidebar_state="auto"
 )
 
-# Funky usernames - who’s who in the zoo with unique voices! 🎭🐾🎙️
+# Funky usernames - who’s who in the zoo! 🎭🐾
 FUN_USERNAMES = {
     "CosmicJester 🌌": "en-US-AriaNeural",
     "PixelPanda 🐼": "en-US-JennyNeural",
@@ -46,17 +44,17 @@ FUN_USERNAMES = {
     "ByteBuster 💾": "en-GB-RyanNeural",
     "GalacticGopher 🌍": "en-AU-WilliamNeural",
     "RocketRaccoon 🚀": "en-CA-LiamNeural",
-    "EchoElf 🧝": "en-US-AnaNeural",
-    "PhantomFox 🦊": "en-US-BrandonNeural",
-    "WittyWizard 🧙": "en-GB-ThomasNeural",
-    "LunarLlama 🌙": "en-AU-FreyaNeural",
-    "SolarSloth ☀️": "en-CA-LindaNeural",
-    "AstroAlpaca 🦙": "en-US-ChristopherNeural",
-    "CyberCoyote 🐺": "en-GB-ElliotNeural",
-    "MysticMoose 🦌": "en-AU-JamesNeural",
-    "GlitchGnome 🧚": "en-CA-EthanNeural",
-    "VortexViper 🐍": "en-US-AmberNeural",
-    "ChronoChimp 🐒": "en-GB-LibbyNeural"
+    "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"
 }
 
 # Folders galore - organizing chaos! 📂🌀
@@ -65,12 +63,10 @@ VOTE_DIR = "vote_logs"
 STATE_FILE = "user_state.txt"
 AUDIO_DIR = "audio_logs"
 HISTORY_DIR = "history_logs"
-MEDIA_DIR = "media_files"
 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)
-os.makedirs(MEDIA_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")
@@ -110,26 +106,6 @@ 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 = {}
-if 'media_notifications' not in st.session_state:
-    st.session_state.media_notifications = []
-if 'last_chat_update' not in st.session_state:
-    st.session_state.last_chat_update = 0
-if 'displayed_chat_lines' not in st.session_state:
-    st.session_state.displayed_chat_lines = []
-if 'old_val' not in st.session_state:
-    st.session_state.old_val = ""
-if 'last_query' not in st.session_state:
-    st.session_state.last_query = ""
-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 'pasted_image_data' not in st.session_state:
-    st.session_state.pasted_image_data = None
-if 'quote_line' not in st.session_state:
-    st.session_state.quote_line = None
-if 'refresh_rate' not in st.session_state:
-    st.session_state.refresh_rate = 5  # Default refresh rate
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix():
@@ -158,12 +134,6 @@ def log_action(username, action):
             f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: {action}\n")
         user_log[action] = current_time
 
-# Clean text - strip the fancy stuff! 🧹📝
-def clean_text_for_tts(text):
-    cleaned = re.sub(r'[#*!\[\]]+', '', text)
-    cleaned = ' '.join(cleaned.split())
-    return cleaned[:200] if cleaned else "No text to speak"
-
 # Chat saver - words locked tight! 💬🔒
 async def save_chat_entry(username, message):
     await asyncio.to_thread(log_action, username, "💬🔒 - Chat saver - words locked tight!")
@@ -171,13 +141,10 @@ async def save_chat_entry(username, message):
     entry = f"[{timestamp}] {username}: {message}"
     await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').write(f"{entry}\n"))
     voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
-    cleaned_message = clean_text_for_tts(message)
-    audio_file = await async_edge_tts_generate(cleaned_message, voice)
+    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")
-    await broadcast_message(f"{username}|{message}", "chat")
-    st.session_state.last_chat_update = time.time()
 
 # Chat loader - history unleashed! 📜🚀
 async def load_chat():
@@ -204,7 +171,7 @@ async def get_user_list(chat_content):
 async def has_joined_before(client_id, chat_content):
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "🚪🔍 - Join checker - been here before?")
-    return any(f"Client-{client_id}" in line for line in chat_content.split('\n'))
+    return any(f"Client-{client_id} has joined" in line for line in chat_content.split('\n'))
 
 # Suggestion maker - old quips resurface! 💡📝
 async def get_message_suggestions(chat_content, prefix):
@@ -234,7 +201,7 @@ async def load_votes(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:
         content = await asyncio.to_thread(f.read)
-        lines = content.strip().split('\n')[2:]
+        lines = content.strip().split('\n')[2:]  # Process content after reading
     votes = {}
     user_votes = set()
     for line in lines:
@@ -262,13 +229,8 @@ async def async_edge_tts_generate(text, voice, rate=0, pitch=0, file_format="mp3
     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")
-    try:
-        await communicate.save(filename)
-        return filename if os.path.exists(filename) else None
-    except edge_tts.exceptions.NoAudioReceived:
-        with open(HISTORY_FILE, 'a') as f:
-            f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: Audio failed - No audio received for '{text}'\n")
-        return None
+    await communicate.save(filename)
+    return filename if os.path.exists(filename) else None
 
 # Audio player - tunes blast off! 🔊🚀
 def play_and_download_audio(file_path):
@@ -279,24 +241,26 @@ def play_and_download_audio(file_path):
         dl_link = f'<a href="data:audio/mpeg;base64,{b64}" download="{os.path.basename(file_path)}">🎵 Download {os.path.basename(file_path)}</a>'
         st.markdown(dl_link, unsafe_allow_html=True)
 
-# Image saver - pics preserved with naming! 📸💾
-async def save_pasted_image(image, username):
+# Image saver - pics preserved! 📸💾
+async def save_pasted_image(image_data):
+    username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "📸💾 - Image saver - pics preserved!")
     timestamp = format_timestamp_prefix()
-    filename = f"paste_{timestamp}_{username}.png"
-    filepath = os.path.join(MEDIA_DIR, filename)
-    # Directly save the PIL Image object without base64 conversion
-    await asyncio.to_thread(image.save, filepath, "PNG")
-    return filepath
-
-# Video renderer - movies roll with autoplay! 🎥🎬
+    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
+
+# Video renderer - movies roll! 🎥🎬
 async def get_video_html(video_path, width="100%"):
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "🎥🎬 - Video renderer - movies roll!")
-    with open(video_path, 'rb') as f:
-        video_data = await asyncio.to_thread(f.read)
-    video_url = f"data:video/mp4;base64,{base64.b64encode(video_data).decode()}"
-    return f'<video width="{width}" controls autoplay><source src="{video_url}" type="video/mp4">Your browser does not support the video tag.</video>'
+    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>'
 
 # Audio renderer - sounds soar! 🎶✈️
 async def get_audio_html(audio_path, width="100%"):
@@ -322,6 +286,7 @@ async def websocket_handler(websocket, path):
             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:
@@ -356,23 +321,12 @@ async def process_voice_input(audio_bytes):
     username = st.session_state.get('username', 'System 🌟')
     await asyncio.to_thread(log_action, username, "🎤📝 - Voice processor - speech to text!")
     if audio_bytes:
-        text = "Voice input simulation"
+        text = "Voice input simulation"  # Replace with actual speech-to-text logic
         await save_chat_entry(username, text)
 
-# Dummy AI lookup function (replace with actual implementation)
-async def perform_ai_lookup(query, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=False, useArxiv=True, useArxivAudio=False):
-    username = st.session_state.get('username', 'System 🌟')
-    result = f"AI Lookup Result for '{query}' (Arxiv: {useArxiv}, Audio: {useArxivAudio})"
-    await save_chat_entry(username, result)
-    if useArxivAudio:
-        audio_file = await async_edge_tts_generate(result, FUN_USERNAMES.get(username, "en-US-AriaNeural"))
-        if audio_file:
-            st.audio(audio_file)
-
-# Main execution - let’s roll! 🎲🚀
-def main():
-    NODE_NAME, port = get_node_name()
-
+# 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)
 
@@ -382,151 +336,187 @@ def main():
             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()
 
-        # Load and display chat
+        # 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()
 
-        current_time = time.time()
-        if current_time - st.session_state.last_chat_update > 1 or not st.session_state.displayed_chat_lines:
-            new_lines = [line for line in chat_lines if line.strip() and ': ' in line and line not in st.session_state.displayed_chat_lines and not line.startswith('#')]
-            st.session_state.displayed_chat_lines.extend(new_lines)
-            st.session_state.last_chat_update = current_time
-
-        for i, line in enumerate(st.session_state.displayed_chat_lines):
-            col1, col2, col3, col4 = st.columns([3, 1, 1, 2])
-            with col1:
-                st.markdown(line)
-                if "Pasted image:" in line or "Uploaded media:" in line:
-                    file_path = line.split(': ')[-1].strip()
-                    if os.path.exists(file_path):
-                        if file_path.endswith(('.png', '.jpg')):
-                            st.image(file_path, use_container_width=True)
-                        elif file_path.endswith('.mp4'):
-                            st.markdown(await get_video_html(file_path), unsafe_allow_html=True)
-                        elif file_path.endswith('.mp3'):
-                            st.markdown(await get_audio_html(file_path), unsafe_allow_html=True)
-            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)
-                    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()
-            with col4:
-                username = line.split(': ')[1].split(' ')[0]
-                cache_key = f"{line}_{FUN_USERNAMES.get(username, 'en-US-AriaNeural')}"
-                if cache_key not in st.session_state.audio_cache:
-                    cleaned_text = clean_text_for_tts(line.split(': ', 1)[1])
-                    audio_file = await async_edge_tts_generate(cleaned_text, FUN_USERNAMES.get(username, "en-US-AriaNeural"))
-                    st.session_state.audio_cache[cache_key] = audio_file
-                audio_file = st.session_state.audio_cache.get(cache_key)
-                if audio_file:
-                    play_and_download_audio(audio_file)
-
-        if st.session_state.quote_line:
+        # 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")
-            # Add paste button within quote response
-            paste_result_quote = paste_image_button("📋 Paste Image with Quote", key="paste_button_quote")
-            if paste_result_quote.image_data is not None:
-                st.image(paste_result_quote.image_data, caption="Received Image for Quote")
-                filename = await save_pasted_image(paste_result_quote.image_data, st.session_state.username)
-                if filename:
-                    st.session_state.pasted_image_data = filename  # Store for use in quote submission
             if st.button("Send Quote 🚀", key="send_quote"):
-                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:
-                    markdown_response += f"\n- **Image**: ![Pasted Image]({st.session_state.pasted_image_data})"
-                    await save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}")
-                    st.session_state.pasted_image_data = None
-                await save_chat_entry(st.session_state.username, markdown_response)
-                st.session_state.quote_line = None
+                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)
+                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:
-            await save_chat_entry("System 🌟", f"{st.session_state.username} changed name to {new_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))
-        # Add paste button next to message input
-        paste_result_msg = paste_image_button("📋 Paste Image with Message", key="paste_button_msg")
-        if paste_result_msg.image_data is not None:
-            st.image(paste_result_msg.image_data, caption="Received Image for Message")
-            filename = await save_pasted_image(paste_result_msg.image_data, st.session_state.username)
-            if filename:
-                st.session_state.pasted_image_data = filename  # Store for use in message submission
-        if st.button("Send 🚀", key="send_button") and (message.strip() or st.session_state.pasted_image_data):
-            if message.strip():
-                await save_chat_entry(st.session_state.username, message)
+        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:
-                await save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}")
-                st.session_state.pasted_image_data = None
+                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()
 
-        # Main action tabs and model use choices
-        tab_main = st.radio("Action:", ["🎤 Voice", "📸 Media", "🔍 ArXiv", "📝 Editor"], horizontal=True)
-        useArxiv = st.checkbox("Search Arxiv for Research Paper Answers", value=True)
-        useArxivAudio = st.checkbox("Generate Audio File for Research Paper Answers", value=False)
-
-        # Enhanced Media Gallery with Image, Audio, Video
-        st.subheader("Upload Media 🎨🎶🎥")
-        uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp4', 'mp3'])
+        # 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:
-            timestamp = format_timestamp_prefix()
-            username = st.session_state.username
-            ext = uploaded_file.name.split('.')[-1]
-            filename = f"{timestamp}_{username}.{ext}"
-            file_path = os.path.join(MEDIA_DIR, filename)
+            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 {filename}")
-            await save_chat_entry(username, f"Uploaded media: {file_path}")
-            if file_path.endswith('.mp4'):
-                st.session_state.media_notifications.append(file_path)
+            st.success(f"Uploaded {uploaded_file.name}")
 
-        st.subheader("Media Gallery 🎨🎶🎥")
-        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp4") + glob.glob(f"{MEDIA_DIR}/*.mp3")
+        media_files = glob.glob("./*.png") + glob.glob("./*.jpg") + glob.glob("./*.mp3") + glob.glob("./*.mp4")
         if media_files:
-            media_votes = await load_votes(MEDIA_VOTES_FILE)
-            st.write("### All Media Uploads")
-            for media_file in sorted(media_files, key=os.path.getmtime, reverse=True):  # Sort by modification time, newest first
-                filename = os.path.basename(media_file)
+            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)
-                
-                col1, col2 = st.columns([3, 1])
-                with col1:
-                    st.markdown(f"**{filename}**")
-                    if media_file.endswith(('.png', '.jpg')):
-                        st.image(media_file, use_container_width=True)
-                    elif media_file.endswith('.mp4'):
-                        st.markdown(await get_video_html(media_file), unsafe_allow_html=True)
-                    elif media_file.endswith('.mp3'):
-                        st.markdown(await get_audio_html(media_file), unsafe_allow_html=True)
-                with col2:
-                    if st.button(f"👍 {vote_count}", key=f"media_vote_{media_file}"):
-                        await save_vote(MEDIA_VOTES_FILE, media_file, await generate_user_hash(), st.session_state.username)
-                        st.rerun()
-
+                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
@@ -535,15 +525,22 @@ def main():
             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)  # Use synchronous sleep
+            loop.run_until_complete(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)
 
+    # 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()