Update app.py

app.py

@@ -17,6 +17,7 @@ import streamlit.components.v1 as components
 import edge_tts
 from audio_recorder_streamlit import audio_recorder
 import nest_asyncio
+import re
 
 # Patch for nested async - sneaky fix! 🐍✨
 nest_asyncio.apply()
@@ -134,14 +135,24 @@ 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):
+    # Remove Markdown formatting (e.g., #, *, [], ![])
+    cleaned = re.sub(r'[#*!\[\]]+', '', text)
+    # Replace newlines with spaces and strip extra whitespace
+    cleaned = ' '.join(cleaned.split())
+    # Ensure some text exists, max 200 chars to avoid edgeTTS limits
+    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!")
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     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 mapped voice or default
-    audio_file = await async_edge_tts_generate(message, 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:
         with open(HISTORY_FILE, 'a') as f:
             f.write(f"[{timestamp}] {username}: Audio generated - {audio_file}\n")
@@ -229,8 +240,13 @@ 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")
-    await communicate.save(filename)
-    return filename if os.path.exists(filename) else None
+    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
 
 # Audio player - tunes blast off! 🔊🚀
 def play_and_download_audio(file_path):
@@ -385,7 +401,8 @@ def create_streamlit_interface():
                     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"))
+                        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)
@@ -418,7 +435,14 @@ def create_streamlit_interface():
                         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)
+                    try:
+                        await save_chat_entry(st.session_state.username, markdown_response)
+                    except edge_tts.exceptions.NoAudioReceived:
+                        # Log failure but continue without audio
+                        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 = ''