Update app.py

app.py

@@ -1,34 +1,35 @@
 import streamlit as st
 from openai import OpenAI
+from PIL import Image
 import time
 import os
-import pandas as pd
 import uuid
-from datetime import datetime
-
-# Firebase setup
 import firebase_admin
 from firebase_admin import credentials, firestore
 
+# 🔐 Firebase setup
 if not firebase_admin._apps:
     cred = credentials.Certificate("firebase-service-account.json")
     firebase_admin.initialize_app(cred)
 
 db = firestore.client()
 
-# OpenAI setup
+# 🔐 OpenAI setup
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
 
+client = OpenAI(api_key=openai_key)
+
+# 🌐 Streamlit Config
 st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 
-# Session + Firebase user ID
+# 🎯 Session + User ID
 if "user_id" not in st.session_state:
     st.session_state["user_id"] = str(uuid.uuid4())
 
 user_id = st.session_state["user_id"]
 
-# Styling + Branding
+# 🖼️ Branding + Styling
 st.markdown("""
     <style>
     .block-container {padding-top: 1rem; padding-bottom: 0rem;}
@@ -36,16 +37,10 @@ st.markdown("""
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; margin-bottom: 10px; }
     .stChatMessage[data-testid="stChatMessage-user"] { background: #f0f0f0; color: #000000; }
     .stChatMessage[data-testid="stChatMessage-assistant"] { background: #D6E9FE; color: #000000; }
-    div[data-testid="column"] button {
-        border: none;
-        background-color: transparent;
-        font-size: 1.4rem;
-        margin-top: 18px;
-    }
     @keyframes bounceIn {
-        0%   { transform: scale(0.7); opacity: 0; }
-        60%  { transform: scale(1.1); opacity: 1; }
-        80%  { transform: scale(0.95); }
+        0% { transform: scale(0.7); opacity: 0; }
+        60% { transform: scale(1.1); opacity: 1; }
+        80% { transform: scale(0.95); }
         100% { transform: scale(1); }
     }
     .carfind-logo {
@@ -63,37 +58,13 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# Chat input + Clear Chat button
-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:
-            # Delete messages
-            user_doc_ref = db.collection("users").document(user_id)
-            message_collection = user_doc_ref.collection("messages").stream()
-            for msg in message_collection:
-                msg.reference.delete()
-            # Delete user doc
-            user_doc_ref.delete()
-
-            # Clear session and rerun
-            st.session_state.clear()
-            st.rerun()
-
-        except Exception as e:
-            st.error(f"Failed to clear chat: {e}")
-
-# Create or load thread ID from Firebase
+# 🔁 Firebase Chat Functions
 def get_or_create_thread_id():
     doc_ref = db.collection("users").document(user_id)
     doc = doc_ref.get()
     if doc.exists:
         return doc.to_dict()["thread_id"]
     else:
-        client = OpenAI(api_key=openai_key)
         thread = client.beta.threads.create()
         doc_ref.set({
             "thread_id": thread.id,
@@ -101,7 +72,6 @@ def get_or_create_thread_id():
         })
         return thread.id
 
-# Save message to Firestore
 def save_message(role, content):
     db.collection("users").document(user_id).collection("messages").add({
         "role": role,
@@ -109,16 +79,12 @@ def save_message(role, content):
         "timestamp": firestore.SERVER_TIMESTAMP
     })
 
-# Display past chat messages (latest at the top)
 def display_chat_history():
     messages = db.collection("users").document(user_id).collection("messages") \
         .order_by("timestamp").stream()
 
-    messages_list = list(messages)[::-1]  # Reverse to show latest first
-
     assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
-
-    for msg in messages_list:
+    for msg in list(messages)[::-1]:
         data = msg.to_dict()
         if data["role"] == "user":
             st.markdown(
@@ -132,29 +98,38 @@ def display_chat_history():
                 unsafe_allow_html=True
             )
 
-# OpenAI assistant interaction
-if openai_key and assistant_id:
-    client = OpenAI(api_key=openai_key)
-    thread_id = get_or_create_thread_id()
+# 🧭 Tabs: AI Chat | Car Identifier
+tab1, tab2 = st.tabs(["🤖 AI Chat", "🚗 What car is that?"])
+
+# 🤖 AI Chat Tab
+with tab1:
+    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:
+                user_doc_ref = db.collection("users").document(user_id)
+                for msg in user_doc_ref.collection("messages").stream():
+                    msg.reference.delete()
+                user_doc_ref.delete()
+                st.session_state.clear()
+                st.rerun()
+            except Exception as e:
+                st.error(f"Failed to clear chat: {e}")
 
+    thread_id = get_or_create_thread_id()
     display_chat_history()
 
     if user_input:
-        client.beta.threads.messages.create(
-            thread_id=thread_id, role="user", content=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
-            )
+            run = client.beta.threads.runs.create(thread_id=thread_id, assistant_id=assistant_id)
             while True:
-                run_status = client.beta.threads.runs.retrieve(
-                    thread_id=thread_id,
-                    run_id=run.id
-                )
+                run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
                 if run_status.status == "completed":
                     break
                 time.sleep(1)
@@ -163,8 +138,74 @@ if openai_key and assistant_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)  # Ensure Firestore timestamps sort properly
+        time.sleep(0.5)
         st.rerun()
-else:
-    st.error("⚠️ OpenAI key or Assistant ID not found. Please set them as Hugging Face secrets.")
+
+# 🚗 What car is that? Tab
+with tab2:
+    st.header("🖼️ Upload a Car Image for Identification")
+    uploaded_image = st.file_uploader("Upload an image of a car", type=["jpg", "jpeg", "png"])
+
+    if uploaded_image:
+        col1, col2 = st.columns([1.2, 1.8])
+
+        with col1:
+            image = Image.open(uploaded_image)
+            st.image(image, caption="Uploaded Image", use_container_width=True)
+
+        with col2:
+            try:
+                image_thread = client.beta.threads.create()
+                file_response = client.files.create(file=uploaded_image, purpose="assistants")
+
+                client.beta.threads.messages.create(
+                    thread_id=image_thread.id,
+                    role="user",
+                    content=[
+                        {
+                            "type": "image_file",
+                            "image_file": {"file_id": file_response.id}
+                        },
+                        {
+                            "type": "text",
+                            "text": "Please identify this car from the image."
+                        }
+                    ]
+                )
+
+                run = client.beta.threads.runs.create(
+                    thread_id=image_thread.id,
+                    assistant_id=assistant_id
+                )
+
+                with st.spinner("🔍 Analyzing image and identifying the car..."):
+                    while True:
+                        run_status = client.beta.threads.runs.retrieve(
+                            thread_id=image_thread.id,
+                            run_id=run.id
+                        )
+                        if run_status.status == "completed":
+                            break
+                        time.sleep(1)
+
+                messages = client.beta.threads.messages.list(thread_id=image_thread.id)
+                assistant_message = messages.data[0].content[0].text.value
+
+                st.success("✅ Identification Complete")
+                st.markdown(f"""
+                    <div style='
+                        font-size: 15px;
+                        background-color: #1a1f25;
+                        color: #ffffff;
+                        padding: 18px 20px;
+                        border-radius: 10px;
+                        border-left: 6px solid #4CAF50;
+                        box-shadow: 0 0 8px rgba(0, 0, 0, 0.2);
+                    '>
+                        <span style='font-size: 18px; font-weight: bold;'>🔍 Identified Car Details:</span><br><br>
+                        {assistant_message}
+                    </div>
+                """, unsafe_allow_html=True)
+
+            except Exception as e:
+                st.error(f"❌ Error during image analysis: {str(e)}")