Update app.py

app.py

@@ -2,9 +2,13 @@ from utils import *
 
 global data_component
 
-def update_table(query, min_size, max_size):
+def update_table(query, min_size, max_size, selected_subjects=None):
     df = get_df()
     filtered_df = search_and_filter_models(df, query, min_size, max_size)
+    if selected_subjects and len(selected_subjects) > 0:
+        base_columns = ['Models', 'Model Size(B)', 'Data Source', 'Overall']
+        selected_columns = base_columns + selected_subjects
+        filtered_df = filtered_df[selected_columns]
     return filtered_df
 
 with gr.Blocks() as block:
@@ -49,6 +53,16 @@ with gr.Blocks() as block:
                     label="Maximum number of parameters (B)",
                 )
 
+            # 添加学科选择器
+            subject_choices = [col for col in COLUMN_NAMES if col not in ['Models', 'Model Size(B)', 'Data Source', 'Overall']]
+            with gr.Row():
+                subjects_select = gr.CheckboxGroup(
+                    choices=subject_choices,
+                    value=subject_choices,  # 默认全选
+                    label="Select Subjects to Display",
+                    elem_id="subjects-select"
+                )
+
             data_component = gr.components.Dataframe(
                 value=df[COLUMN_NAMES],
                 headers=COLUMN_NAMES,
@@ -60,9 +74,31 @@ with gr.Blocks() as block:
             )
             
             refresh_button = gr.Button("Refresh")
-            search_bar.change(fn=update_table, inputs=[search_bar, min_size_slider, max_size_slider], outputs=data_component)
-            min_size_slider.change(fn=update_table, inputs=[search_bar, min_size_slider, max_size_slider], outputs=data_component)
-            max_size_slider.change(fn=update_table, inputs=[search_bar, min_size_slider, max_size_slider], outputs=data_component)
+            
+            # 更新事件处理
+            def update_with_subjects(*args):
+                return update_table(*args)
+
+            search_bar.change(
+                fn=update_with_subjects, 
+                inputs=[search_bar, min_size_slider, max_size_slider, subjects_select], 
+                outputs=data_component
+            )
+            min_size_slider.change(
+                fn=update_with_subjects, 
+                inputs=[search_bar, min_size_slider, max_size_slider, subjects_select], 
+                outputs=data_component
+            )
+            max_size_slider.change(
+                fn=update_with_subjects, 
+                inputs=[search_bar, min_size_slider, max_size_slider, subjects_select], 
+                outputs=data_component
+            )
+            subjects_select.change(
+                fn=update_with_subjects, 
+                inputs=[search_bar, min_size_slider, max_size_slider, subjects_select], 
+                outputs=data_component
+            )
             refresh_button.click(fn=refresh_data, outputs=data_component)
 
         # table 2