Update app.py

app.py

@@ -4,6 +4,8 @@ import edge_tts
 import time
 import os
 import uuid
+import re
+import html
 import firebase_admin
 from firebase_admin import credentials, firestore
 from openai import OpenAI
@@ -153,6 +155,22 @@ def display_chat_history():
     st.markdown('<div class="chat-history-wrapper">' + "".join(chat_msgs) + '</div>', unsafe_allow_html=True)
     st.markdown('<div id="chat-top-anchor"></div>', unsafe_allow_html=True)
 
+# --- TTS sanitize ---
+def sanitize_for_tts(text):
+    text = html.unescape(text)
+    text = re.sub(r'[^\x00-\x7F]+', ' ', text)
+    text = re.sub(r'\[([^\]]+)\]\([^\)]+\)', r'\1', text)
+    text = re.sub(r'(\*\*|__)(.*?)\1', r'\2', text)
+    text = re.sub(r'(\*|_)(.*?)\1', r'\2', text)
+    text = re.sub(r'^#{1,6}\s+', '', text, flags=re.MULTILINE)
+    text = re.sub(r'^\s*[-*+]\s+', ' • ', text, flags=re.MULTILINE)
+    text = re.sub(r'^\s*\d+\.\s+', ' • ', text, flags=re.MULTILINE)
+    text = re.sub(r'[!?]{2,}', '.', text)
+    text = re.sub(r'\.{3,}', '.', text)
+    text = re.sub(r'\n{2,}', '. ', text)
+    text = re.sub(r'\s+', ' ', text).strip()
+    return text
+
 # --- Edge TTS synth ---
 async def edge_tts_synthesize(text, voice, user_id):
     out_path = f"output_{user_id}.mp3"
@@ -197,6 +215,7 @@ window.setTimeout(function(){
 </script>
 """, unsafe_allow_html=True)
 
+# --- Handle user input
 if user_input:
     thread_id = get_or_create_thread_id()
     client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
@@ -216,7 +235,8 @@ if user_input:
     mute_voice = st.session_state.get("mute_voice", False)
     audio_path = None
     if not mute_voice and assistant_message.strip():
-        audio_path = synthesize_voice(assistant_message, st.session_state["selected_voice"], user_id)
+        clean_text = sanitize_for_tts(assistant_message)
+        audio_path = synthesize_voice(clean_text, st.session_state["selected_voice"], user_id)
         st.session_state["last_audio_path"] = audio_path
 
     time.sleep(0.2)