Update app.py

app.py

@@ -2,20 +2,18 @@ import streamlit as st
 from openai import OpenAI
 import time
 import os
+import pandas as pd
 import datetime
 
-# Get secrets from environment
 generated_user = os.getenv("User")
 generated_password = os.getenv("Password")
 openai_key = os.getenv("openai_key")
 assistant_id = os.getenv("ASSISTANT_ID")
 
-# Setup Streamlit app
 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")
 
-# Style tweaks for chat bubbles
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
@@ -24,19 +22,29 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Transcript saving function
-def save_transcript(thread_id, messages):
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-    filename = f"transcript_{thread_id}_{timestamp}.txt"
-    with open(filename, "w", encoding="utf-8") as file:
-        file.write(f"Chat Transcript - Thread: {thread_id} - Date: {timestamp}\n")
-        file.write("------------------------------------------------------\n\n")
-        for msg in reversed(messages):
-            role = "User" if msg.role == "user" else "Assistant"
-            content = msg.content[0].text.value
-            file.write(f"{role}: {content}\n\n")
-
-# Authentication
+# Load existing transcript excel or create one
+transcript_file = "transcripts.xlsx"
+
+def append_to_transcript(thread_id, role, message):
+    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    new_entry = {
+        "Timestamp": timestamp,
+        "Thread_ID": thread_id,
+        "Role": role,
+        "Message": message
+    }
+
+    try:
+        if os.path.exists(transcript_file):
+            df = pd.read_excel(transcript_file)
+            df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
+        else:
+            df = pd.DataFrame([new_entry])
+
+        df.to_excel(transcript_file, index=False)
+    except Exception as e:
+        st.error(f"Failed to log conversation: {str(e)}")
+
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -54,7 +62,6 @@ if not st.session_state.authenticated:
         else:
             st.error("Incorrect username or password. Please try again.")
 
-# Main chat interface
 else:
     st.divider()
 
@@ -82,6 +89,7 @@ else:
             client.beta.threads.messages.create(
                 thread_id=st.session_state["thread_id"], role="user", content=user_input
             )
+            append_to_transcript(st.session_state["thread_id"], "User", user_input)
 
             try:
                 with st.spinner("Thinking and typing... 💭"):
@@ -101,29 +109,28 @@ else:
                     thread_id=st.session_state["thread_id"]
                 )
 
-                # Save the transcript automatically
-                save_transcript(st.session_state["thread_id"], messages_response.data)
-
                 assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
 
-                # Display entire conversation in proper order
                 for msg in reversed(messages_response.data):
+                    content = msg.content[0].text.value
                     if msg.role == "user":
                         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"👤 <strong>You:</strong> {content}"
                             f"</div>", 
                             unsafe_allow_html=True
                         )
                     else:
                         st.markdown(
                             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"{assistant_icon_html} <strong>Carfind Assistant:</strong> {content}"
                             f"</div>", 
                             unsafe_allow_html=True
                         )
+                        append_to_transcript(st.session_state["thread_id"], "Assistant", content)
 
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
+
     else:
         st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")