pufanyi commited on
Commit
f79866b
·
1 Parent(s): 595fce3

Refactor GET_DEFAULT_CHECKBOX function to remove "Model Name" column

Browse files
Files changed (1) hide show
  1. app.py +22 -18
app.py CHANGED
@@ -80,7 +80,7 @@ def GET_DEFAULT_TEXTBOX():
80
 
81
  def GET_DEFAULT_CHECKBOX(subset):
82
  choices = list(LEADERBOARD_DF[subset].columns)
83
- # choices.remove("Model Name")
84
  # print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
85
  return gr.CheckboxGroup(
86
  choices=choices,
@@ -102,6 +102,8 @@ def init_leaderboard(dataframes):
102
  with gr.Row():
103
  # datatype =
104
  df = gr.Dataframe(data, type="pandas")
 
 
105
 
106
  def refresh(subset):
107
  global LEADERBOARD_DF
@@ -116,23 +118,25 @@ def init_leaderboard(dataframes):
116
  print("Subset:", subset)
117
  print("Search Term:", search_term)
118
  print("Selected Columns:", selected_columns)
119
- try:
120
- filtered_data = dataframes[subset]
121
- if search_term:
122
- filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
123
- filtered_data.sort_values(by="Total", ascending=False, inplace=True)
124
- # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
125
- selected_data = filtered_data[selected_columns]
126
- return gr.DataFrame(
127
- selected_data,
128
- type="pandas",
129
- # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
130
- )
131
- except Exception as e:
132
- print("====================================")
133
- print("Error:", e)
134
- print("====================================")
135
- return refresh(subset)
 
 
136
 
137
  with gr.Row():
138
  refresh_button = gr.Button("Refresh")
 
80
 
81
  def GET_DEFAULT_CHECKBOX(subset):
82
  choices = list(LEADERBOARD_DF[subset].columns)
83
+ choices.remove("Model Name")
84
  # print("Choices:", [c.name for c in fields(AutoEvalColumn) if not c.hidden])
85
  return gr.CheckboxGroup(
86
  choices=choices,
 
102
  with gr.Row():
103
  # datatype =
104
  df = gr.Dataframe(data, type="pandas")
105
+
106
+ old_version = NEWEST_VERSION
107
 
108
  def refresh(subset):
109
  global LEADERBOARD_DF
 
118
  print("Subset:", subset)
119
  print("Search Term:", search_term)
120
  print("Selected Columns:", selected_columns)
121
+ if old_version != subset:
122
+ search_term = None
123
+ selected_columns = GET_DEFAULT_CHECKBOX(subset)
124
+ old_version = subset
125
+ filtered_data = dataframes[subset]
126
+ if search_term:
127
+ filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
128
+ filtered_data.sort_values(by="Total", ascending=False, inplace=True)
129
+ selected_columns.append("Model Name")
130
+ selected_columns = [
131
+ c for c in filtered_data.columns if c in selected_columns
132
+ ]
133
+ # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
134
+ selected_data = filtered_data[selected_columns]
135
+ return gr.DataFrame(
136
+ selected_data,
137
+ type="pandas",
138
+ # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
139
+ )
140
 
141
  with gr.Row():
142
  refresh_button = gr.Button("Refresh")