IAMTFRMZA commited on
Commit
acc6be8
Β·
verified Β·
1 Parent(s): 0e1b8a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import streamlit as st
2
  from openai import OpenAI
3
  import time
4
  import os
@@ -19,14 +19,14 @@ assistant_id = os.getenv("ASSISTANT_ID")
19
  client = OpenAI(api_key=openai_key)
20
 
21
  # 🌐 Streamlit Config
22
- st.set_page_config(page_title="LeaseTracker AI Assistant", layout="wide")
23
 
24
  # 🎯 Session + User ID
25
  if "user_id" not in st.session_state:
26
  st.session_state["user_id"] = str(uuid.uuid4())
27
  user_id = st.session_state["user_id"]
28
 
29
- # πŸ–ΌοΈ LeaseTracker Branding + Styling
30
  st.markdown("""
31
  <style>
32
  .block-container {padding-top: 1rem; padding-bottom: 0rem;}
@@ -34,15 +34,15 @@ st.markdown("""
34
  .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
35
  .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
36
  .stChatMessage[data-testid="stChatMessage-assistant"] { background: #e3f2fd; color: #000000; }
37
- .lt-logo { vertical-align: middle; }
38
  </style>
39
  """, unsafe_allow_html=True)
40
 
41
  st.markdown("""
42
  <div style='text-align: center; margin-top: 20px; margin-bottom: -10px;'>
43
  <span style='display: inline-flex; align-items: center; gap: 8px;'>
44
- <img src='https://leasetracker.co.za/wp-content/uploads/2020/10/Lease-Tracker-Logo.png' width='100' class='lt-logo'/>
45
- <span style='font-size: 12px; color: gray;'>Powered by LeaseTracker</span>
46
  </span>
47
  </div>
48
  """, unsafe_allow_html=True)
@@ -69,13 +69,13 @@ def save_message(role, content):
69
  # πŸ’¬ Display chat history
70
  def display_chat_history():
71
  messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
72
- assistant_icon_html = "<img src='https://leasetracker.co.za/wp-content/uploads/2020/10/Lease-Tracker-Logo.png' width='20' style='vertical-align:middle;'/>"
73
  for msg in list(messages)[::-1]:
74
  data = msg.to_dict()
75
  if data["role"] == "user":
76
  st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-user'>πŸ‘€ <strong>You:</strong> {data['content']}</div>", unsafe_allow_html=True)
77
  else:
78
- st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LeaseTracker Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
79
 
80
  # πŸš€ Main Chat UI
81
  input_col, clear_col = st.columns([9, 1])
 
1
+ import streamlit as st
2
  from openai import OpenAI
3
  import time
4
  import os
 
19
  client = OpenAI(api_key=openai_key)
20
 
21
  # 🌐 Streamlit Config
22
+ st.set_page_config(page_title="LOR Technologies AI Assistant", layout="wide")
23
 
24
  # 🎯 Session + User ID
25
  if "user_id" not in st.session_state:
26
  st.session_state["user_id"] = str(uuid.uuid4())
27
  user_id = st.session_state["user_id"]
28
 
29
+ # πŸ–ΌοΈ LOR Branding + Styling
30
  st.markdown("""
31
  <style>
32
  .block-container {padding-top: 1rem; padding-bottom: 0rem;}
 
34
  .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
35
  .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
36
  .stChatMessage[data-testid="stChatMessage-assistant"] { background: #e3f2fd; color: #000000; }
37
+ .lor-logo { vertical-align: middle; }
38
  </style>
39
  """, unsafe_allow_html=True)
40
 
41
  st.markdown("""
42
  <div style='text-align: center; margin-top: 20px; margin-bottom: -10px;'>
43
  <span style='display: inline-flex; align-items: center; gap: 8px;'>
44
+ <img src='https://lortechnologies.com/wp-content/uploads/2023/03/LOR-Online-Logo.svg' width='100' class='lor-logo'/>
45
+ <span style='font-size: 12px; color: gray;'>Powered by LOR Technologies</span>
46
  </span>
47
  </div>
48
  """, unsafe_allow_html=True)
 
69
  # πŸ’¬ Display chat history
70
  def display_chat_history():
71
  messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
72
+ assistant_icon_html = "<img src='https://lortechnologies.com/wp-content/uploads/2023/03/LOR-Online-Logo.svg' width='20' style='vertical-align:middle;'/>"
73
  for msg in list(messages)[::-1]:
74
  data = msg.to_dict()
75
  if data["role"] == "user":
76
  st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-user'>πŸ‘€ <strong>You:</strong> {data['content']}</div>", unsafe_allow_html=True)
77
  else:
78
+ st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>LOR Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
79
 
80
  # πŸš€ Main Chat UI
81
  input_col, clear_col = st.columns([9, 1])