Upload app.py

app.py

@@ -124,6 +124,20 @@ block = gr.Blocks(css="""
         max-width: 80%;
         margin: auto;
     }
+    /* 添加以下样式来控制表格列宽 */
+    .gradio-dataframe {
+        overflow-x: auto !important;
+    }
+    .gradio-dataframe table {
+        width: 100% !important;
+        white-space: nowrap !important;
+    }
+    .gradio-dataframe td, .gradio-dataframe th {
+        min-width: fit-content !important;
+        max-width: none !important;
+        white-space: pre-wrap !important;
+        padding: 8px !important;
+    }
 """)
 with block:
     with gr.Column(elem_classes="container"):
@@ -133,53 +147,72 @@ with block:
         with gr.Tabs(elem_classes="tab-buttons") as tabs:
             # Table 0
             with gr.TabItem("🏅 WorldScore Benchmark", elem_id="worldscore-tab-table", id=0):
-                with gr.Column():
-                    model_type_filter = gr.CheckboxGroup(
-                        choices=MODEL_TYPE,
-                        value=DEFAULT_MODEL_TYPE,
-                        label="Model Type",
-                        interactive=True
-                    )
-                    ability_filter = gr.CheckboxGroup(
-                        choices=ABILITY,
-                        value=DEFAULT_ABILITY,
-                        label="Ability",
-                        interactive=True
-                    )
+                with gr.Row():
+                    with gr.Column(scale=0.3):
+                        model_type_filter = gr.CheckboxGroup(
+                            choices=MODEL_TYPE,
+                            value=DEFAULT_MODEL_TYPE,
+                            label="Model Type",
+                            interactive=True
+                        )
+                        ability_filter = gr.CheckboxGroup(
+                            choices=ABILITY,
+                            value=DEFAULT_ABILITY,
+                            label="Ability",
+                            interactive=True
+                        )
+                    with gr.Column(scale=0.7):
+                        sort_by_filter = gr.Radio(
+                            choices=TASK_INFO,
+                            value=DEFAULT_INFO[0],
+                            label="Sort by",
+                            interactive=True
+                        )
                     
                 data_component = gr.components.Dataframe(
                     value=get_baseline_df(),
                     headers=COLUMN_NAMES,
-                    type="pandas", 
+                    type="pandas",
                     datatype=DATA_TITILE_TYPE,
                     interactive=False,
                     visible=True,
+                    wrap=True
                 )
         
-                def on_filter_change(model_types, abilities):
+                def on_filter_change(model_types, abilities, sort_by):
                     df = get_baseline_df()
                     # Filter by selected model types
                     df = df[df['Model Type'].isin(model_types)]
                     # Filter by selected abilities 
                     df = df[df['Ability'].isin(abilities)]
+                    # Sort by selected sort by
+                    df = df.sort_values(by=sort_by, ascending=False)
+                    
                     return gr.Dataframe(
                         value=df,
                         headers=COLUMN_NAMES,
                         type="pandas",
                         datatype=DATA_TITILE_TYPE,
                         interactive=False,
-                        visible=True
+                        visible=True,
+                        wrap=True
                     )
 
                 model_type_filter.change(
                     fn=on_filter_change,
-                    inputs=[model_type_filter, ability_filter],
+                    inputs=[model_type_filter, ability_filter, sort_by_filter],
                     outputs=data_component
                 )
 
                 ability_filter.change(
                     fn=on_filter_change,
-                    inputs=[model_type_filter, ability_filter], 
+                    inputs=[model_type_filter, ability_filter, sort_by_filter], 
+                    outputs=data_component
+                )
+                
+                sort_by_filter.change(
+                    fn=on_filter_change,
+                    inputs=[model_type_filter, ability_filter, sort_by_filter],
                     outputs=data_component
                 )
                     
@@ -255,17 +288,22 @@ with block:
                     )
 
 
-    def refresh_data():
-        value = get_baseline_df()
-        return value, DEFAULT_MODEL_TYPE, DEFAULT_ABILITY
+        def refresh_data():
+            value = get_baseline_df()
+            data_component.value = value
+            model_type_filter.value = DEFAULT_MODEL_TYPE
+            ability_filter.value = DEFAULT_ABILITY
+            sort_by_filter.value = DEFAULT_INFO[0]
 
-    with gr.Row(elem_classes="container"):
-        data_run = gr.Button("Refresh")
-        data_run.click(
-            refresh_data, 
-            inputs=[], 
-            outputs=[data_component, model_type_filter, ability_filter]
-        )
+            return data_component, model_type_filter, ability_filter, sort_by_filter
+
+        with gr.Row(elem_classes="container"):
+            data_run = gr.Button("Refresh")
+            data_run.click(
+                refresh_data, 
+                inputs=[], 
+                outputs=[data_component, model_type_filter, ability_filter, sort_by_filter]
+            )
 
 
 block.launch()