Update app.py

app.py

@@ -45,6 +45,18 @@ def refresh_data():
     benchmark_types = ["similarity", "function", "family", "affinity"]
     download_from_hub(benchmark_types)
 
+            # Define a function to update metrics based on benchmark type selection
+def update_metrics(selected_benchmarks):
+    updated_metrics = set()
+    for benchmark in selected_benchmarks:
+        updated_metrics.update(benchmark_metric_mapping.get(benchmark, []))
+    return list(updated_metrics)
+        
+            # Define a function to update the leaderboard
+def update_leaderboard(selected_methods, selected_metrics):
+    updated_df = get_baseline_df(selected_methods, selected_metrics)
+    return updated_df
+                
 block = gr.Blocks()
 
 with block:
@@ -59,13 +71,27 @@ with block:
             metric_names = leaderboard.columns.tolist()
             metrics_with_method = metric_names.copy()
             metric_names.remove('Method')  # Remove method_name from the metric options
+
+            benchmark_metric_mapping = {
+                "similarity": [metric for metric in metric_names if metric.startswith('sim_')],
+                "function": [metric for metric in metric_names if metric.startswith('func')],
+                "family": [metric for metric in metric_names if metric.startswith('fam_')],
+                "affinity": [metric for metric in metric_names if metric.startswith('aff_')],
+            }
             
             # Leaderboard section with method and metric selectors
             leaderboard_method_selector = gr.CheckboxGroup(
                 choices=method_names, label="Select Methods for the Leaderboard", value=method_names, interactive=True
             )
+
+            benchmark_type_selector = gr.CheckboxGroup(
+                choices=list(benchmark_metric_mapping.keys()), 
+                label="Select Benchmark Types", 
+                value=None,  # Initially select all benchmark types
+                interactive=True
+            )
             leaderboard_metric_selector = gr.CheckboxGroup(
-                choices=metric_names, label="Select Metrics for the Leaderboard", value=metric_names, interactive=True
+                choices=metric_names, label="Select Metrics for the Leaderboard", value=None, interactive=True
             )
 
             # Display the filtered leaderboard
@@ -89,6 +115,14 @@ with block:
                 inputs=[leaderboard_method_selector, leaderboard_metric_selector], 
                 outputs=data_component
             )
+        
+            # Update metrics when benchmark type changes
+            benchmark_type_selector.change(
+                lambda selected_benchmarks: update_metrics(selected_benchmarks),
+                inputs=[benchmark_type_selector],
+                outputs=leaderboard_metric_selector
+            )
+
             leaderboard_metric_selector.change(
                 get_baseline_df, 
                 inputs=[leaderboard_method_selector, leaderboard_metric_selector],