Create chat_display.py

src/chat_display.py

@@ -0,0 +1,96 @@
+"""
+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
+
+
+def get_avatars():
+    """
+    Get avatar images for the user and assistant.
+    
+    Returns:
+        tuple: (user_avatar, assistant_avatar) - URLs or emoji fallbacks
+    """
+    # Define paths to your company logo images
+    user_logo_path = "src/assets/icon3.png"
+    assistant_logo_path = "src/assets/logo.png"
+
+    # Get base64 encoded images or use default emoji as fallback
+    user_logo_base64 = get_image_base64(user_logo_path)
+    assistant_logo_base64 = get_image_base64(assistant_logo_path)
+
+    # Create avatar URLs with base64 data
+    user_avatar = f"data:image/png;base64,{user_logo_base64}" if user_logo_base64 else "👤"
+    assistant_avatar = f"data:image/png;base64,{assistant_logo_base64}" if assistant_logo_base64 else "🤖"
+    
+    return user_avatar, assistant_avatar
+
+
+def display_chat_history():
+    """
+    Display the chat history from session state.
+    """
+    user_avatar, assistant_avatar = get_avatars()
+    
+    for message in st.session_state.history:
+        if message["role"] == "user":
+            # Right-aligned container for user messages
+            with st.container():
+                col1, col2 = st.columns([2, 10])
+                with col2:
+                    with st.chat_message("user", avatar=user_avatar):
+                        st.write(message["content"])
+        else:
+            # Left-aligned container for assistant messages
+            with st.container():
+                col1, col2 = st.columns([10, 2])
+                with col1:
+                    with st.chat_message("assistant", avatar=assistant_avatar):
+                        # Display the response text without reasoning or sources sections
+                        cleaned_response = remove_reasoning_and_sources(message["content"])
+                        st.markdown(cleaned_response)
+                        
+                        # Only display the explanation in an expander if it exists AND has actual content
+                        if message.get("explanation") and has_meaningful_content(message.get("explanation")):
+                            # Clean up the explanation text
+                            cleaned_explanation = clean_explanation(message["explanation"])
+                            
+                            with st.expander("Show Reasoning"):
+                                st.markdown(cleaned_explanation)
+                                if message.get("evidence"):
+                                    st.markdown("---")
+                                    st.markdown("**Legend:** 🔓 = Open Access (full text available)")
+
+
+def show_typing_indicator():
+    """
+    Display a typing indicator when the system is processing a response.
+    """
+    if st.session_state.processing:
+        user_avatar, assistant_avatar = get_avatars()
+        with st.container():
+            col1, col2 = st.columns([10, 2])
+            with col1:
+                with st.chat_message("assistant", avatar=assistant_avatar):
+                    st.markdown("""
+                    <div class="typing-indicator">
+                        <span></span>
+                        <span></span>
+                        <span></span>
+                    </div>
+                    <p style="margin: 0; color: rgba(255,255,255,0.6) !important; font-size: 0.8rem;">Analyzing your query...</p>
+                    """, unsafe_allow_html=True)
+
+
+def display_legal_disclaimer():
+    """
+    Display the legal disclaimer at the bottom of the page.
+    """
+    st.markdown("""
+    <div class="footer-text">
+        For informational purposes only. Not a substitute for professional medical advice.
+    </div>
+    """, unsafe_allow_html=True)