Update app.py

app.py

@@ -5,21 +5,21 @@ import time
 import datetime
 import os
 
-# Load Hugging Face secrets
+# 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 UI setup
+# 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
+# Light / Dark mode toggle
 dark_mode = st.toggle("🌙 Switch to Light Mode", value=True)
 
-# Custom chat bubble styling
+# Custom styling for bubbles
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
@@ -46,21 +46,21 @@ if not st.session_state.authenticated:
         else:
             st.error("Incorrect username or password. Please try again.")
 
-# Main chat interface
+# Chat UI after login
 else:
     st.divider()
 
     if "messages" not in st.session_state:
         st.session_state["messages"] = []
 
-    # Display chat history
+    # Show chat history
     for msg in st.session_state["messages"]:
         if msg["role"] == "assistant":
-            message(f"![carfind-logo](https://www.carfind.co.za/images/Carfind-Icon.svg) **Carfind Assistant:** {msg['content']}", is_user=False)
+            message(f"🚗 **Carfind Assistant:** {msg['content']}", is_user=False)
         else:
             message(msg["content"], is_user=True, avatar_style="adventurer")
 
-    # Input + clear button
+    # 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")
@@ -71,7 +71,6 @@ else:
             st.success("Chat cleared.")
             st.rerun()
 
-    # Handle chat
     if openai_key and assistant_id:
         client = OpenAI(api_key=openai_key)
 
@@ -99,12 +98,10 @@ else:
                             break
                         time.sleep(1)
 
-                    # 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
+                    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-logo](https://www.carfind.co.za/images/Carfind-Icon.svg) **Carfind Assistant:** {assistant_reply}", is_user=False)
+                message(f"🚗 **Carfind Assistant:** {assistant_reply}", is_user=False)
 
                 save_transcript(st.session_state.messages)