V3

src/ui_helpers.py

@@ -1,104 +0,0 @@
-import os
-import base64
-import streamlit as st
-
-
-def get_image_base64(image_path):
-    """Encode an image file to base64."""
-    try:
-        if os.path.exists(image_path):
-            with open(image_path, "rb") as img_file:
-                return base64.b64encode(img_file.read()).decode()
-        else:
-            st.warning(f"Image not found: {image_path}")
-            return None
-    except Exception as e:
-        st.warning(f"Error loading image: {e}")
-        return None
-
-
-def load_avatars():
-    """Load avatar images for user and assistant."""
-    # Define paths to company logo images
-    user_logo_path = os.path.join("src/assets", "icon3.png")
-    assistant_logo_path = os.path.join("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 initialize_session_state():
-    """Initialize all necessary session state variables."""
-    import uuid
-    from datetime import datetime
-    
-    if 'history' not in st.session_state:
-        st.session_state.history = []
-    if 'consultation_id' not in st.session_state:
-        st.session_state.consultation_id = str(uuid.uuid4())[:8]
-    if 'use_rag' not in st.session_state:
-        st.session_state.use_rag = True
-    if 'processing' not in st.session_state:
-        st.session_state.processing = False
-    if 'show_report_form' not in st.session_state:
-        st.session_state.show_report_form = False
-    if 'report_step' not in st.session_state:
-        st.session_state.report_step = 0
-    if 'patient_info' not in st.session_state:
-        st.session_state.patient_info = {"name": "", "age": "", "gender": ""}
-    if 'pdf_data' not in st.session_state:
-        st.session_state.pdf_data = None
-    if 'show_email_form' not in st.session_state:
-        st.session_state.show_email_form = False
-
-
-def display_chat_history(history, user_avatar, assistant_avatar):
-    """Display the chat history with proper formatting."""
-    from text_processors import remove_reasoning_and_sources, has_meaningful_content, clean_explanation
-    
-    for message in 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 display_typing_indicator(assistant_avatar):
-    """Display a typing indicator while processing a response."""
-    from styles import get_typing_indicator_html
-    
-    with st.container():
-        col1, col2 = st.columns([10, 2])
-        with col1:
-            with st.chat_message("assistant", avatar=assistant_avatar):
-                st.markdown(get_typing_indicator_html(), unsafe_allow_html=True)