Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 from streamlit_chat import message
 from openai import OpenAI
 import time
-import datetime
 import os
 
 generated_user = os.getenv("User")
@@ -14,11 +13,12 @@ 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")
 
+# Updated styling for readability
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
-    .stChatMessage[data-testid="stChatMessage-user"] { background: #333; color: white; }
-    .stChatMessage[data-testid="stChatMessage-assistant"] { background: #E4E9F7; color: black; }
+    .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
+    .stChatMessage[data-testid="stChatMessage-assistant"] { background: #D6E9FE; color: #000000; }
     </style>
 """, unsafe_allow_html=True)
 
@@ -84,22 +84,27 @@ else:
                             break
                         time.sleep(1)
 
-                # ✅ Retrieve entire conversation history
+                # Retrieve entire conversation history
                 messages_response = client.beta.threads.messages.list(
                     thread_id=st.session_state["thread_id"]
                 )
 
-                # Inline assistant icon HTML
+                # Assistant icon
                 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
+                # Display conversation
                 for msg in reversed(messages_response.data):
                     if msg.role == "user":
-                        message(msg.content[0].text.value, is_user=True, avatar_style="adventurer")
+                        # Add simple user avatar (👤) for each message
+                        st.markdown(
+                            f"<div style='background-color:#f0f0f0; color:#000000; padding:10px; border-radius:8px; margin:5px 0;'>"
+                            f"👤 <strong>You:</strong> {msg.content[0].text.value}"
+                            f"</div>", 
+                            unsafe_allow_html=True
+                        )
                     else:
-                        # 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"<div style='background-color:#D6E9FE; color:#000000; 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