Update app.py

app.py

@@ -14,12 +14,12 @@ import base64
 import io
 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
 import pytz
 import shutil
+from urllib.parse import urlencode
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -60,23 +60,12 @@ FUN_USERNAMES = {
     "ChronoChimp 🐒": "en-GB-LibbyNeural"
 }
 
-# Folders galore - organizing chaos! 📂🌀
-CHAT_DIR = "chat_logs"
-VOTE_DIR = "vote_logs"
+# Top-level files (no subdirectories)
+CHAT_FILE = "global_chat.md"
+QUOTE_VOTES_FILE = "quote_votes.md"
+MEDIA_VOTES_FILE = "media_votes.md"
+HISTORY_FILE = "chat_history.md"
 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")
-MEDIA_VOTES_FILE = os.path.join(VOTE_DIR, "media_votes.md")
-HISTORY_FILE = os.path.join(HISTORY_DIR, "chat_history.md")
 
 # Fancy digits - numbers got style! 🔢💃
 UNICODE_DIGITS = {i: f"{i}\uFE0F⃣" for i in range(10)}
@@ -133,14 +122,14 @@ if 'refresh_rate' not in st.session_state:
     st.session_state.refresh_rate = 5
 if 'base64_cache' not in st.session_state:
     st.session_state.base64_cache = {}
-if 'transcript_history' not in st.session_state:
-    st.session_state.transcript_history = []
-if 'last_transcript' not in st.session_state:
-    st.session_state.last_transcript = ""
 if 'image_hashes' not in st.session_state:
     st.session_state.image_hashes = set()
 if 'gallery_columns' not in st.session_state:
     st.session_state.gallery_columns = 1  # Default gallery tiles
+if 'user_id' not in st.session_state:
+    st.session_state.user_id = None
+if 'user_hash' not in st.session_state:
+    st.session_state.user_hash = None
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix(username):
@@ -189,21 +178,43 @@ def clean_text_for_tts(text):
     return cleaned[:200] if cleaned else "No text to speak"
 
 # Chat saver - words locked tight! 💬🔒
-async def save_chat_entry(username, message, is_markdown=False):
+async def save_chat_entry(username, message, is_markdown=False, quote_line=None):
     await asyncio.to_thread(log_action, username, "💬🔒 - Chat saver - words locked tight!")
     central = pytz.timezone('US/Central')
     timestamp = datetime.now(central).strftime("%Y-%m-%d %H:%M:%S")
+    user_history_file = f"{username}_history.md"
+    voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
+    indent = "  " if quote_line else ""  # Nesting for replies
     if is_markdown:
-        entry = f"[{timestamp}] {username}:\n```markdown\n{message}\n```"
+        entry = f"{indent}[{timestamp}] {username}:\n{indent}```markdown\n{indent}{message}\n{indent}```"
     else:
-        entry = f"[{timestamp}] {username}: {message}"
+        entry = f"{indent}[{timestamp}] {username}: {message}"
+    if quote_line:
+        entry = f"{indent}> {quote_line}\n{entry}"
+
+    # Save to global chat file
     await asyncio.to_thread(lambda: open(CHAT_FILE, 'a').write(f"{entry}\n"))
-    voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
+
+    # Save to user-specific history file
+    if not os.path.exists(user_history_file):
+        with open(user_history_file, 'w') as f:
+            f.write(f"# Chat History for {username} (Voice: {voice})\n\n")
+    with open(user_history_file, 'a') as f:
+        f.write(f"{entry}\n")
+
     cleaned_message = clean_text_for_tts(message)
     audio_file = await async_edge_tts_generate(cleaned_message, voice)
     if audio_file:
         with open(HISTORY_FILE, 'a') as f:
             f.write(f"[{timestamp}] {username} ({voice}): Audio generated - {audio_file}\n")
+        with open(user_history_file, 'a') as f:
+            f.write(f"{indent}[{timestamp}] Audio: {audio_file}\n")
+        
+        # Embed audio in global chat if it's an image or media reference
+        if message.startswith("Pasted image:") or message.startswith("Uploaded media:"):
+            with open(CHAT_FILE, 'a') as f:
+                f.write(f"{indent}[{timestamp}] Audio: {audio_file}\n")
+
     await broadcast_message(f"{username}|{message}", "chat")
     st.session_state.last_chat_update = time.time()
     return audio_file
@@ -212,15 +223,15 @@ async def save_chat_entry(username, message, is_markdown=False):
 async def save_chat_history_with_image(username, image_path):
     central = pytz.timezone('US/Central')
     timestamp = datetime.now(central).strftime("%Y-%m-%d_%H-%M-%S")
-    history_filename = f"chat_history_{timestamp}-by-{username}.md"
-    history_filepath = os.path.join(HISTORY_DIR, history_filename)
+    user_history_file = f"{username}_history.md"
     chat_content = await load_chat()
     voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
-    with open(history_filepath, 'w') as f:
-        f.write(f"# Chat History at {timestamp} by {username} (Voice: {voice})\n\n")
-        f.write(f"## Image Shared: {os.path.basename(image_path)}\n")
-        f.write(chat_content)
-    return history_filepath
+    if not os.path.exists(user_history_file):
+        with open(user_history_file, 'w') as f:
+            f.write(f"# Chat History for {username} (Voice: {voice})\n\n")
+    with open(user_history_file, 'a') as f:
+        f.write(f"[{timestamp}] {username} (Voice: {voice}) Shared Image: {os.path.basename(image_path)}\n")
+        f.write(f"```markdown\n{chat_content}\n```\n")
 
 # Chat loader - history unleashed! 📜🚀
 async def load_chat():
@@ -305,7 +316,7 @@ async def async_edge_tts_generate(text, voice, rate=0, pitch=0, file_format="mp3
     await asyncio.to_thread(log_action, username, "🎶🌟 - Audio maker - voices come alive!")
     timestamp = format_timestamp_prefix(username)
     filename = f"{timestamp}.{file_format}"
-    filepath = os.path.join(AUDIO_DIR, filename)
+    filepath = filename  # Top-level file
     communicate = edge_tts.Communicate(text, voice, rate=f"{rate:+d}%", pitch=f"{pitch:+d}Hz")
     try:
         await communicate.save(filepath)
@@ -336,7 +347,7 @@ async def save_pasted_image(image, username):
         return None
     timestamp = format_timestamp_prefix(username)
     filename = f"{timestamp}-{img_hash}.png"
-    filepath = os.path.join(MEDIA_DIR, filename)
+    filepath = filename  # Top-level file
     await asyncio.to_thread(image.save, filepath, "PNG")
     st.session_state.image_hashes.add(img_hash)
     await save_chat_history_with_image(username, filepath)
@@ -405,14 +416,6 @@ async def run_websocket_server():
         st.session_state.server_running = True
         await server.wait_closed()
 
-# Voice processor - speech to text! 🎤📝
-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"
-        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 🌟')
@@ -425,194 +428,28 @@ async def perform_ai_lookup(query, vocal_summary=True, extended_refs=False, titl
 
 # Delete all user files function
 def delete_user_files():
-    protected_files = {'app.py', 'requirements.txt', 'README.md'}
+    protected_files = {'app.py', 'requirements.txt', 'README.md', CHAT_FILE, QUOTE_VOTES_FILE, MEDIA_VOTES_FILE, HISTORY_FILE, STATE_FILE}
     deleted_files = []
-    directories = [MEDIA_DIR, AUDIO_DIR, CHAT_DIR, VOTE_DIR, HISTORY_DIR]
-    for directory in directories:
-        if os.path.exists(directory):
-            for root, _, files in os.walk(directory):
-                for file in files:
-                    file_path = os.path.join(root, file)
-                    if os.path.basename(file_path) not in protected_files:
-                        try:
-                            os.remove(file_path)
-                            deleted_files.append(file_path)
-                        except Exception as e:
-                            st.error(f"Failed to delete {file_path}: {e}")
+    for file in os.listdir('.'):
+        if file not in protected_files and not file.endswith('_history.md'):
             try:
-                shutil.rmtree(directory, ignore_errors=True)
-                os.makedirs(directory, exist_ok=True)
+                os.remove(file)
+                deleted_files.append(file)
             except Exception as e:
-                st.error(f"Failed to remove directory {directory}: {e}")
+                st.error(f"Failed to delete {file}: {e}")
     st.session_state.image_hashes.clear()
     st.session_state.audio_cache.clear()
     st.session_state.base64_cache.clear()
     st.session_state.displayed_chat_lines.clear()
     return deleted_files
 
-# ASR Component HTML
-ASR_HTML = """
-<html>
-<head>
-    <title>Continuous Speech Demo</title>
-    <style>
-        body { 
-            font-family: sans-serif; 
-            padding: 20px; 
-            max-width: 800px;
-            margin: 0 auto;
-        }
-        button { 
-            padding: 10px 20px; 
-            margin: 10px 5px;
-            font-size: 16px;
-        }
-        #status { 
-            margin: 10px 0;
-            padding: 10px;
-            background: #e8f5e9;
-            border-radius: 4px;
-        }
-        #output {
-            white-space: pre-wrap;
-            padding: 15px;
-            background: #f5f5f5;
-            border-radius: 4px;
-            margin: 10px 0;
-            min-height: 100px;
-            max-height: 400px;
-            overflow-y: auto;
-        }
-        .controls {
-            margin: 10px 0;
-        }
-    </style>
-</head>
-<body>
-    <div class="controls">
-        <button id="start">Start Listening</button>
-        <button id="stop" disabled>Stop Listening</button>
-        <button id="clear">Clear Text</button>
-    </div>
-    <div id="status">Ready</div>
-    <div id="output"></div>
-    
-    <script>
-        if (!('webkitSpeechRecognition' in window)) {
-            alert('Speech recognition not supported');
-        } else {
-            const recognition = new webkitSpeechRecognition();
-            const startButton = document.getElementById('start');
-            const stopButton = document.getElementById('stop');
-            const clearButton = document.getElementById('clear');
-            const status = document.getElementById('status');
-            const output = document.getElementById('output');
-            let fullTranscript = '';
-            let lastUpdateTime = Date.now();
-
-            recognition.continuous = true;
-            recognition.interimResults = true;
-
-            const startRecognition = () => {
-                try {
-                    recognition.start();
-                    status.textContent = 'Listening...';
-                    startButton.disabled = true;
-                    stopButton.disabled = false;
-                } catch (e) {
-                    console.error(e);
-                    status.textContent = 'Error: ' + e.message;
-                }
-            };
-
-            window.addEventListener('load', () => {
-                setTimeout(startRecognition, 1000);
-            });
-
-            startButton.onclick = startRecognition;
-
-            stopButton.onclick = () => {
-                recognition.stop();
-                status.textContent = 'Stopped';
-                startButton.disabled = false;
-                stopButton.disabled = true;
-            };
-
-            clearButton.onclick = () => {
-                fullTranscript = '';
-                output.textContent = '';
-                sendDataToPython({value: '', dataType: "json"});
-            };
-
-            recognition.onresult = (event) => {
-                let interimTranscript = '';
-                let finalTranscript = '';
-
-                for (let i = event.resultIndex; i < event.results.length; i++) {
-                    const transcript = event.results[i][0].transcript;
-                    if (event.results[i].isFinal) {
-                        finalTranscript += transcript + '\\n';
-                    } else {
-                        interimTranscript += transcript;
-                    }
-                }
-
-                if (finalTranscript || (Date.now() - lastUpdateTime > 5000)) {
-                    if (finalTranscript) {
-                        fullTranscript += finalTranscript;
-                    }
-                    lastUpdateTime = Date.now();
-                    output.textContent = fullTranscript + (interimTranscript ? '... ' + interimTranscript : '');
-                    output.scrollTop = output.scrollHeight;
-                    sendDataToPython({value: fullTranscript, dataType: "json"});
-                }
-            };
-
-            recognition.onend = () => {
-                if (!stopButton.disabled) {
-                    try {
-                        recognition.start();
-                        console.log('Restarted recognition');
-                    } catch (e) {
-                        console.error('Failed to restart recognition:', e);
-                        status.textContent = 'Error restarting: ' + e.message;
-                        startButton.disabled = false;
-                        stopButton.disabled = true;
-                    }
-                }
-            };
-
-            recognition.onerror = (event) => {
-                console.error('Recognition error:', event.error);
-                status.textContent = 'Error: ' + event.error;
-                if (event.error === 'not-allowed' || event.error === 'service-not-allowed') {
-                    startButton.disabled = false;
-                    stopButton.disabled = true;
-                }
-            };
-        }
-
-        function sendDataToPython(data) {
-            window.parent.postMessage({
-                isStreamlitMessage: true,
-                type: "streamlit:setComponentValue",
-                ...data
-            }, "*");
-        }
-
-        window.addEventListener('load', function() {
-            window.setTimeout(function() {
-                window.parent.postMessage({
-                    isStreamlitMessage: true,
-                    type: "streamlit:setFrameHeight",
-                    height: document.documentElement.clientHeight
-                }, "*");
-            }, 0);
-        });
-    </script>
-</body>
-</html>
-"""
+# Query parameter checker
+def check_query_params():
+    query_params = st.query_params if hasattr(st, 'query_params') else st.experimental_get_query_params()
+    user_id = query_params.get("q", [None])[0]
+    if user_id:
+        st.session_state.user_id = user_id
+    return user_id
 
 # Main execution - let’s roll! 🎲🚀
 def main():
@@ -622,6 +459,10 @@ def main():
     asyncio.set_event_loop(loop)
 
     async def async_interface():
+        # Generate user ID and hash if not set
+        if not st.session_state.user_id:
+            st.session_state.user_id = str(uuid.uuid4())
+            st.session_state.user_hash = await generate_user_hash()
         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'))]
@@ -629,33 +470,22 @@ def main():
             st.session_state.voice = FUN_USERNAMES[st.session_state.username]
             st.markdown(f"**🎙️ Voice Selected**: {st.session_state.voice} 🗣️ for {st.session_state.username}")
 
+        # Check query params and set user ID
+        check_query_params()
+        user_url = f"/q={st.session_state.user_id}"
+        st.write(f"Your unique URL path: [{user_url}]({user_url})")
+
         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! 🎉 User ID: {st.session_state.user_id}")
 
         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()
-
-        # Continuous Speech Input (ASR)
-        st.subheader("🎤 Continuous Speech Input")
-        asr_component = components.html(ASR_HTML, height=400)
-        if asr_component and isinstance(asr_component, dict) and 'value' in asr_component:
-            transcript = asr_component['value'].strip()
-            if transcript and transcript != st.session_state.last_transcript:
-                st.session_state.transcript_history.append(transcript)
-                await save_chat_entry(st.session_state.username, transcript, is_markdown=True)
-                st.session_state.last_transcript = transcript
-                st.rerun()
-
         # Unified Chat History at Top
         st.subheader(f"{START_ROOM} Chat History 💬")
         chat_content = await load_chat()
         chat_lines = chat_content.split('\n')
-        chat_lines = [line for line in chat_lines if line.strip() and ': ' in line and not line.startswith('#')]
+        chat_lines = [line for line in chat_content.split('\n') if line.strip() and not line.startswith('#')]
         if chat_lines:
             col1, col2 = st.columns([2, 1])
             with col1:
@@ -664,7 +494,7 @@ def main():
                     col_text, col_audio = st.columns([3, 1])
                     with col_text:
                         if "```markdown" in line:
-                            markdown_content = re.search(r'```markdown\n(.*?)```', line, re.DOTALL)
+                            markdown_content = re.search(r'```markdown\n(.*?)(?=\n\s*\[|$)', line, re.DOTALL)
                             if markdown_content:
                                 st.markdown(markdown_content.group(1))
                             else:
@@ -672,13 +502,17 @@ def main():
                         else:
                             st.markdown(line)
                     with col_audio:
-                        username = line.split(': ')[1].split(' ')[0]
+                        username = line.split(': ')[1].split(' ')[0] if ': ' in line else "Unknown"
                         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])
-                            voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
-                            audio_file = await async_edge_tts_generate(cleaned_text, voice)
-                            st.session_state.audio_cache[cache_key] = audio_file
+                        if cache_key not in st.session_state.audio_cache and "Audio:" in line:
+                            audio_ref = line.split("Audio: ")[-1].strip()
+                            if os.path.exists(audio_ref):
+                                st.session_state.audio_cache[cache_key] = audio_ref
+                            else:
+                                cleaned_text = clean_text_for_tts(line.split(': ', 1)[1].split("Audio:")[0])
+                                voice = st.session_state.voice if username == st.session_state.username else FUN_USERNAMES.get(username, "en-US-AriaNeural")
+                                audio_file = await async_edge_tts_generate(cleaned_text, voice)
+                                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)
@@ -696,13 +530,13 @@ def main():
                     filename = await save_pasted_image(paste_result_quote.image_data, st.session_state.username)
                     if filename:
                         st.session_state.pasted_image_data = filename
-            if st.button("Send Quote 🚀", key="send_quote"):
+            if st.button("Send Quote �<|control678|>", 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}")
+                    await save_chat_entry(st.session_state.username, f"Pasted image: {st.session_state.pasted_image_data}", quote_line=st.session_state.quote_line)
                     st.session_state.pasted_image_data = None
-                await save_chat_entry(st.session_state.username, markdown_response, is_markdown=True)
+                await save_chat_entry(st.session_state.username, markdown_response, is_markdown=True, quote_line=st.session_state.quote_line)
                 st.session_state.quote_line = None
                 st.session_state.message_text = ''
                 st.rerun()
@@ -744,7 +578,7 @@ def main():
                 if filename:
                     st.session_state.pasted_image_data = filename
 
-        tab_main = st.radio("Action:", ["🎤 Voice", "📸 Media", "🔍 ArXiv", "📝 Editor"], horizontal=True)
+        tab_main = st.radio("Action:", ["📸 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)
 
@@ -757,7 +591,7 @@ def main():
             file_hash = hashlib.md5(uploaded_file.getbuffer()).hexdigest()[:8]
             if file_hash not in st.session_state.image_hashes:
                 filename = f"{timestamp}-{file_hash}.{ext}"
-                file_path = os.path.join(MEDIA_DIR, filename)
+                file_path = filename  # Top-level file
                 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}")
@@ -791,7 +625,7 @@ def main():
         st.subheader("Media Gallery 🎨🎶🎥")
         gallery_columns = st.slider("Number of Gallery Tiles", 1, 20, st.session_state.gallery_columns)
         st.session_state.gallery_columns = gallery_columns
-        media_files = glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.mp4")
+        media_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.mp4")
         if media_files:
             media_votes = await load_votes(MEDIA_VOTES_FILE)
             seen_files = set()
@@ -807,7 +641,7 @@ def main():
                         if media_file.endswith(('.png', '.jpg')):
                             st.image(media_file, use_container_width=True)
                         elif media_file.endswith('.mp4'):
-                            st.markdown(get_video_html(media_file), unsafe_allow_html=True)  # Verbatim autoplay
+                            st.markdown(get_video_html(media_file), unsafe_allow_html=True)
                         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()