Update app.py

app.py

@@ -3,38 +3,36 @@ from openai import OpenAI
 import time
 import os
 import pandas as pd
+import uuid
 from datetime import datetime
 
-# Load environment variables
+# Firebase setup
+import firebase_admin
+from firebase_admin import credentials, firestore
+
+if not firebase_admin._apps:
+    cred = credentials.Certificate("firebase-service-account.json")
+    firebase_admin.initialize_app(cred)
+
+db = firestore.client()
+
+# OpenAI setup
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
 
-# Set up transcript log
-transcript_file = "transcripts.xlsx"
-if not os.path.exists(transcript_file):
-    df = pd.DataFrame(columns=["timestamp", "thread_id", "role", "message"])
-    df.to_excel(transcript_file, index=False)
-
-def append_to_transcript(thread_id, role, message):
-    df_existing = pd.read_excel(transcript_file)
-    new_entry = pd.DataFrame({
-        "timestamp": [datetime.now().strftime("%Y-%m-%d %H:%M:%S")],
-        "thread_id": [thread_id],
-        "role": [role],
-        "message": [message]
-    })
-    updated_df = pd.concat([df_existing, new_entry], ignore_index=True)
-    updated_df.to_excel(transcript_file, index=False)
-
-# Configure page
 st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 
-# Inject custom styling
+# Session + Firebase 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
 st.markdown("""
     <style>
     .block-container {padding-top: 1rem; padding-bottom: 0rem;}
     header {visibility: hidden;}
-    .st-emotion-cache-18ni7ap {visibility: hidden;}
     .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; }
@@ -44,100 +42,129 @@ st.markdown("""
         font-size: 1.4rem;
         margin-top: 18px;
     }
-
-    /* Logo bounce animation */
     @keyframes bounceIn {
         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 {
         animation: bounceIn 0.6s ease-out;
     }
     </style>
 """, unsafe_allow_html=True)
 
-# Initialize session
-if "thread_id" not in st.session_state:
-    st.session_state["thread_id"] = None
-
-# Carfind logo + powered by text (animated)
-st.markdown(
-    """
+st.markdown("""
     <div style='text-align: center; margin-top: 20px; margin-bottom: -10px;'>
         <span style='display: inline-flex; align-items: center; gap: 8px;'>
             <img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='30' class='carfind-logo'/>
             <span style='font-size: 14px; color: gray;'>Powered by Carfind</span>
         </span>
     </div>
-    """,
-    unsafe_allow_html=True
-)
+""", unsafe_allow_html=True)
 
-# Chat input + inline clear button
+# 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"):
-        st.session_state["thread_id"] = None
-        st.rerun()
+        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()
 
-# Chat logic
+        except Exception as e:
+            st.error(f"Failed to clear chat: {e}")
+
+# Create or load thread ID from Firebase
+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,
+            "created_at": firestore.SERVER_TIMESTAMP
+        })
+        return thread.id
+
+# Save message to Firestore
+def save_message(role, content):
+    db.collection("users").document(user_id).collection("messages").add({
+        "role": role,
+        "content": 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:
+        data = msg.to_dict()
+        if data["role"] == "user":
+            st.markdown(
+                f"<div class='stChatMessage' data-testid='stChatMessage-user'>"
+                f"👤 <strong>You:</strong> {data['content']}</div>", unsafe_allow_html=True
+            )
+        else:
+            st.markdown(
+                f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>"
+                f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {data['content']}</div>",
+                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()
 
-    if user_input:
-        if not st.session_state["thread_id"]:
-            thread = client.beta.threads.create()
-            st.session_state["thread_id"] = thread.id
+    display_chat_history()
 
+    if user_input:
         client.beta.threads.messages.create(
-            thread_id=st.session_state["thread_id"], role="user", content=user_input
+            thread_id=thread_id, role="user", content=user_input
         )
-        append_to_transcript(st.session_state["thread_id"], "user", user_input)
+        save_message("user", user_input)
 
-        try:
-            with st.spinner("Thinking and typing... 💭"):
-                run = client.beta.threads.runs.create(
-                    thread_id=st.session_state["thread_id"], assistant_id=assistant_id
-                )
-                while True:
-                    run_status = client.beta.threads.runs.retrieve(
-                        thread_id=st.session_state["thread_id"], run_id=run.id
-                    )
-                    if run_status.status == "completed":
-                        break
-                    time.sleep(1)
-
-            messages_response = client.beta.threads.messages.list(
-                thread_id=st.session_state["thread_id"]
+        with st.spinner("Thinking and typing... 💭"):
+            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
+                )
+                if run_status.status == "completed":
+                    break
+                time.sleep(1)
 
-            assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
-            messages_sorted = sorted(messages_response.data, key=lambda x: x.created_at, reverse=True)
-
-            for msg in messages_sorted:
-                if msg.role == "user":
-                    st.markdown(
-                        f"<div class='stChatMessage' data-testid='stChatMessage-user'>"
-                        f"👤 <strong>You:</strong> {msg.content[0].text.value}</div>",
-                        unsafe_allow_html=True
-                    )
-                else:
-                    response_text = msg.content[0].text.value
-                    append_to_transcript(st.session_state["thread_id"], "assistant", response_text)
-                    st.markdown(
-                        f"<div class='stChatMessage' data-testid='stChatMessage-assistant'>"
-                        f"{assistant_icon_html} <strong>Carfind Assistant:</strong> {response_text}</div>",
-                        unsafe_allow_html=True
-                    )
+        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)
 
-        except Exception as e:
-            st.error(f"❌ An error occurred: {str(e)}")
+        time.sleep(0.5)  # Ensure Firestore timestamps sort properly
+        st.rerun()
 else:
-    st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")
+    st.error("⚠️ OpenAI key or Assistant ID not found. Please set them as Hugging Face secrets.")