Update app.py

app.py

@@ -3,6 +3,7 @@ import asyncio
 import edge_tts
 import time
 import os
+import re
 import uuid
 import firebase_admin
 from firebase_admin import credentials, firestore
@@ -19,34 +20,24 @@ openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("assistant_id")
 client = OpenAI(api_key=openai_key)
 
-VOICE_OPTIONS = {
-    "Jenny (US, Female)": "en-US-JennyNeural",
-    "Aria (US, Female)": "en-US-AriaNeural",
-    "Ryan (UK, Male)": "en-GB-RyanNeural",
-    "Natasha (AU, Female)": "en-AU-NatashaNeural",
-    "William (AU, Male)": "en-AU-WilliamNeural",
-    "Libby (UK, Female)": "en-GB-LibbyNeural",
-    "Leah (SA, Female)": "en-ZA-LeahNeural",
-    "Luke (SA, Male)": "en-ZA-LukeNeural"
-}
-
-st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
+# ---- Voice Settings ----
+FIXED_VOICE_NAME = "Jenny (US, Female)"
+FIXED_VOICE = "en-US-JennyNeural"
 
 # --- State setup
 if "user_id" not in st.session_state:
     st.session_state["user_id"] = str(uuid.uuid4())
 user_id = st.session_state["user_id"]
 
-if "mute_voice" not in st.session_state:
-    st.session_state["mute_voice"] = False
 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 "selected_voice" not in st.session_state:
-    st.session_state["selected_voice"] = "Jenny (US, Female)"
 
-# --- CSS for logo, chat, and input bar at top ---
+# --- Page config ---
+st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
+
+# --- CSS Styling ---
 st.markdown("""
     <style>
     .block-container {padding-top: 1rem;}
@@ -64,21 +55,19 @@ st.markdown("""
     .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
     .stChatMessage[data-testid="stChatMessage-assistant"] { background: #e3f2fd; color: #000000; }
     .chat-history-wrapper {
-        margin-top: 0.5em; padding-bottom: 2em; min-height: 60vh;
+        margin-top: 0.5em; margin-bottom: 5em; height: 65vh; overflow-y: auto; padding: 0 0.5em;
     }
-    .input-top-bar {
-        position: sticky; top: 0; z-index: 20;
-        background: #191b22; padding: 0.5em 0 0.25em 0;
-        display: flex; align-items: center; gap: 0.4em;
-        border-bottom: 1px solid #22232c;
+    .chat-input-bar {
+        position: fixed; bottom: 0; width: 100%; z-index: 100;
+        background: #191b22; padding: 0.6em 1em; border-top: 1px solid #22232c;
+        display: flex; align-items: center; gap: 0.5em;
     }
-    .input-top-bar .element-container { flex: 1 1 auto; }
-    .input-top-bar input { font-size: 1.12em !important; }
-    .clear-chat-btn { background: none; border: none; font-size: 1.4em; color: #666; cursor: pointer; }
+    .chat-input-bar input { width: 100%; font-size: 1.1em; }
+    .clear-chat-btn { background: none; border: none; font-size: 1.4em; color: #999; cursor: pointer; }
     </style>
 """, unsafe_allow_html=True)
 
-# --- Top Branding, Mini Logo ---
+# --- Top Branding ---
 st.markdown("""
 <div class="lor-brand-bar">
     <img src="https://lortechnologies.com/wp-content/uploads/2023/03/LOR-Online-Logo.svg" class="logo-mini" />
@@ -86,30 +75,6 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-# --- Sidebar: audio/voice controls
-with st.sidebar:
-    st.markdown("### Voice Settings & Controls")
-    selected_voice = st.selectbox(
-        "Select assistant voice", list(VOICE_OPTIONS.keys()),
-        index=list(VOICE_OPTIONS.keys()).index(st.session_state["selected_voice"])
-    )
-    st.session_state["selected_voice"] = selected_voice
-
-    last_audio = st.session_state.get("last_audio_path")
-    mute_voice = st.session_state.get("mute_voice", False)
-    if last_audio and os.path.exists(last_audio):
-        st.audio(last_audio, format="audio/mp3", autoplay=not mute_voice)
-        if st.button("🔁 Replay Voice"):
-            st.audio(last_audio, format="audio/mp3", autoplay=True)
-    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()
-
 # --- Firestore helpers ---
 def get_or_create_thread_id():
     doc_ref = db.collection("users").document(user_id)
@@ -140,7 +105,7 @@ def display_chat_history():
     messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
     assistant_icon_html = "<img src='https://raw.githubusercontent.com/AndrewLORTech/lortechwebsite/main/lorain.jpg' width='22' style='vertical-align:middle; border-radius:50%;'/>"
     chat_msgs = []
-    for msg in list(messages)[::-1]:   # Most recent first!
+    for msg in list(messages):
         data = msg.to_dict()
         if data["role"] == "user":
             chat_msgs.append(
@@ -151,57 +116,47 @@ def display_chat_history():
                 f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LORAIN:</strong> {data['content']}</div>"
             )
     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)
 
 # --- Edge TTS synth ---
+def sanitize_tts_text(text):
+    text = re.sub(r'[^\w\s\.\,]', '', text)  # remove emojis & special chars
+    text = text.replace('.co.za', 'dot coza')
+    return text
+
 async def edge_tts_synthesize(text, voice, user_id):
     out_path = f"output_{user_id}.mp3"
     communicate = edge_tts.Communicate(text, voice)
     await communicate.save(out_path)
     return out_path
 
-def synthesize_voice(text, voice_key, user_id):
-    voice = VOICE_OPTIONS[voice_key]
+def synthesize_voice(text, user_id):
+    voice = FIXED_VOICE
+    sanitized = sanitize_tts_text(text)
     out_path = f"output_{user_id}.mp3"
-    if st.session_state["last_tts_text"] != text or not os.path.exists(out_path) or st.session_state.get("last_voice") != voice:
-        with st.spinner(f"Generating voice ({voice_key})..."):
-            asyncio.run(edge_tts_synthesize(text, voice, user_id))
-        st.session_state["last_tts_text"] = text
+    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
-        st.session_state["last_voice"] = voice
     return out_path
 
-# --- INPUT AT THE TOP ---
-with st.container():
-    st.markdown('<div class="input-top-bar">', unsafe_allow_html=True)
-    col1, col2 = st.columns([10, 1])
-    user_input = col1.chat_input("Type your message here...")
-    if col2.button("🗑️", help="Clear Chat", key="clear-chat-top"):
-        clear_chat_history()
-    st.markdown('</div>', unsafe_allow_html=True)
-
-# --- CHAT: display under input, latest on top ---
+# --- CHAT: display history ---
 display_chat_history()
 
-# --- JS: auto-scroll to top on new message ---
-st.markdown("""
-<script>
-window.onload = function() {
-    var anchor = document.getElementById("chat-top-anchor");
-    if(anchor){ anchor.scrollIntoView({ behavior: "smooth", block: "start" }); }
-};
-window.setTimeout(function(){
-    var anchor = document.getElementById("chat-top-anchor");
-    if(anchor){ anchor.scrollIntoView({ behavior: "smooth", block: "start" }); }
-}, 200);
-</script>
-""", 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])
+user_input = col1.chat_input("Type your message here...")
+if col2.button("🗑️", help="Clear Chat", key="clear-chat-bottom"):
+    clear_chat_history()
+st.markdown('</div>', unsafe_allow_html=True)
 
+# --- PROCESS USER INPUT ---
 if user_input:
-    # --- OpenAI Assistant Response ---
     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)
+
     with st.spinner("Thinking and typing... 💭"):
         run = client.beta.threads.runs.create(thread_id=thread_id, assistant_id=assistant_id)
         while True:
@@ -209,16 +164,15 @@ if user_input:
             if run_status.status == "completed":
                 break
             time.sleep(1)
+
     messages_response = client.beta.threads.messages.list(thread_id=thread_id)
     latest_response = sorted(messages_response.data, key=lambda x: x.created_at)[-1]
     assistant_message = latest_response.content[0].text.value
     save_message("assistant", assistant_message)
 
-    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)
-        st.session_state["last_audio_path"] = audio_path
+    audio_path = synthesize_voice(assistant_message, user_id)
+    if os.path.exists(audio_path):
+        st.audio(audio_path, format="audio/mp3", autoplay=True)
 
     time.sleep(0.2)
     st.rerun()