Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,22 +26,23 @@ 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 |
-
# 🖼️ Branding + Styling
|
| 30 |
st.markdown("""
|
| 31 |
<style>
|
| 32 |
.block-container {padding-top: 1rem; padding-bottom: 0rem;}
|
| 33 |
header {visibility: hidden;}
|
| 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: #
|
|
|
|
| 37 |
</style>
|
| 38 |
""", unsafe_allow_html=True)
|
| 39 |
|
| 40 |
st.markdown("""
|
| 41 |
<div style='text-align: center; margin-top: 20px; margin-bottom: -10px;'>
|
| 42 |
<span style='display: inline-flex; align-items: center; gap: 8px;'>
|
| 43 |
-
<img src='https://
|
| 44 |
-
<span style='font-size: 12px; color: gray;'>Powered by
|
| 45 |
</span>
|
| 46 |
</div>
|
| 47 |
""", unsafe_allow_html=True)
|
|
@@ -68,13 +69,13 @@ def save_message(role, content):
|
|
| 68 |
# 💬 Display chat history
|
| 69 |
def display_chat_history():
|
| 70 |
messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
|
| 71 |
-
assistant_icon_html = "<img src='https://
|
| 72 |
for msg in list(messages)[::-1]:
|
| 73 |
data = msg.to_dict()
|
| 74 |
if data["role"] == "user":
|
| 75 |
st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-user'>👤 <strong>You:</strong> {data['content']}</div>", unsafe_allow_html=True)
|
| 76 |
else:
|
| 77 |
-
st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>
|
| 78 |
|
| 79 |
# 🚀 Main Chat UI
|
| 80 |
input_col, clear_col = st.columns([9, 1])
|
|
@@ -97,6 +98,7 @@ thread_id = get_or_create_thread_id()
|
|
| 97 |
display_chat_history()
|
| 98 |
|
| 99 |
if user_input:
|
|
|
|
| 100 |
client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
|
| 101 |
save_message("user", user_input)
|
| 102 |
|
|
|
|
| 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;}
|
| 33 |
header {visibility: hidden;}
|
| 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])
|
|
|
|
| 98 |
display_chat_history()
|
| 99 |
|
| 100 |
if user_input:
|
| 101 |
+
# Send user message to OpenAI thread
|
| 102 |
client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
|
| 103 |
save_message("user", user_input)
|
| 104 |
|