Update app.py

app.py

@@ -1,10 +1,32 @@
-import streamlit as st
+import streamlit as st 
 from openai import OpenAI
 import time
 import os
 import uuid
 import firebase_admin
 from firebase_admin import credentials, firestore
+# ------------------ Authentication ------------------
+VALID_USERS = {
+    "[email protected]": "Pass.123",
+}
+
+def login():
+    st.title("🔐 Login Required")
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
+    if st.button("Login"):
+        if VALID_USERS.get(email) == password:
+            st.session_state.authenticated = True
+            st.rerun()
+        else:
+            st.error("❌ Incorrect email or password.")
+
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+
+if not st.session_state.authenticated:
+    login()
+    st.stop()
 
 # 🔐 Firebase setup
 if not firebase_admin._apps:
@@ -15,7 +37,7 @@ db = firestore.client()
 
 # 🔐 OpenAI setup
 openai_key = os.getenv("openai_key")
-assistant_id = os.getenv("assistant_id")
+assistant_id = os.getenv("ASSISTANT_ID")
 client = OpenAI(api_key=openai_key)
 
 # 🌐 Streamlit Config
@@ -26,7 +48,7 @@ if "user_id" not in st.session_state:
     st.session_state["user_id"] = str(uuid.uuid4())
 user_id = st.session_state["user_id"]
 
-# 🖼️ LOR Branding + Styling
+# 🖼️ LORTech Branding + Styling
 st.markdown("""
     <style>
     .block-container {padding-top: 1rem; padding-bottom: 0rem;}
@@ -34,7 +56,7 @@ st.markdown("""
     .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; }
-    .lor-logo { vertical-align: middle; }
+    .lt-logo { vertical-align: middle; }
     </style>
 """, unsafe_allow_html=True)
 
@@ -75,7 +97,7 @@ def display_chat_history():
         if data["role"] == "user":
             st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-user'>👤 <strong>You:</strong> {data['content']}</div>", unsafe_allow_html=True)
         else:
-            st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LOR Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
+            st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LeaseTracker Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
 
 # 🚀 Main Chat UI
 input_col, clear_col = st.columns([9, 1])
@@ -119,5 +141,3 @@ if user_input:
     st.rerun()
 
 
-
-