Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from openai import OpenAI
-from PIL import Image
 import time
 import os
 import uuid
@@ -47,7 +46,7 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# Function to get or create a thread ID for chat
+# 🔁 Get or create a thread ID
 def get_or_create_thread_id():
     doc_ref = db.collection("users").document(user_id)
     doc = doc_ref.get()
@@ -58,7 +57,7 @@ def get_or_create_thread_id():
         doc_ref.set({"thread_id": thread.id, "created_at": firestore.SERVER_TIMESTAMP})
         return thread.id
 
-# Save message to the Firestore database
+# 💾 Save a message
 def save_message(role, content):
     db.collection("users").document(user_id).collection("messages").add({
         "role": role,
@@ -66,7 +65,7 @@ def save_message(role, content):
         "timestamp": firestore.SERVER_TIMESTAMP
     })
 
-# Display chat history
+# 💬 Display chat history
 def display_chat_history():
     messages = db.collection("users").document(user_id).collection("messages").order_by("timestamp").stream()
     assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='20' style='vertical-align:middle;'/>"
@@ -77,10 +76,11 @@ def display_chat_history():
         else:
             st.markdown(f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>{assistant_icon_html} <strong>Carfind Assistant:</strong> {data['content']}</div>", unsafe_allow_html=True)
 
-# 🚀 Main Assistant Chat Interface
+# 🚀 Main Chat UI
 input_col, clear_col = st.columns([9, 1])
 with input_col:
     user_input = st.chat_input("Type your message here...")
+
 with clear_col:
     if st.button("🗑️", key="clear-chat", help="Clear Chat"):
         try:
@@ -99,6 +99,7 @@ display_chat_history()
 if user_input:
     client.beta.threads.messages.create(thread_id=thread_id, role="user", content=user_input)
     save_message("user", user_input)
+
     with st.spinner("Thinking and typing... 💭"):
         run = client.beta.threads.runs.create(thread_id=thread_id, assistant_id=assistant_id)
         while True:
@@ -106,9 +107,11 @@ if user_input:
             if run_status.status == "completed":
                 break
             time.sleep(1)
+
     messages_response = client.beta.threads.messages.list(thread_id=thread_id)
     latest_response = sorted(messages_response.data, key=lambda x: x.created_at)[-1]
     assistant_message = latest_response.content[0].text.value
+
     save_message("assistant", assistant_message)
     time.sleep(0.5)
     st.rerun()