checkbox

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 
 # Function to load data from a given CSV file
-def load_data(model,version):
+def load_data(model,version,metrics):
     version = version.replace("%", "p")
     file_path = f'versions/{model}-{version}.csv'  # Replace with your file paths
     df = pd.read_csv(file_path)
@@ -16,6 +16,14 @@ def load_data(model,version):
                     "ROUGE Retain", "Truth Ratio Retain", "Prob. Retain", 
                     "ROUGE Forget", "Truth Ratio Forget", "Prob. Forget", 
                     ]
+    #based on the metrics, remove the columns that are not needed
+    if "ROUGE" not in metrics:
+        column_names = [x for x in column_names if "ROUGE" not in x]
+    if "Truth Ratio" not in metrics:
+        column_names = [x for x in column_names if "Truth Ratio" not in x]
+    if "Prob." not in metrics:
+        column_names = [x for x in column_names if "Prob." not in x]
+
     #if there is a column with name WD, modify each entry in Method to include WD: method (WD = wd)
     if "WD" in df.columns:
         #get the WD column entry for each row and add it to the method name
@@ -30,6 +38,11 @@ def load_data(model,version):
     
     return df
 
+
+# def style_leaderboard(df):
+    # make color red for background if column has "Forget" in it
+
+
 # Function for searching in the leaderboard
 def search_leaderboard(df, query):
     if query == "":
@@ -68,6 +81,11 @@ with demo:
                     label="🔄 Select Base Model",
                     value="llama",
                 )
+                metrics_checkbox = gr.CheckboxGroup(
+                    label="Select Metrics",
+                    choices=["ROUGE", "Truth Ratio", "Prob."],
+                    default=["ROUGE", "Truth Ratio", "Prob."],
+                )
 
             with gr.Row():
                 search_bar = gr.Textbox(
@@ -83,36 +101,29 @@ with demo:
 
             version_dropdown.change(
                 change_version,
-                inputs=[model_dropdown,version_dropdown],
+                inputs=[model_dropdown,version_dropdown,metrics_checkbox],
                 outputs=leaderboard_table
             )
 
             model_dropdown.change(
                 change_version,
-                inputs=[model_dropdown,version_dropdown],
+                inputs=[model_dropdown,version_dropdown,metrics_checkbox],
                 outputs=leaderboard_table
             )
 
             search_bar.change(
                 search_leaderboard,
-                inputs=[leaderboard_table, search_bar],
+                inputs=[leaderboard_table, search_bar,metrics_checkbox],
                 outputs=leaderboard_table
             )
 
-        # with gr.TabItem("Plots"):
-        #     version_dropdown_plots = gr.Dropdown(
-        #             choices=["llama", "phi", "stable-lm"],
-        #             label="🔄 Select Base Model",
-        #             value="llama",
-        #         )
+            metrics_checkbox.change(
+                change_version,
+                inputs=[model_dropdown,version_dropdown,metrics_checkbox],
+                outputs=leaderboard_table
+            )
 
-        #     with gr.Row():
-        #         methods_checkbox = gr.CheckboxGroup(
-        #             label="Select Methods",
-        #             choices=list(load_data("llama", "10%")['Method'].unique()),  # To be populated dynamically
-        #         )
 
-        #     plot_output = gr.Plot()
 
         #     # Dynamically update the choices for the methods checkbox
         #     def update_method_choices(version):