Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 import time
 import re
 import requests
-import json
 from PIL import Image
 from io import BytesIO
 from openai import OpenAI
@@ -133,13 +132,18 @@ with tab1:
                     messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
                     for m in reversed(messages.data):
                         if m.role == "assistant":
-                            reply = m.content[0].text.value
-                            st.session_state.messages.append({"role": "assistant", "content": reply})
-                            image_matches = re.findall(
-                                r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
-                                reply
+                            reply = m.content[0].text.value.strip()
+                            is_duplicate = any(
+                                reply in msg["content"] or msg["content"] in reply
+                                for msg in st.session_state.messages if msg["role"] == "assistant"
                             )
-                            st.session_state.image_urls = image_matches
+                            if not is_duplicate:
+                                st.session_state.messages.append({"role": "assistant", "content": reply})
+                                image_matches = re.findall(
+                                    r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
+                                    reply
+                                )
+                                st.session_state.image_urls = image_matches
                             break
                 else:
                     st.error("❌ Assistant failed to respond.")
@@ -162,7 +166,6 @@ with tab1:
                     st.image(img, caption=f"📷 {raw_url.split('/')[-1]}", use_container_width=True)
                 except Exception:
                     continue
-                    
 
 
 # ------------------ Tab 2: Visual Reference Search ------------------