document_reader/utils/session_state.py

import streamlit as st


def reset_state(upload_id=None):
    if upload_id:
        st.session_state['uploads'][upload_id] = {
            'file_groups': None,
            'values_raw': None,
            'values_display': None,
            'results_transformed': None,
            'temp_dir': None,
            'application_form': None,
            'memo': None
        }
        st.session_state['current_upload'] = upload_id
    else:
        current = st.session_state.get('current_upload')
        if current:
            st.session_state['uploads'][current]['values_raw'] = None
            st.session_state['uploads'][current]['values_display'] = None
            st.session_state['uploads'][current]['results_transformed'] = None
            # st.session_state['uploads'][current]['application_form'] = None
            # st.session_state['uploads'][current]['temp_dir'] = None