session_manager.py

# Initialize a global variable to hold the session state reference
session_state = None

def set_session_state(state):
    global session_state
    session_state = state

def get_session_state():
    return session_state