Update app.py

app.py

@@ -105,16 +105,24 @@ def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False))]
 
 
+# def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
+#     always_here_cols = [
+#         AutoEvalColumn.model_type_symbol.name,
+#         AutoEvalColumn.model.name,
+#     ]
+#     # We use COLS to maintain sorting
+#     filtered_df = df[
+#         always_here_cols + [c for c in COLS if c in df.columns and c in columns]# + [AutoEvalColumn.dummy.name]
+#     ]
+#     return filtered_df
+
 def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
     always_here_cols = [
         AutoEvalColumn.model_type_symbol.name,
         AutoEvalColumn.model.name,
     ]
-    # We use COLS to maintain sorting
-    filtered_df = df[
-        always_here_cols + [c for c in COLS if c in df.columns and c in columns]# + [AutoEvalColumn.dummy.name]
-    ]
-    return filtered_df
+    selected_cols = always_here_cols + [c for c in columns if c in df.columns]
+    return df[selected_cols]
 
 
 def filter_queries(query: str, filtered_df: pd.DataFrame):
@@ -261,7 +269,10 @@ with demo:
                     )
 
             leaderboard_df_filtered = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
+            initial_columns = [c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value
+            leaderboard_df_filtered = select_columns(leaderboard_df_filtered, initial_columns)
 
+            
             leaderboard_table = gr.components.Dataframe(
                 value=leaderboard_df_filtered,
                 headers=[c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value,
@@ -270,9 +281,6 @@ with demo:
                 interactive=False,
                 visible=True,
             )
-            print("Leaderboard table initial value:")
-            print(leaderboard_table.value)
-            print(f"Leaderboard table shape: {leaderboard_table.value.shape if isinstance(leaderboard_table.value, pd.DataFrame) else 'Not a DataFrame'}")
             
             # Dummy leaderboard for handling the case when the user uses backspace key
             hidden_leaderboard_table_for_search = gr.components.Dataframe(