Update app.py

app.py

@@ -18,6 +18,8 @@ from audio_recorder_streamlit import audio_recorder
 import nest_asyncio
 import re
 from streamlit_paste_button import paste_image_button
+import pytz
+import shutil
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -140,7 +142,8 @@ if 'image_hashes' not in st.session_state:
 
 # Timestamp wizardry - clock ticks with flair! ⏰🎩
 def format_timestamp_prefix(username):
-    now = datetime.now()
+    central = pytz.timezone('US/Central')
+    now = datetime.now(central)
     return f"{now.strftime('%I-%M-%p-ct-%m-%d-%Y')}-by-{username}"
 
 # Compute image hash from binary data
@@ -172,8 +175,9 @@ def log_action(username, action):
     user_log = {k: v for k, v in user_log.items() if current_time - v < 10}
     st.session_state.action_log[username] = user_log
     if action not in user_log:
+        central = pytz.timezone('US/Central')
         with open(HISTORY_FILE, 'a') as f:
-            f.write(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: {action}\n")
+            f.write(f"[{datetime.now(central).strftime('%Y-%m-%d %H:%M:%S')}] {username}: {action}\n")
         user_log[action] = current_time
 
 # Clean text - strip the fancy stuff! 🧹📝
@@ -185,13 +189,14 @@ def clean_text_for_tts(text):
 # Chat saver - words locked tight! 💬🔒
 async def save_chat_entry(username, message, is_markdown=False):
     await asyncio.to_thread(log_action, username, "💬🔒 - Chat saver - words locked tight!")
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    central = pytz.timezone('US/Central')
+    timestamp = datetime.now(central).strftime("%Y-%m-%d %H:%M:%S")
     if is_markdown:
         entry = f"[{timestamp}] {username}:\n```markdown\n{message}\n```"
     else:
         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")  # Use username-specific voice
+    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)
     if audio_file:
@@ -199,6 +204,7 @@ async def save_chat_entry(username, message, is_markdown=False):
             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()
+    return audio_file
 
 # Chat loader - history unleashed! 📜🚀
 async def load_chat():
@@ -238,7 +244,8 @@ async def get_message_suggestions(chat_content, prefix):
 # Vote saver - cheers recorded! 👍📊
 async def save_vote(file, item, user_hash, username, comment=""):
     await asyncio.to_thread(log_action, username, "👍📊 - Vote saver - cheers recorded!")
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    central = pytz.timezone('US/Central')
+    timestamp = datetime.now(central).strftime("%Y-%m-%d %H:%M:%S")
     entry = f"[{timestamp}] {user_hash} voted for {item}"
     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"))
@@ -289,7 +296,8 @@ async def async_edge_tts_generate(text, voice, rate=0, pitch=0, file_format="mp3
         return filepath if os.path.exists(filepath) 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")
+            central = pytz.timezone('US/Central')
+            f.write(f"[{datetime.now(central).strftime('%Y-%m-%d %H:%M:%S')}] {username}: Audio failed - No audio received for '{text}'\n")
         return None
 
 # Audio player - tunes blast off! 🔊🚀
@@ -397,6 +405,40 @@ async def perform_ai_lookup(query, vocal_summary=True, extended_refs=False, titl
         if audio_file:
             st.audio(audio_file)
 
+# Delete all user files function
+def delete_user_files():
+    protected_files = {'app.py', 'requirements.txt', 'README.md'}
+    deleted_files = []
+    
+    # Directories to clear
+    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}")
+            # Remove empty directories
+            try:
+                shutil.rmtree(directory, ignore_errors=True)
+                os.makedirs(directory, exist_ok=True)  # Recreate empty directory
+            except Exception as e:
+                st.error(f"Failed to remove directory {directory}: {e}")
+    
+    # Clear session state caches
+    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>
@@ -574,7 +616,7 @@ 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()))
             st.session_state.voice = FUN_USERNAMES[st.session_state.username]
-            st.markdown(f"**🎙️ Voice Selected**: {st.session_state.voice} 🗣️")
+            st.markdown(f"**🎙️ Voice Selected**: {st.session_state.voice} 🗣️ for {st.session_state.username}")
 
         st.title(f"🤖🧠MMO {st.session_state.username}📝🔬")
         st.markdown(f"Welcome to {START_ROOM} - chat, vote, upload, paste images, and enjoy quoting! 🎉")
@@ -671,6 +713,7 @@ def main():
             if paste_result_quote.image_data is not None:
                 if isinstance(paste_result_quote.image_data, str):
                     st.session_state.message_text = paste_result_quote.image_data
+                    st.text_area("Add your response", key="quote_response", value=st.session_state.message_text)
                 else:
                     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)
@@ -687,19 +730,21 @@ def main():
                 st.session_state.message_text = ''
                 st.rerun()
 
-        new_username = st.selectbox("Change Name", [""] + list(FUN_USERNAMES.keys()), index=0)
+        current_selection = st.session_state.username if st.session_state.username in FUN_USERNAMES else ""
+        new_username = st.selectbox("Change Name and Voice", [""] + list(FUN_USERNAMES.keys()), index=(list(FUN_USERNAMES.keys()).index(current_selection) + 1 if current_selection else 0), format_func=lambda x: f"{x} ({FUN_USERNAMES.get(x, 'No Voice')})" if x else "Select a name")
         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}")
             st.session_state.username = new_username
             st.session_state.voice = FUN_USERNAMES[new_username]
-            st.markdown(f"**🎙️ Voice Changed**: {st.session_state.voice} 🗣️")
+            st.markdown(f"**🎙️ Voice Changed**: {st.session_state.voice} 🗣️ for {st.session_state.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))
+        message = st.text_input(f"Message as {st.session_state.username} (Voice: {st.session_state.voice})", key="message_input", value=st.session_state.message_text)
         paste_result_msg = paste_image_button("📋 Paste Image or Text with Message", key="paste_button_msg")
         if paste_result_msg.image_data is not None:
             if isinstance(paste_result_msg.image_data, str):
                 st.session_state.message_text = paste_result_msg.image_data
+                st.text_input(f"Message as {st.session_state.username} (Voice: {st.session_state.voice})", key="message_input_paste", value=st.session_state.message_text)
             else:
                 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)
@@ -707,7 +752,9 @@ def main():
                     st.session_state.pasted_image_data = filename
         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, is_markdown=True)
+                audio_file = await save_chat_entry(st.session_state.username, message, is_markdown=True)
+                if audio_file:
+                    st.session_state.audio_cache[f"{message}_{FUN_USERNAMES[st.session_state.username]}"] = audio_file
             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
@@ -735,6 +782,20 @@ def main():
                 if file_path.endswith('.mp4'):
                     st.session_state.media_notifications.append(file_path)
 
+        # Big Red Delete Button
+        st.subheader("🛑 Danger Zone")
+        if st.button("Try Not To Delete It All On Your First Day", key="delete_all", help="Deletes all user-added files!", type="primary", use_container_width=True):
+            deleted_files = delete_user_files()
+            if deleted_files:
+                st.markdown("### 🗑️ Deleted Files:\n" + "\n".join([f"- `{file}`" for file in deleted_files]))
+            else:
+                st.markdown("### 🗑️ Nothing to Delete!")
+            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()
+            st.rerun()
+
         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")
         if media_files: