Update app.py

app.py

@@ -5,21 +5,21 @@ import time
 import datetime
 import os
 
-# Load secrets from Hugging Face environment
+# Load Hugging Face secrets
 generated_user = os.getenv("User")
 generated_password = os.getenv("Password")
 openai_key = os.getenv("openai_key")
-assistant_id = os.getenv("ASSISTANT_ID")  # Now using the Hugging Face secret!
+assistant_id = os.getenv("ASSISTANT_ID")
 
-# Streamlit page setup
+# Streamlit UI 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
+# Light/Dark mode toggle
 dark_mode = st.toggle("🌙 Switch to Light Mode", value=True)
 
-# Add custom CSS for styling bubbles
+# Custom chat bubble styling
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
@@ -28,7 +28,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Login authentication
+# Authentication
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -46,7 +46,7 @@ if not st.session_state.authenticated:
         else:
             st.error("Incorrect username or password. Please try again.")
 
-# Main chat after login
+# Main chat interface
 else:
     st.divider()
 
@@ -56,11 +56,11 @@ else:
     # Display chat history
     for msg in st.session_state["messages"]:
         if msg["role"] == "assistant":
-            message(msg["content"], is_user=False, avatar="https://www.carfind.co.za/images/Carfind-Icon.svg")
+            message(f"![carfind-logo](https://www.carfind.co.za/images/Carfind-Icon.svg) **Carfind Assistant:** {msg['content']}", is_user=False)
         else:
             message(msg["content"], is_user=True, avatar_style="adventurer")
 
-    # Chat input and clear button side-by-side
+    # Input + clear button
     input_col, clear_col = st.columns([8, 1])
     with input_col:
         user_input = st.text_input("Type your message here...", key="chat_input")
@@ -71,6 +71,7 @@ else:
             st.success("Chat cleared.")
             st.rerun()
 
+    # Handle chat
     if openai_key and assistant_id:
         client = OpenAI(api_key=openai_key)
 
@@ -82,12 +83,10 @@ else:
                 log.write("--- End Chat ---\n")
 
         if user_input:
-            # Display user message
             st.session_state.messages.append({"role": "user", "content": user_input})
             message(user_input, is_user=True, avatar_style="adventurer")
 
             try:
-                # Start OpenAI thread and show typing spinner
                 with st.spinner("Thinking and typing... 💭"):
                     thread = client.beta.threads.create()
                     client.beta.threads.messages.create(thread_id=thread.id, role="user", content=user_input)
@@ -100,18 +99,17 @@ else:
                             break
                         time.sleep(1)
 
-                    # Get assistant reply
+                    # Get the assistant reply
                     response_messages = client.beta.threads.messages.list(thread_id=thread.id)
                     assistant_reply = response_messages.data[0].content[0].text.value
 
                 st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
-                message(assistant_reply, is_user=False, avatar="https://www.carfind.co.za/images/Carfind-Icon.svg")
+                message(f"![carfind-logo](https://www.carfind.co.za/images/Carfind-Icon.svg) **Carfind Assistant:** {assistant_reply}", is_user=False)
 
-                # Save chat log
                 save_transcript(st.session_state.messages)
 
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
 
     else:
-        st.error("⚠️ API key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")
+        st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")