Update app.py

app.py

@@ -11,7 +11,7 @@ openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
 
 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.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")
 
 dark_mode = st.toggle("🌙 Switch to Light Mode", value=True)
@@ -61,19 +61,19 @@ else:
         client = OpenAI(api_key=openai_key)
 
         if user_input:
-            # Create thread once and reuse it
+            # Create thread if it doesn't exist
             if not st.session_state["thread_id"]:
                 thread = client.beta.threads.create()
                 st.session_state["thread_id"] = thread.id
 
-            # Add user message to thread
+            # Add user message to the thread
             client.beta.threads.messages.create(
                 thread_id=st.session_state["thread_id"], role="user", content=user_input
             )
 
             try:
                 with st.spinner("Thinking and typing... 💭"):
-                    # Run assistant
+                    # Trigger assistant run
                     run = client.beta.threads.runs.create(
                         thread_id=st.session_state["thread_id"], assistant_id=assistant_id
                     )
@@ -86,17 +86,26 @@ else:
                             break
                         time.sleep(1)
 
-                # ✅ Fetch the entire conversation history:
+                # ✅ Retrieve entire conversation history
                 messages_response = client.beta.threads.messages.list(
                     thread_id=st.session_state["thread_id"]
                 )
 
-                # Display the entire conversation (in chronological order)
+                # Inline assistant icon HTML
+                assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
+
+                # Display chat history in correct order
                 for msg in reversed(messages_response.data):
                     if msg.role == "user":
                         message(msg.content[0].text.value, is_user=True, avatar_style="adventurer")
                     else:
-                        message(f"**Carfind Assistant:** {msg.content[0].text.value}", is_user=False, avatar_style="bottts")
+                        # Use markdown manually to render inline icon and text
+                        st.markdown(
+                            f"<div style='background-color:#00529B; color:white; padding:10px; border-radius:8px; margin:5px 0;'>"
+                            f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {msg.content[0].text.value}"
+                            f"</div>", 
+                            unsafe_allow_html=True
+                        )
 
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")