Update app.py

app.py

@@ -33,6 +33,8 @@ if "last_tts_text" not in st.session_state:
     st.session_state["last_tts_text"] = ""
 if "last_audio_path" not in st.session_state:
     st.session_state["last_audio_path"] = ""
+if "is_thinking" not in st.session_state:
+    st.session_state["is_thinking"] = False
 
 # --- Page config ---
 st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
@@ -119,7 +121,7 @@ def display_chat_history():
 
 # --- Edge TTS synth ---
 def sanitize_tts_text(text):
-    text = re.sub(r'[^\w\s\.\,]', '', text)  # remove emojis & special chars
+    text = re.sub(r'[^\w\s\.\,\!\?\:\;\'\"]', '', text)  # keep basic punctuation
     text = text.replace('.co.za', 'dot coza')
     return text
 
@@ -134,15 +136,29 @@ def synthesize_voice(text, user_id):
     sanitized = sanitize_tts_text(text)
     out_path = f"output_{user_id}.mp3"
     if st.session_state["last_tts_text"] != sanitized or not os.path.exists(out_path):
-        with st.spinner(f"Generating voice ({FIXED_VOICE_NAME})..."):
-            asyncio.run(edge_tts_synthesize(sanitized, voice, user_id))
-        st.session_state["last_tts_text"] = sanitized
-        st.session_state["last_audio_path"] = out_path
+        try:
+            with st.spinner(f"Generating voice ({FIXED_VOICE_NAME})..."):
+                asyncio.run(edge_tts_synthesize(sanitized, voice, user_id))
+            st.session_state["last_tts_text"] = sanitized
+            st.session_state["last_audio_path"] = out_path
+        except Exception as e:
+            st.warning(f"TTS Error: {e}")
+            return None
     return out_path
 
 # --- CHAT: display history ---
 display_chat_history()
 
+# --- LORAIN is thinking indicator ---
+if st.session_state.get("is_thinking", False):
+    st.markdown("""
+    <div style="
+        text-align:center; color:#ddd; font-size: 14px;
+        margin-top: -1em; margin-bottom: 0.5em;">
+        🤖 <em>LORAIN is thinking...</em>
+    </div>
+    """, unsafe_allow_html=True)
+
 # --- INPUT BAR (floating at bottom) ---
 st.markdown('<div class="chat-input-bar">', unsafe_allow_html=True)
 col1, col2 = st.columns([10, 1])
@@ -153,6 +169,8 @@ st.markdown('</div>', unsafe_allow_html=True)
 
 # --- PROCESS USER INPUT ---
 if user_input:
+    st.session_state["is_thinking"] = True  # start thinking
+
     thread_id = get_or_create_thread_id()
     client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
     save_message("user", user_input)
@@ -171,8 +189,24 @@ if user_input:
     save_message("assistant", assistant_message)
 
     audio_path = synthesize_voice(assistant_message, user_id)
-    if os.path.exists(audio_path):
+    if audio_path and os.path.exists(audio_path):
         st.audio(audio_path, format="audio/mp3", autoplay=True)
 
+    st.session_state["is_thinking"] = False  # stop thinking
+
     time.sleep(0.2)
     st.rerun()
+
+# --- Auto-scroll JS ---
+st.markdown("""
+<script>
+window.onload = function() {
+    var chatWrapper = document.querySelector('.chat-history-wrapper');
+    if(chatWrapper){ chatWrapper.scrollTop = chatWrapper.scrollHeight; }
+};
+setTimeout(function(){
+    var chatWrapper = document.querySelector('.chat-history-wrapper');
+    if(chatWrapper){ chatWrapper.scrollTop = chatWrapper.scrollHeight; }
+}, 300);
+</script>
+""", unsafe_allow_html=True)