Update app.py

app.py

@@ -313,6 +313,10 @@ def select(df, data: gr.SelectData):
     print('selected_values', selected_values, selected_indices)
     return gr.CheckboxGroup(list(selected_values.keys()))
 
+def init_comparison_data():
+    global selected_values
+    return gr.CheckboxGroup(choices=list(selected_values.keys()), value=list(selected_values.keys())) 
+
 def generate_spider_chart(df, selected_keys):
     global selected_values
     print('generate_spider_chart', selected_values, selected_keys)
@@ -425,10 +429,10 @@ with demo:
                             filter_columns_groupDtype = gr.Dropdown(choices=[i.value.name for i in GroupDtype], label="Group Size", multiselect=False, value="All", interactive=True,)
 
                     with gr.Row():
-                        with gr.Column(scale=5):
-                            model_comparison = gr.CheckboxGroup(label="Accuracy Comparison on Selected Models", choices=list(selected_values.keys()), value=[], interactive=True, elem_id="model_comparison")
+                        with gr.Column(scale=6):
+                            model_comparison = gr.CheckboxGroup(label="Accuracy Comparison on Selected Models", choices=list(selected_values.keys()), value=list(selected_values.keys()), interactive=True, elem_id="model_comparison")
                         with gr.Column(scale=1):
-                            spider_btn = gr.Button("Spider Chart Generation")                   
+                            spider_btn = gr.Button("Compare")                 
       
                     
             leaderboard_table = gr.components.Dataframe(
@@ -451,6 +455,7 @@ with demo:
             leaderboard_table.select(select, leaderboard_table, model_comparison)
             spider_btn.click(generate_spider_chart, [leaderboard_table, model_comparison], map)
             spider_btn.click(lambda: Modal(visible=True), None, modal)
+            demo.load(init_comparison_data, None, model_comparison)
             
             # Dummy leaderboard for handling the case when the user uses backspace key
             hidden_leaderboard_table_for_search = gr.components.Dataframe(