Update app.py

app.py

@@ -1,72 +1,105 @@
 import streamlit as st
 from openai import OpenAI
 import time
+import datetime
 
-st.set_page_config(page_title="Carfind.co.za Ai Assistant")
+# Set your API key as a variable (securely from env in HF or paste here)
+openai_key = "YOUR_OPENAI_API_KEY_HERE"
 
-st.title("Carfind.co.za Ai Assistant")
+st.set_page_config(page_title="Carfind.co.za AI Assistant")
+
+st.title("Carfind.co.za AI Assistant")
 st.caption("Chat with Carfind.co.za and Find your Next Car Fast")
 
-# Sidebar for API Key input
-with st.sidebar:
-    OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
-
-if OPENAI_API_KEY:
-    client = OpenAI(api_key=OPENAI_API_KEY)
-else:
-    st.error("Please enter your C2 Group of Technologies Access Key to continue.")
-    st.stop()
-
-ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
-
-# Initialize session state for chat history
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
-
-# Display chat history
-for message in st.session_state.messages:
-    role, content = message["role"], message["content"]
-    st.chat_message(role).write(content)
-
-# Process user input
-if prompt := st.chat_input():
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
-
-    try:
-        # Create a new thread for conversation
-        thread = client.beta.threads.create()
-        thread_id = thread.id
-
-        # Send user message to OpenAI API
-        client.beta.threads.messages.create(
-            thread_id=thread_id,
-            role="user",
-            content=prompt
-        )
-
-        # Run the assistant to generate a response
-        run = client.beta.threads.runs.create(
-            thread_id=thread_id,
-            assistant_id=ASSISTANT_ID
-        )
-
-        # Wait for response
-        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)
-
-        # Retrieve assistant response
-        messages = client.beta.threads.messages.list(thread_id=thread_id)
-        assistant_message = messages.data[0].content[0].text.value
-
-        # Display assistant's response
-        st.chat_message("assistant").write(assistant_message)
-
-        # Store in session state
-        st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-    
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+# Simple login system
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+
+if not st.session_state.authenticated:
+    st.subheader("Login")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+
+    if st.button("Login"):
+        if username == "andrew@lortechnologies.com" and password == "Pass.123":
+            st.session_state.authenticated = True
+            st.success("Login successful!")
+            st.experimental_rerun()
+        else:
+            st.error("Invalid username or password.")
+
+# Show chat interface only if logged in
+if st.session_state.authenticated:
+    # Sidebar with clear chat button and chat log link
+    with st.sidebar:
+        if st.button("🧹 Clear Chat"):
+            st.session_state.messages = []
+            st.success("Chat history cleared.")
+        st.markdown("---")
+        st.markdown("**Chat logs are saved locally in `chat_logs.txt`.**")
+
+    # Initialize OpenAI client
+    client = OpenAI(api_key=openai_key)
+
+    ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
+
+    # Initialize session state for chat history
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = []
+
+    # Display chat history
+    for message in st.session_state.messages:
+        role, content = message["role"], message["content"]
+        st.chat_message(role).write(content)
+
+    # Helper function to save chat transcripts
+    def save_transcript(messages):
+        with open("chat_logs.txt", "a") as f:
+            f.write(f"\n--- New Chat ({datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) ---\n")
+            for msg in messages:
+                f.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
+            f.write("--- End Chat ---\n")
+
+    # Process user input
+    if prompt := st.chat_input("Type your message here..."):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        st.chat_message("user").write(prompt)
+
+        try:
+            # Create a new thread
+            thread = client.beta.threads.create()
+            thread_id = thread.id
+
+            # Send user message
+            client.beta.threads.messages.create(
+                thread_id=thread_id,
+                role="user",
+                content=prompt
+            )
+
+            # Run assistant
+            run = client.beta.threads.runs.create(
+                thread_id=thread_id,
+                assistant_id=ASSISTANT_ID
+            )
+
+            # Wait for completion
+            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)
+
+            # Get assistant response
+            messages = client.beta.threads.messages.list(thread_id=thread_id)
+            assistant_message = messages.data[0].content[0].text.value
+
+            # Display and store response
+            st.chat_message("assistant").write(assistant_message)
+            st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+
+            # Save conversation transcript every interaction
+            save_transcript(st.session_state.messages)
+
+        except Exception as e:
+            st.error(f"Error: {str(e)}")