add graph tab

app.py

@@ -92,6 +92,36 @@ def get_dataset(
     return leaderboard_table
 
 
+def get_bar_plot(
+    histone_tasks: List[str],
+    regulatory_tasks: List[str],
+    rna_tasks: List[str],
+    target_metric: str = "MCC",
+    aggregation_method: str = "mean",
+):
+    tasks = histone_tasks + regulatory_tasks + rna_tasks
+
+    aggr_fn = getattr(np, aggregation_method)
+    scores = _ORIGINAL_DF[target_metric].apply(retrieve_array_from_text).apply(aggr_fn)
+    scores = scores.apply(format_number)
+    df = _ORIGINAL_DF.drop(columns=_METRICS)
+    df["Score"] = scores / len(tasks)
+    df = df.query(f"Dataset == {tasks}")
+
+    bar_plot = gr.BarPlot.update(
+        df,
+        x="Model",
+        y="Score",
+        color="Dataset",
+        width=500,
+        x_label_angle=-45,
+        x_title="Model",
+        y_title="Score",
+        color_legend_title="Downstream Task",
+    )
+    return bar_plot
+
+
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Image(banner_url, height=160, scale=1)
@@ -137,7 +167,12 @@ with gr.Blocks() as demo:
                 elem_id="leaderboard-table",
             )
 
-        with gr.TabItem("📈 Metrics", elem_id="od-benchmark-tab-table", id=1):
+        with gr.TabItem("📈 Graph", elem_id="od-benchmark-tab-table", id=2):
+            bar_plot = gr.BarPlot(
+                elem_id="leaderboard-bar-plot",
+            )
+
+        with gr.TabItem("ℹ️ Metrics", elem_id="od-benchmark-tab-table", id=1):
             gr.Markdown("Hey hey hey", elem_classes="markdown-text")
 
     gr.Markdown(f"Last updated on **{_LAST_UPDATED}**", elem_classes="markdown-text")
@@ -182,4 +217,35 @@ with gr.Blocks() as demo:
         outputs=dataframe,
     )
 
+    histone_tasks.change(
+        get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+    regulatory_tasks.change(
+        get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+    rna_tasks.change(
+        get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+    metric_choice.change(
+        get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+    aggr_choice.change(
+        get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+    demo.load(
+        fn=get_bar_plot,
+        inputs=[histone_tasks, regulatory_tasks, rna_tasks, metric_choice, aggr_choice],
+        outputs=bar_plot,
+    )
+
 demo.launch()