utils/session_state.py

# src/utils/session_state.py
import streamlit as st

def initialize_session_state():
    """Initialize session state variables"""
    session_vars = {
        'chat_history': [],
        'current_chat': False,
        'selected_docs': [],
        'qa_system': None,
        'documents_initialized': False,
        'processed_files': set()
    }
    
    for var, default_value in session_vars.items():
        if var not in st.session_state:
            st.session_state[var] = default_value