Update app.py

app.py

@@ -8,18 +8,17 @@ import firebase_admin
 from firebase_admin import credentials, firestore
 from openai import OpenAI
 
-# ---- Firebase setup ----
+# Firebase setup
 if not firebase_admin._apps:
     cred = credentials.Certificate("firebase-service-account.json")
     firebase_admin.initialize_app(cred)
 db = firestore.client()
 
-# ---- OpenAI setup ----
+# OpenAI setup
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("assistant_id")
 client = OpenAI(api_key=openai_key)
 
-# ---- Edge TTS voices ----
 VOICE_OPTIONS = {
     "Jenny (US, Female)": "en-US-JennyNeural",
     "Aria (US, Female)": "en-US-AriaNeural",
@@ -33,6 +32,7 @@ VOICE_OPTIONS = {
 
 st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
 
+# State
 if "user_id" not in st.session_state:
     st.session_state["user_id"] = str(uuid.uuid4())
 user_id = st.session_state["user_id"]
@@ -46,18 +46,26 @@ 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)"
 
-# --- Branding & Styling ---
+# CSS for floating bar illusion (margin for chat, input pinned visually)
 st.markdown("""
     <style>
-    .block-container {padding-top: 1rem; padding-bottom: 0rem;}
-    header {visibility: hidden;}
+    .block-container {padding-top: 1rem;}
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
     .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
     .stChatMessage[data-testid="stChatMessage-assistant"] { background: #e3f2fd; color: #000000; }
     .lt-logo { vertical-align: middle; }
-    .fixed-bottom-bar { position: fixed; bottom: 0; left: 0; width: 100vw; background: #181a22; padding: 1.2em 0.5em 0.8em 0.5em; z-index: 9999; box-shadow: 0 -2px 8px rgba(0,0,0,0.06);}
-    .chat-area { margin-bottom: 6.1em; }
-    .st-emotion-cache-1avcm0n { justify-content: flex-end !important; }
+    .footer-fakebar {
+        position: fixed;
+        left: 0; bottom: 0;
+        width: 100vw;
+        background: #181a22;
+        box-shadow: 0 -2px 8px rgba(0,0,0,0.10);
+        padding: 1.2em 0.5em 0.8em 0.5em;
+        z-index: 9999;
+    }
+    .footer-fakebar .element-container { flex: 1 1 auto; }
+    .footer-fakebar input { font-size: 1.15em !important; }
+    .footer-placeholder { height: 90px; }
     </style>
 """, unsafe_allow_html=True)
 st.markdown("""
@@ -69,27 +77,21 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# --- Sidebar: All audio/controls here ---
+# Sidebar: audio/voice controls
 with st.sidebar:
     st.markdown("### Voice Settings & Controls")
     selected_voice = st.selectbox(
-        "Select assistant voice",
-        list(VOICE_OPTIONS.keys()),
+        "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):
         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
@@ -99,7 +101,6 @@ with st.sidebar:
             st.session_state["mute_voice"] = False
             st.rerun()
 
-# --- Firestore helpers ---
 def get_or_create_thread_id():
     doc_ref = db.collection("users").document(user_id)
     doc = doc_ref.get()
@@ -126,24 +127,22 @@ def clear_chat_history():
     st.rerun()
 
 def display_chat_history():
-    st.markdown("<div class='chat-area'>", unsafe_allow_html=True)
     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='24' style='vertical-align:middle; border-radius:50%;'/>"
+    chat_msgs = []
     for msg in list(messages)[::-1]:
         data = msg.to_dict()
         if data["role"] == "user":
-            st.markdown(
-                f"<div class='stChatMessage' data-testid='stChatMessage-user'>👤 <strong>You:</strong> {data['content']}</div>",
-                unsafe_allow_html=True,
+            chat_msgs.append(
+                f"<div class='stChatMessage' data-testid='stChatMessage-user'>👤 <strong>You:</strong> {data['content']}</div>"
             )
         else:
-            st.markdown(
-                f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LORAIN:</strong> {data['content']}</div>",
-                unsafe_allow_html=True,
+            chat_msgs.append(
+                f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LORAIN:</strong> {data['content']}</div>"
             )
-    st.markdown("</div>", unsafe_allow_html=True)
+    st.markdown("".join(chat_msgs[::-1]), unsafe_allow_html=True)
 
-# --- Edge TTS synth ---
+# TTS
 async def edge_tts_synthesize(text, voice, user_id):
     out_path = f"output_{user_id}.mp3"
     communicate = edge_tts.Communicate(text, voice)
@@ -153,7 +152,6 @@ async def edge_tts_synthesize(text, voice, user_id):
 def synthesize_voice(text, voice_key, user_id):
     voice = VOICE_OPTIONS[voice_key]
     out_path = f"output_{user_id}.mp3"
-    # Only synthesize if text changed or file missing or voice changed
     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))
@@ -162,40 +160,40 @@ def synthesize_voice(text, voice_key, user_id):
         st.session_state["last_voice"] = voice
     return out_path
 
-# --- Main Chat UI (text only!) ---
-thread_id = get_or_create_thread_id()
+# --- Chat history and spacer ---
 display_chat_history()
+st.markdown('<div class="footer-placeholder"></div>', unsafe_allow_html=True)
 
-# --- Fixed bottom bar: input + clear chat ---
+# --- "Floating" Chat Input and Clear Chat button ---
 with st.container():
-    cols = st.columns([10, 1])
-    user_input = cols[0].chat_input("Type your message here...")
-    if cols[1].button("🗑️", help="Clear Chat"):
+    st.markdown('<div class="footer-fakebar">', 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)
 
 if user_input:
     # --- OpenAI Assistant Response ---
-    client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
+    client.beta.threads.messages.create(thread_id=get_or_create_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)
+        run = client.beta.threads.runs.create(thread_id=get_or_create_thread_id(), assistant_id=assistant_id)
         while True:
-            run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
+            run_status = client.beta.threads.runs.retrieve(thread_id=get_or_create_thread_id(), run_id=run.id)
             if run_status.status == "completed":
                 break
             time.sleep(1)
-    messages_response = client.beta.threads.messages.list(thread_id=thread_id)
+    messages_response = client.beta.threads.messages.list(thread_id=get_or_create_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)
 
-    # --- TTS: Speak unless muted ---
     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
-        # No audio in main area! Sidebar will autoplay due to above code.
 
     time.sleep(0.2)
     st.rerun()