Update src/vis_utils.py

src/vis_utils.py

@@ -247,26 +247,24 @@ def update_metric_choices(benchmark_type):
         return (
             gr.update(choices=metric_names, value=metric_names[0], visible=True),
             gr.update(choices=metric_names, value=metric_names[1], visible=True),
-            gr.update(visible=False), gr.update(visible=False), 
-            gr.update(visible=False), gr.update(visible=False)
+            gr.update(visible=False), gr.update(visible=False),  gr.update(visible=False)
         )
     elif benchmark_type == 'function':
         # Show aspect and dataset type selectors for function
         aspect_types = benchmark_specific_metrics[benchmark_type]['aspect_types']
-        dataset_types = benchmark_specific_metrics[benchmark_type]['dataset_types']
+        metric_types = benchmark_specific_metrics[benchmark_type]['dataset_types']
         return (
             gr.update(visible=False), gr.update(visible=False),
             gr.update(choices=aspect_types, value=aspect_types[0], visible=True),
-            gr.update(choices=dataset_types, value=dataset_types[0], visible=True),
-            gr.update(visible=False), gr.update(visible=False)
+            gr.update(visible=False), 
+            gr.update(choices=metric_types, value=metric_types[0], visible=True)
         )
     elif benchmark_type == 'family':
         # Show dataset and metric selectors for family
         datasets = benchmark_specific_metrics[benchmark_type]['datasets']
         metrics = benchmark_specific_metrics[benchmark_type]['metrics']
         return (
-            gr.update(visible=False), gr.update(visible=False),
-            gr.update(visible=False), gr.update(visible=False),
+            gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
             gr.update(choices=datasets, value=datasets[0], visible=True),
             gr.update(choices=metrics, value=metrics[0], visible=True)
         )
@@ -275,8 +273,8 @@ def update_metric_choices(benchmark_type):
         metrics = benchmark_specific_metrics[benchmark_type]
         return (
             gr.update(visible=False), gr.update(visible=False),
-            gr.update(visible=False), gr.update(visible=False),
+            gr.update(visible=False),
             gr.update(visible=False), gr.update(choices=metrics, value=metrics[0], visible=True)
         )
         
-    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)