add raw data view

app.py

@@ -110,6 +110,28 @@ def get_data_split(dfs, set_name):
     return df.reset_index(drop=True)
 
 
+def clean_df_for_display(df):
+
+    df = df.loc[
+        :,
+        [
+            "Model",
+            "rating",
+            "MMLU",
+            "MT-bench (score)",
+            "Release Date",
+            "Organization",
+            "License",
+            "Link",
+        ],
+    ].rename(columns={"rating": "ELO Score", "MT-bench (score)": "MT-Bench"})
+
+    df["Release Date"] = df["Release Date"].astype(str)
+    df.sort_values("ELO Score", ascending=False, inplace=True)
+    df.reset_index(drop=True, inplace=True)
+    return df
+
+
 def filter_df(min_score, max_models_per_month, set_selector, org_selector):
     df = get_data_split(merged_dfs, set_name=set_selector)
 
@@ -169,7 +191,7 @@ def build_plot(toggle_annotations, filtered_df):
                 arrowhead=0,
             )
 
-    return fig
+    return fig, clean_df_for_display(filtered_df)
 
 
 set_dark_mode = """
@@ -260,44 +282,73 @@ with gr.Blocks(
                 )
 
     # Show plot
-    plot = gr.Plot()
     filtered_df = gr.State()
+    with gr.Group():
+        with gr.Tab("Plot"):
+            plot = gr.Plot()
+        with gr.Tab("Raw Data"):
+
+            display_df = gr.DataFrame()
 
     demo.load(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     min_score.change(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     max_models_per_month.change(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     toggle_annotations.change(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     set_selector.change(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     org_selector.change(
         fn=filter_df,
         inputs=[min_score, max_models_per_month, set_selector, org_selector],
         outputs=filtered_df,
-    ).then(fn=build_plot, inputs=[toggle_annotations, filtered_df], outputs=plot)
+    ).then(
+        fn=build_plot,
+        inputs=[toggle_annotations, filtered_df],
+        outputs=[plot, display_df],
+    )
 
     gr.Markdown(
         """