Update app.py

app.py

@@ -69,7 +69,7 @@ def save_message(role, content):
 # 💬 Display chat history
 def display_chat_history():
     messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
-    assistant_icon_html = "<img src='https://huggingface.co/spaces/IAMTFRMZA/lortechassistant/blob/main/lorain.jpg' width='20' style='vertical-align:middle;'/>"
+    assistant_icon_html = "<img src='https://lortechnologies.com/wp-content/uploads/2023/03/LOR-Online-Logo.svg' width='20' style='vertical-align:middle;'/>"
     for msg in list(messages)[::-1]:
         data = msg.to_dict()
         if data["role"] == "user":
@@ -77,7 +77,7 @@ def display_chat_history():
         else:
             st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LORAIN:</strong> {data['content']}</div>", unsafe_allow_html=True)
 
-# 🚀 Main Chat UI
+# --- Main Chat UI ---
 input_col, clear_col = st.columns([9, 1])
 with input_col:
     user_input = st.chat_input("Type your message here...")
@@ -100,22 +100,24 @@ display_chat_history()
 if "mute_voice" not in st.session_state:
     st.session_state["mute_voice"] = False
 
-def synthesize_and_play(text, mute):
-    if not mute:
-        # Only call OpenAI TTS if unmuted
+if "last_tts_text" not in st.session_state:
+    st.session_state["last_tts_text"] = ""
+
+def synthesize_voice(text, user_id):
+    audio_path = f"output_{user_id}.mp3"
+    # Only synthesize if text changed or file doesn't exist
+    if st.session_state["last_tts_text"] != text or not os.path.exists(audio_path):
         with st.spinner("Synthesizing voice with GPT-4o..."):
             speech_response = client.audio.speech.create(
-                model="tts-1", # 'tts-1' or 'tts-1-hd'
-                voice="nova",  # or "alloy", "echo", "fable", etc.
+                model="tts-1",
+                voice="nova",
                 input=text,
                 response_format="mp3"
             )
-            audio_path = f"output_{user_id}.mp3"
             with open(audio_path, "wb") as f:
                 f.write(speech_response.content)
-            st.audio(audio_path, format="audio/mp3", autoplay=True)
-            # Optionally remove the file after play to save disk
-            # os.remove(audio_path)
+        st.session_state["last_tts_text"] = text
+    return audio_path
 
 if user_input:
     # Send user message to OpenAI thread
@@ -135,20 +137,24 @@ if user_input:
     assistant_message = latest_response.content[0].text.value
     save_message("assistant", assistant_message)
 
-    # 👇 Voice controls (auto-speak enabled, can mute)
-    col1, col2 = st.columns([1, 1])
-    with col1:
-        if st.button("🔊 Play Voice", key="unmute"):
-            st.session_state["mute_voice"] = False
-            synthesize_and_play(assistant_message, False)
-    with col2:
-        if st.button("🔇 Mute Voice", key="mute"):
-            st.session_state["mute_voice"] = True
-            st.info("Voice output muted for this and future messages.")
+    # Voice autoplay unless muted
+    mute_voice = st.session_state.get("mute_voice", False)
+    audio_path = synthesize_voice(assistant_message, user_id) if not mute_voice else None
 
-    # Play voice unless muted
-    synthesize_and_play(assistant_message, st.session_state["mute_voice"])
+    if not mute_voice and audio_path:
+        st.audio(audio_path, format="audio/mp3", autoplay=True)
+    elif mute_voice:
+        st.info("🔇 Voice is muted. To enable assistant speech, click 'Unmute Voice' below and ask another question.")
+
+    # Single mute/unmute button
+    if not mute_voice:
+        if st.button("🔇 Mute Voice"):
+            st.session_state["mute_voice"] = True
+            st.rerun()
+    else:
+        if st.button("🔊 Unmute Voice"):
+            st.session_state["mute_voice"] = False
+            st.rerun()
 
-    # Force Streamlit to rerun so chat refreshes and you get a new prompt
     time.sleep(0.5)
     st.rerun()