Removed pycache (hopefully), and fixed assay names

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+*.pyc

about.py

@@ -1,7 +1,21 @@
 import os
 from huggingface_hub import HfApi
 
-ASSAY_LIST = ["AC-SINS_pH7.4", "PSP_CHO", "HIC", "Tm2", "Titer"]
+ASSAY_LIST = ["AC-SINS_pH7.4", "PR_CHO", "HIC", "Tm2", "Titer"]
+ASSAY_RENAME = {
+    "AC-SINS_pH7.4": "Self-association",
+    "PR_CHO": "Polyreactivity",
+    "HIC": "Hydrophobicity",
+    "Tm2": "Thermostability",
+    "Titer": "Titer",
+}
+ASSAY_EMOJIS = {
+    "AC-SINS_pH7.4": "🧲",
+    "PR_CHO": "🎯",
+    "HIC": "💧",
+    "Tm2": "🌡️",
+    "Titer": "🧪",
+}
 TOKEN = os.environ.get("HF_TOKEN")
 CACHE_PATH=os.getenv("HF_HOME", ".")
 API = HfApi(token=TOKEN)

app.py

@@ -7,7 +7,7 @@ from gradio_leaderboard import Leaderboard
 from evaluation import evaluate_problem
 
 from utils import read_submission_from_hub, write_results
-from about import ASSAY_LIST
+from about import ASSAY_LIST, ASSAY_RENAME, ASSAY_EMOJIS
 
 def evaluate_boundary(filename):
     print(filename)
@@ -36,8 +36,9 @@ def get_leaderboard_table(assay: str | None = None):
     # to_show['user'] = to_show['user'].apply(lambda x: make_user_clickable(x)).astype(str)
     
     # Previously hosted on HF hub, local for now
-    column_order = ["model", "feature", "assay", "spearman"]
+    column_order = ["model", "feature", "property", "spearman", "spearman_abs"]  # "assay", 
     df = pd.read_csv("data/metrics_all.csv")
+    df["property"] = df["assay"].map(ASSAY_RENAME)
     if assay is not None:
         df = df[df['assay'] == assay]
     df = df[column_order]
@@ -48,11 +49,11 @@ def get_leaderboard_object(assay: str | None = None):
     df = get_leaderboard_table(assay=assay)
     filter_columns = ["model"]
     if assay is None:
-        filter_columns.append("assay")
+        filter_columns.append("property")
     Leaderboard(
         value=df,
         datatype=["str", "str", "str", "number"],
-        select_columns=["model", "assay", "feature", "spearman"],
+        select_columns=["model", "feature", "property", "spearman"],
         search_columns=["model"],
         # hide_columns=["spearman_abs"],
         filter_columns=filter_columns,
@@ -77,8 +78,8 @@ with gr.Blocks() as demo:
 
         # Procedurally make these 5 tabs
         for assay in ASSAY_LIST:
-            with gr.TabItem(assay, elem_id=f"abdev-benchmark-tab-table"):
-                gr.Markdown(f"# {assay}")
+            with gr.TabItem(f"{ASSAY_EMOJIS[assay]} {ASSAY_RENAME[assay]}", elem_id=f"abdev-benchmark-tab-table"):
+                gr.Markdown(f"# {ASSAY_RENAME[assay]} (measured by {assay})")
                 get_leaderboard_object(assay=assay)
         
         with gr.TabItem("❔About", elem_id="abdev-benchmark-tab-table"):