Update app.py

app.py

@@ -10,11 +10,44 @@ generated_user = os.getenv("User")
 generated_password = os.getenv("Password")
 openai_key = os.getenv("openai_key")
 
-# Streamlit app configuration
+# Streamlit page config
 st.set_page_config(page_title="Carfind.co.za AI Assistant", layout="wide")
 st.title("🚗 Carfind.co.za AI Assistant")
 st.caption("Chat with Carfind.co.za and find your next car fast")
 
+# Theme toggle
+if "theme" not in st.session_state:
+    st.session_state.theme = "dark"
+
+toggle = st.toggle("Switch to Light Mode")
+if toggle:
+    st.session_state.theme = "light"
+else:
+    st.session_state.theme = "dark"
+
+if st.session_state.theme == "light":
+    st.markdown(
+        """
+        <style>
+        body { background-color: #f0f0f0; color: #000; }
+        .chat-bubble-user { background-color: #d1e7dd; color: #000; }
+        .chat-bubble-assistant { background-color: #f8d7da; color: #000; }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+else:
+    st.markdown(
+        """
+        <style>
+        body { background-color: #0e0e0e; color: #fff; }
+        .chat-bubble-user { background-color: #004085; color: #fff; }
+        .chat-bubble-assistant { background-color: #155724; color: #fff; }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
 # Initialize authentication state
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
@@ -41,11 +74,11 @@ else:
 
     st.divider()
 
-    # Display chat history
+    # Display chat history with colored bubbles
     for i, msg in enumerate(st.session_state["messages"]):
-        message(msg["content"], is_user=(msg["role"] == "user"), key=str(i))
+        bubble_class = "chat-bubble-user" if msg["role"] == "user" else "chat-bubble-assistant"
+        message(msg["content"], is_user=(msg["role"] == "user"), key=str(i), avatar_style="adventurer")
 
-    # Chat input with clear chat icon
     col_input, col_clear = st.columns([10, 1])
 
     with col_input:
@@ -54,56 +87,57 @@ else:
     with col_clear:
         if st.button("🗑️", help="Clear chat"):
             st.session_state.messages = []
-            st.experimental_rerun()
+            st.rerun()
 
     # Handle OpenAI assistant
     if user_input and openai_key:
         st.session_state.messages.append({"role": "user", "content": user_input})
 
-        client = OpenAI(api_key=openai_key)
-        ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
-
-        def save_transcript(messages):
-            with open("chat_logs.txt", "a") as log:
-                log.write(f"\n--- New Chat ({datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) ---\n")
-                for msg in messages:
-                    log.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
-                log.write("--- End Chat ---\n")
-
-        try:
-            thread = client.beta.threads.create()
-            thread_id = thread.id
-
-            client.beta.threads.messages.create(
-                thread_id=thread_id,
-                role="user",
-                content=user_input
-            )
-
-            run = client.beta.threads.runs.create(
-                thread_id=thread_id,
-                assistant_id=ASSISTANT_ID
-            )
-
-            while True:
-                run_status = client.beta.threads.runs.retrieve(
+        with st.spinner("Thinking and typing... 💬"):
+            client = OpenAI(api_key=openai_key)
+            ASSISTANT_ID = "asst_5gQR21fOsmHil11FGBzEArA7"
+
+            def save_transcript(messages):
+                with open("chat_logs.txt", "a") as log:
+                    log.write(f"\n--- New Chat ({datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}) ---\n")
+                    for msg in messages:
+                        log.write(f"{msg['role'].capitalize()}: {msg['content']}\n")
+                    log.write("--- End Chat ---\n")
+
+            try:
+                thread = client.beta.threads.create()
+                thread_id = thread.id
+
+                client.beta.threads.messages.create(
+                    thread_id=thread_id,
+                    role="user",
+                    content=user_input
+                )
+
+                run = client.beta.threads.runs.create(
                     thread_id=thread_id,
-                    run_id=run.id
+                    assistant_id=ASSISTANT_ID
                 )
-                if run_status.status == "completed":
-                    break
-                time.sleep(1)
 
-            messages_response = client.beta.threads.messages.list(thread_id=thread_id)
-            assistant_reply = messages_response.data[0].content[0].text.value
+                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)
+
+                messages_response = client.beta.threads.messages.list(thread_id=thread_id)
+                assistant_reply = messages_response.data[0].content[0].text.value
 
-            st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
-            save_transcript(st.session_state.messages)
+                st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
+                save_transcript(st.session_state.messages)
 
-            st.experimental_rerun()
+                st.rerun()
 
-        except Exception as e:
-            st.error(f"An error occurred: {str(e)}")
+            except Exception as e:
+                st.error(f"An error occurred: {str(e)}")
 
     elif not openai_key:
         st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")