Update src/chat_display.py

src/chat_display.py

@@ -5,6 +5,7 @@ This module handles the chat display components and rendering.
 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
 
 
 def get_avatars():
@@ -31,11 +32,26 @@ def get_avatars():
 
 def display_chat_history():
     """
-    Display the chat history from session state.
+    Display the chat history from the database.
     """
     user_avatar, assistant_avatar = get_avatars()
     
-    for message in st.session_state.history:
+    # Get full history from database
+    history = get_full_history()
+    
+    # Show message count for debugging
+    # st.caption(f"Displaying {len(history)} messages")
+    
+    # Option to display a limited number of recent messages (UI performance)
+    # Configure this as needed based on your UI requirements
+    max_display_messages = 20
+    display_history = history[-max_display_messages:] if len(history) > max_display_messages else history
+    
+    # Show indicator if some messages are hidden
+    if len(history) > max_display_messages:
+        st.caption(f"Showing most recent {max_display_messages} of {len(history)} messages")
+    
+    for message in display_history:
         if message["role"] == "user":
             # Right-aligned container for user messages
             with st.container():