Update app.py

app.py

@@ -139,30 +139,11 @@ 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":
-                            raw_reply = m.content[0].text.value.strip()
-
-                            # ✅ Format cleanup and enhancement
-                            cleaned_reply = raw_reply
-                            cleaned_reply = re.sub(r'\[source[\d\.]+\]', '', cleaned_reply)
-                            cleaned_reply = re.sub(r'(?i)^document references', '### 📄 Document References', cleaned_reply, flags=re.MULTILINE)
-                            cleaned_reply = re.sub(r'(?i)^key points', '### 🔑 Key Points', cleaned_reply, flags=re.MULTILINE)
-                            cleaned_reply = re.sub(r'(?i)^some possible questions', '### ❓ Some Possible Questions', cleaned_reply, flags=re.MULTILINE)
-                            cleaned_reply = re.sub(r'(?i)^other references', '### 📚 Other References', cleaned_reply, flags=re.MULTILINE)
-                            cleaned_reply = re.sub(r'(?i)^section[:]', '**Section:**', cleaned_reply, flags=re.MULTILINE)
-                            cleaned_reply = re.sub(r'(Page \d+)', r'**\1**', cleaned_reply)
-                            cleaned_reply = re.sub(r'(https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png)', r'[🖼️ View Referenced Image](\1)', cleaned_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
-                                )
+                            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)
                                 if image_matches:
                                     st.session_state.image_url = image_matches[0]
                                     st.session_state.image_updated = True
@@ -187,7 +168,6 @@ 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"