Update app.py

app.py

@@ -379,4 +379,192 @@ def create_streamlit_interface():
         """, 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
+        st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy quoting! 🎉")
+
+        if not st.session_state.server_task:
+            st.session_state.server_task = loop.create_task(run_websocket_server())
+
+        audio_bytes = audio_recorder()
+        if audio_bytes:
+            await process_voice_input(audio_bytes)
+            st.rerun()
+
+        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:
+                        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
+                    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, st.session_state.username)
+                            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()
+
+        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, st.session_state.username)
+                        if filename:
+                            markdown_response += f"\n- **Image**: ![Pasted Image]({filename})"
+                            st.session_state.pasted_image_data = None
+                    try:
+                        await save_chat_entry(st.session_state.username, markdown_response)
+                    except edge_tts.exceptions.NoAudioReceived:
+                        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                        with open(HISTORY_FILE, 'a') as f:
+                            f.write(f"[{timestamp}] {st.session_state.username}: Quote saved without audio - No audio received\n")
+                        await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').write(f"[{timestamp}] {st.session_state.username}: {markdown_response}\n"))
+                loop.run_until_complete(process_quote())
+                del st.session_state.quote_line
+                st.session_state.message_text = ''
+                st.rerun()
+
+        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 = 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, st.session_state.username))
+                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()
+
+        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
+        )
+
+        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)
+            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)  # Trigger autoplay for all clients
+
+        # Organize gallery by user
+        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp3") + glob.glob(f"{MEDIA_DIR}/*.mp4")
+        if media_files:
+            media_votes = loop.run_until_complete(load_votes(MEDIA_VOTES_FILE))
+            users = sorted(set(f.split('_')[-1].split('.')[0] for f in media_files))  # Extract usernames
+            for user in users:
+                with st.expander(f"{user}'s Media"):
+                    user_files = [f for f in media_files if user in f]
+                    for media_file in user_files:
+                        vote_count = media_votes.get(media_file, 0)
+                        if vote_count > 0:
+                            col1, col2, col3 = st.columns([3, 1, 1])
+                            with col1:
+                                if media_file.endswith(('.png', '.jpg')):
+                                    st.image(media_file, use_container_width=True, caption=os.path.basename(media_file))
+                                elif media_file.endswith('.mp3'):
+                                    st.markdown(await get_audio_html(media_file), unsafe_allow_html=True)
+                                    st.caption(os.path.basename(media_file))
+                                elif media_file.endswith('.mp4'):
+                                    st.markdown(await get_video_html(media_file), unsafe_allow_html=True)
+                                    st.caption(os.path.basename(media_file))
+                                    if media_file in st.session_state.media_notifications:
+                                        st.session_state.media_notifications.remove(media_file)  # Autoplay once
+                            with col2:
+                                if st.button(f"👍 {vote_count}", key=f"media_vote_{media_file}"):
+                                    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))
+                                    st.session_state.message_text = ''
+                                    st.rerun()
+                            with col3:
+                                if st.button("🗑️", key=f"media_delete_{media_file}"):
+                                    await asyncio.to_thread(os.remove, media_file)
+                                    st.rerun()
+
+        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()
+
+        st.sidebar.subheader("Chat History 📜")
+        with open(HISTORY_FILE, 'r') as f:
+            history_content = f.read()
+        st.sidebar.markdown(history_content)
+
+    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()