[Add] metric selection

app.py

@@ -9,7 +9,7 @@ from src.dataloading import get_leaderboard_models_cached, get_leaderboard_datas
 # Set matplotlib backend for non-GUI environments
 plt.switch_backend('Agg')
 
-def create_heatmap(selected_models, selected_dataset):
+def create_heatmap(selected_models, selected_dataset, selected_metric):
     if not selected_models or not selected_dataset:
         return None
     
@@ -78,6 +78,13 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
             allow_custom_value=False,
             info="Search and select multiple models"
         )
+
+    metric_dropdown = gr.Dropdown(
+        choices=["Kappa_p (prob.)", "Kappa_p (det.)", "Error Consistency"],
+        label="Select Metric",
+        default="Kappa_p (prob.)",
+        info="Select a similarity metric to compute"
+    )
     
     generate_btn = gr.Button("Generate Heatmap", variant="primary")
     heatmap = gr.Image(label="Similarity Heatmap", visible=True)
@@ -88,7 +95,7 @@ with gr.Blocks(title="LLM Similarity Analyzer") as demo:
         queue=False
     ).then(
         fn=create_heatmap,
-        inputs=[model_dropdown, dataset_dropdown],
+        inputs=[model_dropdown, dataset_dropdown, metric_dropdown],
         outputs=heatmap
     )