Update src/chat_display.py

src/chat_display.py

@@ -6,6 +6,7 @@ import streamlit as st
 import os
 from utils import has_meaningful_content, remove_reasoning_and_sources, clean_explanation, get_image_base64
 from session_state import get_full_history
+import re
 
 
 def get_avatars():
@@ -68,6 +69,12 @@ def display_chat_history():
                             # Clean up the explanation text
                             cleaned_explanation = clean_explanation(message["explanation"])
                             
+                            # Additional cleaning to remove any source information from reasoning
+                            # Remove any sources/references sections
+                            cleaned_explanation = re.sub(r'(?i)(\n+\s*sources:|\n+\s*references:|\n+\s*\*{0,2}sources\*{0,2}:?|\n+\s*\*{0,2}references\*{0,2}:?|\n+\s*#{1,3}\s*sources|\n+\s*#{1,3}\s*references).*', '', cleaned_explanation, flags=re.DOTALL)
+                            # Remove any ## Sources heading and content
+                            cleaned_explanation = re.sub(r'#{1,3}\s+Sources.*', '', cleaned_explanation, flags=re.DOTALL)
+                            
                             with st.expander("Show Reasoning"):
                                 st.markdown(cleaned_explanation)