Update app.py

app.py

@@ -17,17 +17,9 @@ from audio_recorder_streamlit import audio_recorder
 import nest_asyncio
 import re
 import pytz
-import anthropic
-import openai
-from PyPDF2 import PdfReader
-import threading
-import json
-import zipfile
 from gradio_client import Client
-from dotenv import load_dotenv
 from streamlit_marquee import streamlit_marquee
 from collections import defaultdict, Counter
-import pandas as pd
 
 # Patch asyncio for nesting
 nest_asyncio.apply()
@@ -65,12 +57,6 @@ AUDIO_DIR = "audio_logs"
 STATE_FILE = "user_state.txt"
 CHAT_FILE = os.path.join(CHAT_DIR, "sim_chat.md")
 
-# API Keys
-load_dotenv()
-anthropic_key = os.getenv('ANTHROPIC_API_KEY', st.secrets.get('ANTHROPIC_API_KEY', ""))
-openai_api_key = os.getenv('OPENAI_API_KEY', st.secrets.get('OPENAI_API_KEY', ""))
-openai_client = openai.OpenAI(api_key=openai_api_key)
-
 # Timestamp Helper
 def format_timestamp_prefix(username=""):
     central = pytz.timezone('US/Central')
@@ -86,7 +72,7 @@ def init_session_state():
         'chat_history': [], 'marquee_settings': {
             "background": "#1E1E1E", "color": "#FFFFFF", "font-size": "14px",
             "animationDuration": "20s", "width": "100%", "lineHeight": "35px"
-        }, 'username': None, 'autosend': True, 'autosearch': True, 'last_message': "",
+        }, 'username': None, 'autosend': True, 'last_message': "",
         'mp3_files': {}, 'timer_start': time.time(), 'auto_refresh': True, 'refresh_rate': 10
     }
     for k, v in defaults.items():
@@ -95,10 +81,10 @@ def init_session_state():
 
 # Marquee Helpers
 def update_marquee_settings_ui():
-    st.sidebar.markdown("### 🎯 Marquee Settings")
+    st.sidebar.markdown("### 🎯 Marquee Settings 🎨")
     cols = st.sidebar.columns(2)
     with cols[0]:
-        st.session_state['marquee_settings']['background'] = st.color_picker("🎨 Background", "#1E1E1E")
+        st.session_state['marquee_settings']['background'] = st.color_picker("🌈 Background", "#1E1E1E")
         st.session_state['marquee_settings']['color'] = st.color_picker("✍️ Text", "#FFFFFF")
     with cols[1]:
         st.session_state['marquee_settings']['font-size'] = f"{st.slider('📏 Size', 10, 24, 14)}px"
@@ -131,7 +117,7 @@ def get_download_link(file, file_type="mp3"):
     with open(file, "rb") as f:
         b64 = base64.b64encode(f.read()).decode()
     mime_types = {"mp3": "audio/mpeg", "png": "image/png", "md": "text/markdown"}
-    return f'<a href="data:{mime_types.get(file_type, "application/octet-stream")};base64,{b64}" download="{os.path.basename(file)}">{FILE_EMOJIS.get(file_type, "Download")} Download {os.path.basename(file)}</a>'
+    return f'<a href="data:{mime_types.get(file_type, "application/octet-stream")};base64,{b64}" download="{os.path.basename(file)}">{FILE_EMOJIS.get(file_type, "📥")} Download {os.path.basename(file)}</a>'
 
 def save_username(username):
     with open(STATE_FILE, 'w') as f:
@@ -143,6 +129,13 @@ def load_username():
             return f.read().strip()
     return None
 
+def load_mp3_viewer():
+    mp3_files = sorted(glob.glob("*.mp3"), key=os.path.getmtime)
+    for i, mp3 in enumerate(mp3_files, 1):
+        filename = os.path.basename(mp3)
+        if filename not in st.session_state['mp3_files']:
+            st.session_state['mp3_files'][filename] = (i, mp3)
+
 # Audio Processing
 async def async_edge_tts_generate(text, voice, username):
     cache_key = f"{text[:100]}_{voice}"
@@ -187,35 +180,13 @@ async def load_chat():
     lines = content.split('\n')
     return list(dict.fromkeys(line for line in lines if line.strip()))
 
-# Claude and ArXiv Search
-async def perform_claude_search(query, username):
-    client = anthropic.Anthropic(api_key=anthropic_key)
-    response = client.messages.create(
-        model="claude-3-sonnet-20240229",
-        max_tokens=1000,
-        messages=[{"role": "user", "content": query}]
-    )
-    result = response.content[0].text
-    voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
-    full_text = f"Claude says: {result}"
-    md_file, audio_file = await save_chat_entry(username, full_text, voice, True)
-    return md_file, audio_file, result
-
-async def perform_arxiv_search(query, username, claude_result=None):
-    if claude_result is None:
-        client = anthropic.Anthropic(api_key=anthropic_key)
-        claude_response = client.messages.create(
-            model="claude-3-sonnet-20240229",
-            max_tokens=1000,
-            messages=[{"role": "user", "content": query}]
-        )
-        claude_result = claude_response.content[0].text
-    enhanced_query = f"{query}\n\n{claude_result}"
+# ArXiv Search with Gradio Client
+async def perform_arxiv_search(query, username):
     gradio_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
     refs = gradio_client.predict(
-        enhanced_query, 10, "Semantic Search", "mistralai/Mixtral-8x7B-Instruct-v0.1", api_name="/update_with_rag_md"
+        query, 10, "Semantic Search", "mistralai/Mixtral-8x7B-Instruct-v0.1", api_name="/update_with_rag_md"
     )[0]
-    result = f"🔎 ArXiv Results:\n\n{refs}"
+    result = f"🔎 ArXiv Insights:\n\n{refs}"
     voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
     md_file, audio_file = await save_chat_entry(username, result, voice, True)
     return md_file, audio_file
@@ -235,6 +206,7 @@ async def websocket_handler(websocket, path):
                 username, content = message.split('|', 1)
                 voice = FUN_USERNAMES.get(username, "en-US-AriaNeural")
                 await save_chat_entry(username, content, voice)
+                await perform_arxiv_search(content, username)  # ArXiv response for every chat input
     except websockets.ConnectionClosed:
         await save_chat_entry("System 🌟", f"{username} has left {START_ROOM}!", "en-US-AriaNeural")
     finally:
@@ -267,21 +239,22 @@ def start_websocket_server():
 # Sim-Oriented Mad Libs
 def generate_mad_libs_story(username, inputs):
     story = f"""
-    In the wilds of Colorado, {username} stumbled upon {inputs['plural_noun']}. 
-    These {inputs['adjective1']} critters were {inputs['verb_ing']} all over the Rockies, 
-    freaking out climbers like {inputs['celebrity']}. 
-    Holding a {inputs['object']}, {username} felt {inputs['emotion']} as a {inputs['animal']} 
-    {inputs['verb_past']} past, yelling {inputs['exclamation']} near {inputs['place']}. 
-    Later, munching {inputs['food']} with a {inputs['adjective2']} {inputs['body_part']}, 
-    {username} looped “{inputs['song']}” {inputs['number']} times, 
-    deciding to {inputs['verb_present']} as a {inputs['occupation']} 
-    in a {inputs['color']} {inputs['noun']}—all ending with a {inputs['sound']}!
+    🌄 In the wilds of Colorado, {username} stumbled upon {inputs['plural_noun']}. 
+    🌟 These {inputs['adjective1']} critters were {inputs['verb_ing']} all over the Rockies, 
+    🎤 freaking out climbers like {inputs['celebrity']}. 
+    🛠️ Holding a {inputs['object']}, {username} felt {inputs['emotion']} as a {inputs['animal']} 
+    ⏳ {inputs['verb_past']} past, yelling {inputs['exclamation']} near {inputs['place']}. 
+    🍽️ Later, munching {inputs['food']} with a {inputs['adjective2']} {inputs['body_part']}, 
+    🎵 {username} looped “{inputs['song']}” {inputs['number']} times, 
+    💼 deciding to {inputs['verb_present']} as a {inputs['occupation']} 
+    🎨 in a {inputs['color']} {inputs['noun']}—all ending with a {inputs['sound']}!
     """
     return story.strip()
 
 # Main Interface
 def main():
     init_session_state()
+    load_mp3_viewer()
     saved_username = load_username()
     if saved_username and saved_username in FUN_USERNAMES:
         st.session_state.username = saved_username
@@ -291,88 +264,73 @@ def main():
         asyncio.run(save_chat_entry("System 🌟", f"{st.session_state.username} has joined {START_ROOM}!", "en-US-AriaNeural"))
     save_username(st.session_state.username)
 
-    st.title(f"{Site_Name} - {st.session_state.username}")
-    st.subheader("Chat, Search, or Create a Colorado Sim Story! 😂🌲")
+    st.title(f"🎉 {Site_Name} - {st.session_state.username} 🏔️")
+    st.subheader("🌲 Chat, Explore Audio, or Create a Sim Story! 🎤")
     update_marquee_settings_ui()
     chat_text = " ".join([line.split(": ")[-1] for line in asyncio.run(load_chat()) if ": " in line])
-    display_marquee(f"🏔️ {START_ROOM} | {st.session_state.username} | {chat_text}", st.session_state['marquee_settings'], "welcome")
+    display_marquee(f"🏔️ {START_ROOM} | 🎙️ {st.session_state.username} | 💬 {chat_text}", st.session_state['marquee_settings'], "welcome")
 
-    tab_main = st.radio("Action:", ["🎤 Chat & Voice", "🔍 Research", "✨ Mad Libs Sim"], horizontal=True)
-    st.checkbox("Autosend Chat", key="autosend")
-    st.checkbox("Autosearch Research", key="autosearch")
+    tab_main = st.radio("🎮 Action:", ["🎤 Chat & Voice", "🎵 Audio Gallery", "✨ Mad Libs Sim"], horizontal=True)
+    st.checkbox("🚀 Autosend Chat", key="autosend")
 
     if tab_main == "🎤 Chat & Voice":
-        st.subheader(f"{START_ROOM} Chat 💬")
+        st.subheader(f"💬 {START_ROOM} Chat 🌟")
         chat_content = asyncio.run(load_chat())
         with st.container():
             st.code("\n".join(f"{i+1}. {line}" for i, line in enumerate(chat_content)), language="python")
 
-        message = st.text_input(f"Message as {st.session_state.username}", key="message_input")
+        message = st.text_input(f"💬 Message as {st.session_state.username}", key="message_input", placeholder="Type your wild thoughts! 🌈")
         if message and message != st.session_state.last_message:
             st.session_state.last_message = message
-            col_send, col_claude, col_arxiv = st.columns([1, 1, 1])
+            col_send = st.columns(1)[0]
             with col_send:
-                if st.session_state.autosend or st.button("Send 🚀"):
+                if st.session_state.autosend or st.button("🚀 Send Message 🎙️"):
                     voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
                     md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, message, voice, True))
                     if audio_file:
                         play_and_download_audio(audio_file)
                     st.rerun()
-            with col_claude:
-                if st.button("🧠 Claude"):
-                    md_file, audio_file, _ = asyncio.run(perform_claude_search(message, st.session_state.username))
-                    if audio_file:
-                        play_and_download_audio(audio_file)
-                    st.rerun()
-            with col_arxiv:
-                if st.button("🔍 ArXiv"):
-                    md_file, audio_file = asyncio.run(perform_arxiv_search(message, st.session_state.username))
-                    if audio_file:
-                        play_and_download_audio(audio_file)
-                    st.rerun()
 
-    elif tab_main == "🔍 Research":
-        st.subheader("🔍 Research with Claude & ArXiv")
-        q = st.text_input("Query:", key="research_query")
-        if q and (st.session_state.autosearch or st.button("🔍 Run")):
-            st.session_state.last_message = q
-            md_file_claude, audio_file_claude, claude_result = asyncio.run(perform_claude_search(q, st.session_state.username))
-            if audio_file_claude:
-                play_and_download_audio(audio_file_claude)
-            md_file_arxiv, audio_file_arxiv = asyncio.run(perform_arxiv_search(q, st.session_state.username, claude_result))
-            if audio_file_arxiv:
-                play_and_download_audio(audio_file_arxiv)
+    elif tab_main == "🎵 Audio Gallery":
+        st.subheader("🎵 Audio Gallery 🎧")
+        mp3_files = sorted(glob.glob("*.mp3"), key=os.path.getmtime)
+        if mp3_files:
+            for i, mp3 in enumerate(mp3_files):
+                with st.expander(f"🎵 {i+1}. {os.path.basename(mp3)}"):
+                    play_and_download_audio(mp3)
+        else:
+            st.write("🎶 No audio files yet—chat or create a story to generate some! 🎤")
 
     elif tab_main == "✨ Mad Libs Sim":
-        st.subheader("Create Your Colorado Sim Story!")
+        st.subheader("✨ Create Your Colorado Sim Story! 🎉")
         col1, col2 = st.columns(2)
         inputs = {}
         with col1:
-            inputs['plural_noun'] = st.text_input("1. Plural Noun", placeholder="e.g., 'tacos'")
-            inputs['adjective1'] = st.text_input("2. Adjective", placeholder="e.g., 'spicy'")
-            inputs['verb_ing'] = st.text_input("3. Verb ending in -ing", placeholder="e.g., 'yeeting'")
-            inputs['celebrity'] = st.text_input("4. Celebrity Name", placeholder="e.g., 'Elon Musk'")
-            inputs['object'] = st.text_input("5. Random Object", placeholder="e.g., 'toaster'")
-            inputs['emotion'] = st.text_input("6. Emotion", placeholder="e.g., 'salty'")
-            inputs['animal'] = st.text_input("7. Animal", placeholder="e.g., 'elk'")
-            inputs['verb_past'] = st.text_input("8. Verb (past tense)", placeholder="e.g., 'yeeted'")
-            inputs['place'] = st.text_input("9. Place", placeholder="e.g., 'Rockies'")
-            inputs['exclamation'] = st.text_input("10. Exclamation", placeholder="e.g., 'Bruh!'")
+            inputs['plural_noun'] = st.text_input("🌟 Plural Noun", placeholder="e.g., 'tacos'", key="plural_noun")
+            inputs['adjective1'] = st.text_input("🎨 Adjective", placeholder="e.g., 'spicy'", key="adjective1")
+            inputs['verb_ing'] = st.text_input("⏳ Verb ending in -ing", placeholder="e.g., 'yeeting'", key="verb_ing")
+            inputs['celebrity'] = st.text_input("🎤 Celebrity Name", placeholder="e.g., 'Elon Musk'", key="celebrity")
+            inputs['object'] = st.text_input("🛠️ Random Object", placeholder="e.g., 'toaster'", key="object")
+            inputs['emotion'] = st.text_input("❤️ Emotion", placeholder="e.g., 'salty'", key="emotion")
+            inputs['animal'] = st.text_input("🦌 Animal", placeholder="e.g., 'elk'", key="animal")
+            inputs['verb_past'] = st.text_input("⏮️ Verb (past tense)", placeholder="e.g., 'yeeted'", key="verb_past")
+            inputs['place'] = st.text_input("🌍 Place", placeholder="e.g., 'Rockies'", key="place")
+            inputs['exclamation'] = st.text_input("❗ Exclamation", placeholder="e.g., 'Bruh!'", key="exclamation")
         with col2:
-            inputs['food'] = st.text_input("11. Food Item", placeholder="e.g., 'pizza'")
-            inputs['adjective2'] = st.text_input("12. Adjective", placeholder="e.g., 'cringe'")
-            inputs['body_part'] = st.text_input("13. Body Part", placeholder="e.g., 'elbow'")
-            inputs['number'] = st.text_input("14. Number", placeholder="e.g., '69'")
-            inputs['song'] = st.text_input("15. Song Title", placeholder="e.g., 'Sweet Caroline'")
-            inputs['verb_present'] = st.text_input("16. Verb (present tense)", placeholder="e.g., 'slay'")
-            inputs['occupation'] = st.text_input("17. Occupation", placeholder="e.g., 'influencer'")
-            inputs['color'] = st.text_input("18. Color", placeholder="e.g., 'teal'")
-            inputs['noun'] = st.text_input("19. Noun", placeholder="e.g., 'chaos'")
-            inputs['sound'] = st.text_input("20. Silly Sound Effect", placeholder="e.g., 'Boop!'")
-
-        if st.button("Generate Sim Story! 🎉"):
+            inputs['food'] = st.text_input("🍽️ Food Item", placeholder="e.g., 'pizza'", key="food")
+            inputs['adjective2'] = st.text_input("✨ Adjective", placeholder="e.g., 'cringe'", key="adjective2")
+            inputs['body_part'] = st.text_input("🦴 Body Part", placeholder="e.g., 'elbow'", key="body_part")
+            inputs['number'] = st.text_input("🔢 Number", placeholder="e.g., '69'", key="number")
+            inputs['song'] = st.text_input("🎵 Song Title", placeholder="e.g., 'Sweet Caroline'", key="song")
+            inputs['verb_present'] = st.text_input("▶️ Verb (present tense)", placeholder="e.g., 'slay'", key="verb_present")
+            inputs['occupation'] = st.text_input("💼 Occupation", placeholder="e.g., 'influencer'", key="occupation")
+            inputs['color'] = st.text_input("🌈 Color", placeholder="e.g., 'teal'", key="color")
+            inputs['noun'] = st.text_input("📋 Noun", placeholder="e.g., 'chaos'", key="noun")
+            inputs['sound'] = st.text_input("🔊 Silly Sound Effect", placeholder="e.g., 'Boop!'", key="sound")
+
+        if st.button("🎉 Generate Sim Story! 🎤"):
             story = generate_mad_libs_story(st.session_state.username, inputs)
-            st.markdown("### Your Colorado Sim Adventure 🏔️")
+            st.markdown("### 🌄 Your Colorado Sim Adventure 🏔️")
             st.write(story)
             voice = FUN_USERNAMES.get(st.session_state.username, "en-US-AriaNeural")
             md_file, audio_file = asyncio.run(save_chat_entry(st.session_state.username, story, voice, True))
@@ -380,8 +338,8 @@ def main():
                 play_and_download_audio(audio_file)
 
     # Sidebar
-    st.sidebar.subheader("Character Settings")
-    new_username = st.sidebar.selectbox("Switch Character", list(FUN_USERNAMES.keys()), index=list(FUN_USERNAMES.keys()).index(st.session_state.username))
+    st.sidebar.subheader("🎭 Character Settings")
+    new_username = st.sidebar.selectbox("🔄 Switch Character", list(FUN_USERNAMES.keys()), index=list(FUN_USERNAMES.keys()).index(st.session_state.username))
     if new_username != st.session_state.username:
         asyncio.run(save_chat_entry("System 🌟", f"{st.session_state.username} switched to {new_username}", "en-US-AriaNeural"))
         st.session_state.username = new_username
@@ -389,14 +347,5 @@ def main():
         save_username(st.session_state.username)
         st.rerun()
 
-    st.sidebar.subheader("Chat History")
-    chat_content = asyncio.run(load_chat())
-    with st.sidebar.expander("Recent Logs"):
-        st.code("\n".join(f"{i+1}. {line}" for i, line in enumerate(chat_content[-10:])), language="python")
-
-    if not st.session_state.get('server_running', False):
-        st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)
-        st.session_state.server_task.start()
-
-if __name__ == "__main__":
-    main()
+    st.sidebar.subheader("💬 Chat History")
+    chat_content = asyncio