| """The App module is the main entry point for the application. |
| |
| Run `streamlit run app.py` to start the app. |
| """ |
|
|
| import pandas as pd |
| import streamlit as st |
| from streamlit_option_menu import option_menu |
|
|
| from src.load import load_context |
| from src.subpages import ( |
| DebugPage, |
| FindDuplicatesPage, |
| HomePage, |
| LossesPage, |
| LossySamplesPage, |
| MetricsPage, |
| MisclassifiedPage, |
| Page, |
| ProbingPage, |
| RandomSamplesPage, |
| RawDataPage, |
| ) |
| from src.subpages.attention import AttentionPage |
| from src.subpages.hidden_states import HiddenStatesPage |
| from src.subpages.inspect import InspectPage |
| from src.utils import classmap |
|
|
| sts = st.sidebar |
| st.set_page_config( |
| layout="wide", |
| page_title="Error Analysis", |
| page_icon="🏷️", |
| ) |
|
|
|
|
| def _show_menu(pages: list[Page]) -> int: |
| with st.sidebar: |
| page_names = [p.name for p in pages] |
| page_icons = [p.icon for p in pages] |
| selected_menu_item = st.session_state.active_page = option_menu( |
| menu_title="ExplaiNER", |
| options=page_names, |
| icons=page_icons, |
| menu_icon="layout-wtf", |
| default_index=0, |
| ) |
| return page_names.index(selected_menu_item) |
| assert False |
|
|
|
|
| def _initialize_session_state(pages: list[Page]): |
| if "active_page" not in st.session_state: |
| for page in pages: |
| st.session_state.update(**page._get_widget_defaults()) |
| st.session_state.update(st.session_state) |
|
|
|
|
| def _write_color_legend(context): |
| def style(x): |
| return [f"background-color: {rgb}; opacity: 1;" for rgb in colors] |
|
|
| labels = list(set([lbl.split("-")[1] if "-" in lbl else lbl for lbl in context.labels])) |
| colors = [st.session_state.get(f"color_{lbl}", "#000000") for lbl in labels] |
|
|
| color_legend_df = pd.DataFrame( |
| [classmap[l] for l in labels], columns=["label"], index=labels |
| ).T |
| st.sidebar.write( |
| color_legend_df.T.style.apply(style, axis=0).set_properties( |
| **{"color": "white", "text-align": "center"} |
| ) |
| ) |
|
|
|
|
| def main(): |
| """The main entry point for the application.""" |
| pages: list[Page] = [ |
| HomePage(), |
| AttentionPage(), |
| HiddenStatesPage(), |
| ProbingPage(), |
| MetricsPage(), |
| LossySamplesPage(), |
| LossesPage(), |
| MisclassifiedPage(), |
| RandomSamplesPage(), |
| FindDuplicatesPage(), |
| InspectPage(), |
| RawDataPage(), |
| DebugPage(), |
| ] |
|
|
| _initialize_session_state(pages) |
|
|
| selected_page_idx = _show_menu(pages) |
| selected_page = pages[selected_page_idx] |
|
|
| if isinstance(selected_page, HomePage): |
| selected_page.render() |
| return |
|
|
| if "model_name" not in st.session_state: |
| |
| st.error("Setup not complete. Please click on 'Home / Setup in left menu bar'") |
| return |
|
|
| context = load_context(**st.session_state) |
| _write_color_legend(context) |
| selected_page.render(context) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
