Update app.py

app.py

@@ -5,21 +5,17 @@ import time
 import datetime
 import os
 
-# Load secrets from Hugging Face environment
 generated_user = os.getenv("User")
 generated_password = os.getenv("Password")
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
 
-# Streamlit page setup
 st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 st.markdown("<h1 style='text-align: center;'>🚗 Carfind.co.za AI Assistant</h1>", unsafe_allow_html=True)
 st.caption("Chat with Carfind.co.za and find your next car fast")
 
-# Light / Dark mode toggle
 dark_mode = st.toggle("🌙 Switch to Light Mode", value=True)
 
-# Custom styling for bubbles
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
@@ -28,7 +24,6 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Authentication
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -46,21 +41,18 @@ if not st.session_state.authenticated:
         else:
             st.error("Incorrect username or password. Please try again.")
 
-# Chat UI after login
 else:
     st.divider()
 
     if "messages" not in st.session_state:
         st.session_state["messages"] = []
 
-    # Show chat history
     for msg in st.session_state["messages"]:
         if msg["role"] == "assistant":
-            message(f"🚗 **Carfind Assistant:** {msg['content']}", is_user=False)
+            message(msg['content'], is_user=False, avatar_url="https://www.carfind.co.za/images/Carfind-Icon.svg")
         else:
             message(msg["content"], is_user=True, avatar_style="adventurer")
 
-    # Input and Clear Chat button
     input_col, clear_col = st.columns([8, 1])
     with input_col:
         user_input = st.text_input("Type your message here...", key="chat_input")
@@ -101,7 +93,7 @@ else:
                     assistant_reply = client.beta.threads.messages.list(thread_id=thread.id).data[0].content[0].text.value
 
                 st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
-                message(f"🚗 **Carfind Assistant:** {assistant_reply}", is_user=False)
+                message(assistant_reply, is_user=False, avatar_url="https://www.carfind.co.za/images/Carfind-Icon.svg")
 
                 save_transcript(st.session_state.messages)