feat: set config page at the start

src/streamlit_app.py

@@ -6,6 +6,14 @@ import pandas as pd
 import streamlit as st
 import plotly.express as px
 
+# ---------------------------------------------------------------------
+# Page config (must be the first Streamlit command)
+# ---------------------------------------------------------------------
+st.set_page_config(
+    page_title="Custom Model Benchmarks",
+    layout="wide",
+)
+
 # ---------------------------------------------------------------------
 # Configuration
 # ---------------------------------------------------------------------
@@ -62,6 +70,17 @@ def load_raw_data():
 
     pearson_df.columns = [c.strip() for c in pearson_df.columns]
     mcc_df.columns = [c.strip() for c in mcc_df.columns]
+
+    # Optional: basic sanity check on required columns
+    # required_p = {"species", "datasets", "pearson correlation"}
+    # required_m = {"species", "datasets", "MCC"}
+    # missing_p = required_p - set(pearson_df.columns)
+    # missing_m = required_m - set(mcc_df.columns)
+    # if missing_p:
+    #     st.error(f"Pearson CSV missing columns: {missing_p}")
+    # if missing_m:
+    #     st.error(f"MCC CSV missing columns: {missing_m}")
+
     return pearson_df, mcc_df
 
 
@@ -216,17 +235,29 @@ def build_bar_df(
     )
 
 
+# ---------------------------------------------------------------------
+# UI helpers
+# ---------------------------------------------------------------------
+
+
+def sidebar_toggle(label: str, value: bool = False, key: str | None = None) -> bool:
+    """
+    Wrapper to use st.sidebar.toggle when available, otherwise fall back to checkbox.
+    This makes the app compatible with older Streamlit versions on Hugging Face.
+    """
+    toggle_fn = getattr(st.sidebar, "toggle", None)
+    if toggle_fn is not None:
+        return toggle_fn(label, value=value, key=key)
+    # Fallback for older Streamlit versions
+    return st.sidebar.checkbox(label, value=value, key=key)
+
+
 # ---------------------------------------------------------------------
 # Streamlit UI
 # ---------------------------------------------------------------------
 
 
 def main():
-    st.set_page_config(
-        page_title="Custom Model Benchmarks",
-        layout="wide",
-    )
-
     st.title("🧬 Custom Model Benchmarks")
     st.markdown(_INTRO)
     st.markdown(f"_Last updated: **{_LAST_UPDATED}**_")
@@ -249,7 +280,7 @@ def main():
     available_species = sorted(df_bench["species"].unique())
     selected_species: List[str] = []
     for sp in available_species:
-        if st.sidebar.toggle(sp, value=True, key=f"species_{benchmark_name}_{sp}"):
+        if sidebar_toggle(sp, value=True, key=f"species_{benchmark_name}_{sp}"):
             selected_species.append(sp)
 
     # Assay toggles (Pearson only), based on filtered species
@@ -266,7 +297,7 @@ def main():
         )
         selected_assays: List[str] = []
         for assay in available_assays:
-            if st.sidebar.toggle(assay, value=True, key=f"assay_{benchmark_name}_{assay}"):
+            if sidebar_toggle(assay, value=True, key=f"assay_{benchmark_name}_{assay}"):
                 selected_assays.append(assay)
     else:
         selected_assays = []
@@ -281,7 +312,7 @@ def main():
             df_for_tracks = df_bench
         available_datasets = sorted(df_for_tracks["datasets"].unique())
         for ds in available_datasets:
-            if st.sidebar.toggle(ds, value=True, key=f"dataset_{benchmark_name}_{ds}"):
+            if sidebar_toggle(ds, value=True, key=f"dataset_{benchmark_name}_{ds}"):
                 selected_datasets.append(ds)
     else:
         selected_datasets = []
@@ -290,7 +321,7 @@ def main():
     st.sidebar.subheader("Models")
     selected_models: List[str] = []
     for model in _ALL_MODELS:
-        if st.sidebar.toggle(model, value=True, key=f"model_{model}"):
+        if sidebar_toggle(model, value=True, key=f"model_{model}"):
             selected_models.append(model)
 
     # --- Main content ---
@@ -333,4 +364,5 @@ def main():
             st.plotly_chart(fig, use_container_width=True)
 
 
-main()
+if __name__ == "__main__":
+    main()