Update app.py

app.py

@@ -141,8 +141,16 @@ with tab1:
                         if m.role == "assistant":
                             raw_reply = m.content[0].text.value.strip()
 
-                            # ✅ Clean formatting and strip noise
-                            cleaned_reply = re.sub(r'\[source[\d\.]+\]', '', raw_reply)
+                            # ✅ 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)