Update app.py

app.py

@@ -30,8 +30,6 @@ if st.session_state.theme == "light":
         """
         <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
@@ -41,8 +39,6 @@ else:
         """
         <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
@@ -74,10 +70,17 @@ else:
 
     st.divider()
 
-    # Display chat history with colored bubbles
+    # Display chat history with avatars
     for i, msg in enumerate(st.session_state["messages"]):
-        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")
+        avatar_url = "https://www.carfind.co.za/images/Carfind-Icon.svg" if msg["role"] == "assistant" else None
+        message(
+            msg["content"],
+            is_user=(msg["role"] == "user"),
+            key=str(i),
+            avatar_style="adventurer" if msg["role"] == "user" else None,
+            seed="user-seed" if msg["role"] == "user" else None,
+            avatar_url=avatar_url
+        )
 
     col_input, col_clear = st.columns([10, 1])
 
@@ -89,55 +92,59 @@ else:
             st.session_state.messages = []
             st.rerun()
 
-    # Handle OpenAI assistant
+    # Handle OpenAI assistant (with loop protection)
     if user_input and openai_key:
-        st.session_state.messages.append({"role": "user", "content": user_input})
-
-        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,
-                    assistant_id=ASSISTANT_ID
-                )
-
-                while True:
-                    run_status = client.beta.threads.runs.retrieve(
+        # Avoid duplicate messages triggering loops
+        if len(st.session_state.messages) == 0 or user_input != st.session_state.messages[-1]["content"]:
+            st.session_state.messages.append({"role": "user", "content": user_input})
+
+            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,
-                        run_id=run.id
+                        role="user",
+                        content=user_input
                     )
-                    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
+                    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)
+
+                    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)
+                    # Prevent infinite repetition by ignoring if assistant echoes input
+                    if assistant_reply.strip().lower() != user_input.strip().lower():
+                        st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
+                        save_transcript(st.session_state.messages)
 
-                st.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.")
+        st.error("OpenAI key not found. Please ensure it is set as a Hugging Face secret.")