Update app.py

app.py

@@ -10,7 +10,7 @@ import pandas as pd
 
 
 basic_component_values = [None] * 6
-leader_component_values = [None] * 5
+leader_component_values = [None]
 
 def make_default_md(arena_df, elo_results):
     total_votes = sum(arena_df["num_battles"]) // 2
@@ -70,12 +70,7 @@ def update_elo_components(max_num_files, elo_results_file):
     if elo_results_file is None:  # Do live update
         battles = clean_battle_data(log_files)
         elo_results = report_elo_analysis_results(battles)
-
         leader_component_values[0] = make_leaderboard_md_live(elo_results)
-        leader_component_values[1] = elo_results["win_fraction_heatmap"]
-        leader_component_values[2] = elo_results["battle_count_heatmap"]
-        leader_component_values[3] = elo_results["bootstrap_elo_rating"]
-        leader_component_values[4] = elo_results["average_win_rate_bar"]
 
     # Basic stats
     basic_stats = report_basic_stats(log_files)
@@ -282,10 +277,10 @@ cat_name_to_explanation = {
     "Overall": "Overall Questions",
 }
 
-def build_leaderboard_tab(elo_results_file, leaderboard_table_file, show_plot=False):
+def build_leaderboard_tab(results_file, leaderboard_table_file, show_plot=False):
     arena_dfs = {}
     category_elo_results = {}
-    if elo_results_file is None:  # Do live update
+    if results_file is None:  # Do live update
         default_md = "Loading ..."
     else:
         with open(elo_results_file, "rb") as fin:
@@ -309,9 +304,49 @@ def build_leaderboard_tab(elo_results_file, leaderboard_table_file, show_plot=Fa
         with gr.Tabs() as tabs:
             # arena table
             arena_table_vals = get_arena_table(arena_df, model_table_df)
-            # with gr.Tab("Arena Elo", id=0):
-            #     md = make_arena_leaderboard_md(arena_df)
-            #     leaderboard_markdown = gr.Markdown(md, elem_id="leaderboard_markdown")
+            with gr.Tab("Arena Elo", id=0):
+                md = make_arena_leaderboard_md(arena_df)
+                leaderboard_markdown = gr.Markdown(md, elem_id="leaderboard_markdown")
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        category_dropdown = gr.Dropdown(choices=list(arena_dfs.keys()), label="Category", value="Overall")
+                    default_category_details = make_category_arena_leaderboard_md(arena_df, arena_df, name="Overall")
+                    with gr.Column(scale=4, variant="panel"):
+                        category_deets = gr.Markdown(default_category_details, elem_id="category_deets")
+
+                elo_display_df = gr.Dataframe(
+                    headers=[
+                        "Rank",
+                        "🤖 Model",
+                        "⭐ Arena Elo",
+                        "Organization",
+                        "License",
+                    ],
+                    datatype=[
+                        "number",
+                        "markdown",
+                        "number",
+                        "str",
+                        "str",
+                    ],
+                    value=arena_table_vals,
+                    elem_id="arena_leaderboard_dataframe",
+                    height=700,
+                    column_widths=[70, 190, 110, 160, 150, 140],
+                    wrap=True,
+                )
+
+                gr.Markdown(
+                    f"""Note: .
+            """,
+                    elem_id="leaderboard_markdown"
+                )
+
+                leader_component_values[:] = [default_md]
+
+            # with gr.Tab("Full Leaderboard", id=0):
+            #     md = make_full_leaderboard_md(elo_results)
+            #     gr.Markdown(md, elem_id="leaderboard_markdown")
             #     with gr.Row():
             #         with gr.Column(scale=2):
             #             category_dropdown = gr.Dropdown(choices=list(arena_dfs.keys()), label="Category", value="Overall")
@@ -319,63 +354,30 @@ def build_leaderboard_tab(elo_results_file, leaderboard_table_file, show_plot=Fa
             #         with gr.Column(scale=4, variant="panel"):
             #             category_deets = gr.Markdown(default_category_details, elem_id="category_deets")
 
-            #     elo_display_df = gr.Dataframe(
+            #     full_table_vals = get_full_table(model_table_df)
+            #     display_df = gr.Dataframe(
             #         headers=[
-            #             "Rank",
             #             "🤖 Model",
-            #             "⭐ Arena Elo",
+            #             "⭐ Task 1",
+            #             "📈 Task 2",
+            #             "📚 Task 3",
             #             "Organization",
             #             "License",
             #         ],
-            #         datatype=[
-            #             "number",
-            #             "markdown",
-            #             "number",
-            #             "str",
-            #             "str",
-            #         ],
-            #         value=arena_table_vals,
-            #         elem_id="arena_leaderboard_dataframe",
+            #         datatype=["markdown", "number", "number", "number", "str", "str"],
+            #         value=full_table_vals,
+            #         elem_id="full_leaderboard_dataframe",
+            #         column_widths=[200, 100, 100, 100, 150, 150],
             #         height=700,
-            #         column_widths=[70, 190, 110, 160, 150, 140],
             #         wrap=True,
             #     )
-
             #     gr.Markdown(
             #         f"""Note: .
-            # """,
+            #         """,
             #         elem_id="leaderboard_markdown"
             #     )
 
             #     leader_component_values[:] = [default_md]
-
-            with gr.Tab("Full Leaderboard", id=0):
-                md = make_full_leaderboard_md(elo_results)
-                gr.Markdown(md, elem_id="leaderboard_markdown")
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        category_dropdown = gr.Dropdown(choices=list(arena_dfs.keys()), label="Category", value="Overall")
-                    default_category_details = make_category_arena_leaderboard_md(arena_df, arena_df, name="Overall")
-                    with gr.Column(scale=4, variant="panel"):
-                        category_deets = gr.Markdown(default_category_details, elem_id="category_deets")
-
-                full_table_vals = get_full_table(model_table_df)
-                display_df = gr.Dataframe(
-                    headers=[
-                        "🤖 Model",
-                        "⭐ Task 1",
-                        "📈 Task 2",
-                        "📚 Task 3",
-                        "Organization",
-                        "License",
-                    ],
-                    datatype=["markdown", "number", "number", "number", "str", "str"],
-                    value=full_table_vals,
-                    elem_id="full_leaderboard_dataframe",
-                    column_widths=[200, 100, 100, 100, 150, 150],
-                    height=700,
-                    wrap=True,
-                )
         if not show_plot:
             gr.Markdown(
                 """ ## Submit your model [here]().