Update app.py

app.py

@@ -139,11 +139,22 @@ 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.strip()
-                            if not any(reply in msg["content"] or msg["content"] in reply for msg in st.session_state.messages if msg["role"] == "assistant"):
-                                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)
+                            raw_reply = m.content[0].text.value.strip()
+
+                            # ✅ Clean formatting and strip noise
+                            cleaned_reply = re.sub(r'\[source[\d\.]+\]', '', raw_reply)
+                            cleaned_reply = re.sub(r'\n{3,}', '\n\n', cleaned_reply)
+                            cleaned_reply = re.sub(r'\*\*+', '**', cleaned_reply)
+                            cleaned_reply = re.sub(r'(?<=\n)([-*]\s[^\n]+)', r'\1\n', cleaned_reply)
+                            cleaned_reply = cleaned_reply.strip()
+
+                            if not any(cleaned_reply in msg["content"] or msg["content"] in cleaned_reply for msg in st.session_state.messages if msg["role"] == "assistant"):
+                                st.session_state.messages.append({"role": "assistant", "content": cleaned_reply})
+
+                                image_matches = re.findall(
+                                    r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
+                                    cleaned_reply
+                                )
                                 if image_matches:
                                     st.session_state.image_url = image_matches[0]
                                     st.session_state.image_updated = True
@@ -168,6 +179,7 @@ with tab1:
             except Exception as e:
                 st.error(f"🖼️ Failed to load image: {e}")
 
+
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"