Update app.py

app.py

@@ -31,10 +31,8 @@ VOICE_OPTIONS = {
     "Luke (SA, Male)": "en-ZA-LukeNeural"
 }
 
-# --- Streamlit Config ---
 st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
 
-# --- User/session state ---
 if "user_id" not in st.session_state:
     st.session_state["user_id"] = str(uuid.uuid4())
 user_id = st.session_state["user_id"]
@@ -48,12 +46,6 @@ if "last_audio_path" not in st.session_state:
 if "selected_voice" not in st.session_state:
     st.session_state["selected_voice"] = "Jenny (US, Female)"
 
-# --- Sidebar for Voice Selection ---
-with st.sidebar:
-    st.markdown("### Voice Settings")
-    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
-
 # --- Branding & Styling ---
 st.markdown("""
     <style>
@@ -63,6 +55,8 @@ st.markdown("""
     .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
     .stChatMessage[data-testid="stChatMessage-assistant"] { background: #e3f2fd; color: #000000; }
     .lt-logo { vertical-align: middle; }
+    .st-emotion-cache-1avcm0n { justify-content: flex-end !important; }
+    .stChatInputContainer { position: fixed !important; bottom: 0; width: 80vw; z-index: 100; left: 10vw; background: #11131a; }
     </style>
 """, unsafe_allow_html=True)
 st.markdown("""
@@ -74,6 +68,33 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
+# --- Sidebar: All audio/controls here ---
+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
+
+    # Audio player, always present if we have an mp3
+    last_audio = st.session_state.get("last_audio_path")
+    mute_voice = st.session_state.get("mute_voice", False)
+
+    # Replay button and audio player
+    if last_audio and os.path.exists(last_audio):
+        # Autoplay if this was just generated, else manual play
+        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)
+
+    # Mute/Unmute
+    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)
@@ -121,26 +142,13 @@ def synthesize_voice(text, voice_key, user_id):
         st.session_state["last_voice"] = voice
     return out_path
 
-# --- Main Chat UI ---
-input_col, clear_col = st.columns([9, 1])
-with input_col:
-    user_input = st.chat_input("Type your message here...")
-
-with clear_col:
-    if st.button("🗑️", key="clear-chat", help="Clear Chat"):
-        try:
-            user_doc_ref = db.collection("users").document(user_id)
-            for msg in user_doc_ref.collection("messages").stream():
-                msg.reference.delete()
-            user_doc_ref.delete()
-            st.session_state.clear()
-            st.rerun()
-        except Exception as e:
-            st.error(f"Failed to clear chat: {e}")
-
+# --- Main Chat UI (text only!) ---
 thread_id = get_or_create_thread_id()
 display_chat_history()
 
+# --- Static Chat Input at Bottom ---
+user_input = st.chat_input("Type your message here...")
+
 if user_input:
     # --- OpenAI Assistant Response ---
     client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
@@ -163,41 +171,7 @@ if user_input:
     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
-        st.audio(audio_path, format="audio/mp3", autoplay=True)
-    elif mute_voice:
-        st.info("🔇 Voice is muted. Click Unmute below to enable assistant speech.")
-
-    # --- Controls (Mute/Unmute/Replay) ---
-    col1, col2 = st.columns([1, 1])
-    with col1:
-        if not mute_voice and st.button("🔇 Mute Voice"):
-            st.session_state["mute_voice"] = True
-            st.rerun()
-        elif mute_voice and st.button("🔊 Unmute Voice"):
-            st.session_state["mute_voice"] = False
-            st.rerun()
-    with col2:
-        # Replay button: Always available if last_audio_path exists
-        if st.session_state.get("last_audio_path") and os.path.exists(st.session_state["last_audio_path"]):
-            if st.button("🔁 Replay Voice"):
-                st.audio(st.session_state["last_audio_path"], format="audio/mp3", autoplay=True)
+        # No audio in main area! Sidebar will autoplay due to above code.
 
     time.sleep(0.2)
     st.rerun()
-else:
-    # Always show last audio with replay if available
-    if st.session_state.get("last_audio_path") and os.path.exists(st.session_state["last_audio_path"]) and not st.session_state["mute_voice"]:
-        st.audio(st.session_state["last_audio_path"], format="audio/mp3", autoplay=False)
-        # Controls: Only show Replay when idle
-        if st.button("🔁 Replay Last Voice"):
-            st.audio(st.session_state["last_audio_path"], format="audio/mp3", autoplay=True)
-
-    # Show mute/unmute in idle state too
-    if not st.session_state["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()