Update app.py

app.py

@@ -4,18 +4,18 @@ import time
 import os
 import datetime
 
-# Load secrets
+# 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
+# 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")
 
-# Styling
+# Style tweaks for chat bubbles
 st.markdown("""
     <style>
     .stChatMessage { max-width: 85%; border-radius: 12px; padding: 8px; }
@@ -24,9 +24,17 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Create transcripts folder
-if not os.path.exists("chat_logs"):
-    os.makedirs("chat_logs")
+# 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
 if "authenticated" not in st.session_state:
@@ -44,14 +52,14 @@ if not st.session_state.authenticated:
             time.sleep(1)
             st.rerun()
         else:
-            st.error("Incorrect username or password.")
+            st.error("Incorrect username or password. Please try again.")
 
+# Main chat interface
 else:
     st.divider()
 
     if "thread_id" not in st.session_state:
         st.session_state["thread_id"] = None
-        st.session_state["chat_history"] = []
 
     input_col, clear_col = st.columns([8, 1])
     with input_col:
@@ -60,7 +68,6 @@ else:
     with clear_col:
         if st.button("🗑️", help="Clear Chat"):
             st.session_state["thread_id"] = None
-            st.session_state["chat_history"] = []
             st.success("Chat cleared.")
             st.rerun()
 
@@ -68,19 +75,16 @@ else:
         client = OpenAI(api_key=openai_key)
 
         if user_input:
-            # Start new thread if needed
             if not st.session_state["thread_id"]:
                 thread = client.beta.threads.create()
                 st.session_state["thread_id"] = thread.id
 
-            # Add user message to OpenAI thread
             client.beta.threads.messages.create(
                 thread_id=st.session_state["thread_id"], role="user", content=user_input
             )
-            st.session_state["chat_history"].append(("user", user_input))
 
             try:
-                with st.spinner("Thinking... 💭"):
+                with st.spinner("Thinking and typing... 💭"):
                     run = client.beta.threads.runs.create(
                         thread_id=st.session_state["thread_id"], assistant_id=assistant_id
                     )
@@ -93,40 +97,33 @@ else:
                             break
                         time.sleep(1)
 
-                # Get assistant response
                 messages_response = client.beta.threads.messages.list(
                     thread_id=st.session_state["thread_id"]
                 )
 
-                # The latest assistant reply
-                assistant_reply = messages_response.data[0].content[0].text.value
-                st.session_state["chat_history"].append(("assistant", assistant_reply))
+                # Save the transcript automatically
+                save_transcript(st.session_state["thread_id"], messages_response.data)
 
-                # Save chat log after each response
-                timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-                with open(f"chat_logs/chat_{timestamp}.txt", "w") as log_file:
-                    for role, content in st.session_state["chat_history"]:
-                        log_file.write(f"{role.capitalize()}: {content}\n")
+                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):
+                    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"</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"</div>", 
+                            unsafe_allow_html=True
+                        )
 
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
-
-        # Assistant icon inline
-        assistant_icon_html = "<img src='https://www.carfind.co.za/images/Carfind-Icon.svg' width='22' style='vertical-align:middle;'/>"
-
-        # Display chat history top-down
-        for role, content in st.session_state["chat_history"]:
-            if role == "user":
-                st.markdown(
-                    f"<div style='background-color:#f0f0f0; color:#000000; padding:10px; border-radius:8px; margin:5px 0;'>"
-                    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> {content}"
-                    f"</div>", unsafe_allow_html=True
-                )
     else:
-        st.error("⚠️ OpenAI key or Assistant ID not found. Make sure both are set as Hugging Face secrets.")
+        st.error("⚠️ OpenAI key or Assistant ID not found. Please ensure both are set as Hugging Face secrets.")