Connect 'Show only open models' checkbox to Winners and Evolution sections

- Modified create_leaderboard_display to return show_open_only_checkbox
- Updated main_page.py to connect checkbox to Winners by Category and Evolution charts
- Pre-generate both all and open-only versions of Winners HTML and Evolution plot
- When checkbox is toggled, both sections now update to show only open models

main_page.py

@@ -36,20 +36,30 @@ def build_page():
 
     test_df, test_tag_map = get_full_leaderboard_data("test")
     if not test_df.empty:
-        create_leaderboard_display(
+        # Get the checkbox returned from create_leaderboard_display
+        show_open_only_checkbox = create_leaderboard_display(
             full_df=test_df,
             tag_map=test_tag_map,
             category_name=CATEGORY_NAME,
             split_name="test"
         )
         
+        # Prepare open-only filtered dataframe for Winners and Evolution
+        if 'Openness' in test_df.columns:
+            test_df_open = test_df[test_df['Openness'].str.lower() == 'open'].copy()
+        else:
+            test_df_open = test_df.copy()
+        
         # --- Winners by Category Section ---
         gr.Markdown("---")
         gr.HTML('<h2>Winners by Category</h2>', elem_id="winners-header")
         gr.Markdown("Top 5 performing systems in each benchmark category.")
         
-        winners_html = create_winners_by_category_html(test_df, top_n=5)
-        gr.HTML(winners_html, elem_id="winners-by-category")
+        # Create both all and open-only versions of winners HTML
+        winners_html_all = create_winners_by_category_html(test_df, top_n=5)
+        winners_html_open = create_winners_by_category_html(test_df_open, top_n=5)
+        
+        winners_component = gr.HTML(winners_html_all, elem_id="winners-by-category")
         
         # --- New Visualization Sections ---
         gr.Markdown("---")
@@ -58,18 +68,35 @@ def build_page():
         gr.HTML('<h2>Evolution Over Time</h2>', elem_id="evolution-header")
         gr.Markdown("Track how model performance has improved over time based on release dates.")
         
-        evolution_fig = create_evolution_over_time_chart(test_df)
-        gr.Plot(value=evolution_fig, elem_id="evolution-chart")
+        # Create both all and open-only versions of evolution chart
+        evolution_fig_all = create_evolution_over_time_chart(test_df)
+        evolution_fig_open = create_evolution_over_time_chart(test_df_open)
+        
+        evolution_component = gr.Plot(value=evolution_fig_all, elem_id="evolution-chart")
         
         gr.Markdown("---")
         
-        # Open Model Accuracy by Size Section
+        # Open Model Accuracy by Size Section (always shows open models only by design)
         gr.HTML('<h2>Open Model Accuracy by Size</h2>', elem_id="size-accuracy-header")
         gr.Markdown("Compare open-weights model performance against their parameter count. The x-axis shows active parameters (relevant for MoE models).")
         
         size_fig = create_accuracy_by_size_chart(test_df)
         gr.Plot(value=size_fig, elem_id="size-accuracy-chart")
         
+        # Connect the open models checkbox to update Winners and Evolution sections
+        if show_open_only_checkbox is not None:
+            def update_winners_and_evolution(show_open_only):
+                if show_open_only:
+                    return winners_html_open, evolution_fig_open
+                else:
+                    return winners_html_all, evolution_fig_all
+            
+            show_open_only_checkbox.change(
+                fn=update_winners_and_evolution,
+                inputs=[show_open_only_checkbox],
+                outputs=[winners_component, evolution_component]
+            )
+        
     else:
         gr.Markdown("No data available.")
 

ui_components.py

@@ -1061,8 +1061,8 @@ def create_leaderboard_display(
                 outputs=[dataframe_component, cost_plot_component, runtime_plot_component]
             )
 
-    # Return the components so they can be referenced elsewhere.
-    return cost_plot_component, runtime_plot_component, dataframe_component
+    # Return the show_open_only_checkbox so it can be used to update other sections
+    return show_open_only_checkbox
 
 # # --- Detailed Benchmark Display ---
 def create_benchmark_details_display(