update scattor and modal

app.py

@@ -45,6 +45,10 @@ from src.tools.plots import (
     create_plot_df,
     create_scores_df,
 )
+from gradio_modal import Modal
+import plotly.graph_objects as go
+
+selected_indices = []
 
 # Start ephemeral Spaces on PRs (see config in README.md)
 #enable_space_ci()
@@ -278,6 +282,35 @@ def filter_models(
 
     return filtered_df
 
+def select(df, data: gr.SelectData):
+    global selected_indices
+    selected_index = data.index[0]  
+    
+    if selected_index in selected_indices:
+        selected_indices.remove(selected_index) 
+    else:
+        selected_indices.append(selected_index)  
+
+    fig = go.Figure()
+    for i in selected_indices:
+        row = df.iloc[i, :]
+        fig.add_trace(go.Scatterpolar(
+            r=[row['Average ⬆️'], row['ARC-c'], row['ARC-e'], row['Boolq'], row['HellaSwag'], row['Lambada'], row['MMLU'], row['Openbookqa'], row['Piqa'], row['Truthfulqa'], row['Winogrande']],
+            theta=['Average ⬆️', 'ARC-c', 'ARC-e', 'Boolq', 'HellaSwag', 'Lambada', 'MMLU', 'Openbookqa', 'Piqa', 'Truthfulqa', 'Winogrande',],
+            fill='toself',
+            name=str(row['Model'])  
+        ))
+    fig.update_layout(
+        polar=dict(
+            radialaxis=dict(
+                visible=True,
+            )),
+        showlegend=True
+    )
+    
+
+    return fig    
+
 leaderboard_df = filter_models(
     df=leaderboard_df, 
     type_query=[t.to_str(" : ") for t in QuantType], 
@@ -377,6 +410,12 @@ with demo:
                 #column_widths=["2%", "33%"] 
             )
 
+            with Modal(visible=False) as modal:
+                map = gr.Plot()
+            
+            leaderboard_table.select(select, leaderboard_table, map)
+            leaderboard_table.select(lambda: Modal(visible=True), None, modal)
+            
             # Dummy leaderboard for handling the case when the user uses backspace key
             hidden_leaderboard_table_for_search = gr.components.Dataframe(
                 value=original_df[COLS],
@@ -448,7 +487,11 @@ with demo:
                 [filter_columns_type],
                 [filter_columns_weightDtype]
             )
+
             
+
+
+    
             for selector in [shown_columns, filter_columns_type, filter_columns_precision, filter_columns_size, filter_columns_parameters, hide_models, filter_columns_computeDtype, filter_columns_weightDtype, filter_columns_doubleQuant, filter_columns_groupDtype]:
                 selector.change(
                     update_table,