Update src/vis_utils.py

src/vis_utils.py

@@ -244,4 +244,45 @@ def plot_affinity_results(file_path, method_names, metric, save_path="./plot_ima
     ax.get_figure().savefig(filename, dpi=400, bbox_inches='tight')
     plt.close()  # Close the plot to free memory
 
-    return filename
+    return filename
+
+def update_metric_choices(benchmark_type):
+    if benchmark_type == 'similarity':
+        # Show x and y metric selectors for similarity
+        metric_names = benchmark_specific_metrics.get(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)
+        )
+    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']
+        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)
+        )
+    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(choices=datasets, value=datasets[0], visible=True),
+            gr.update(choices=metrics, value=metrics[0], visible=True)
+        )
+    elif benchmark_type == 'affinity':
+        # Show single metric selector for affinity
+        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(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)