Update app.py

app.py

@@ -129,6 +129,17 @@ def get_baseline_df():
     df = convert_scores_to_percentage(df)
     return df
 
+def get_baseline_df_quality():
+    submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
+    submission_repo.git_pull()
+    df = pd.read_csv(CSV_DIR)
+    df = get_final_score(df, checkbox_group.value)
+    df = df.sort_values(by="Selected Score", ascending=False)
+    present_columns = MODEL_INFO_TAB_QUALITY + checkbox_group_quality.value
+    df = df[present_columns]
+    df = convert_scores_to_percentage(df)
+    return df
+
 def get_all_df(selected_columns):
     submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
     submission_repo.git_pull()
@@ -178,6 +189,29 @@ def on_filter_model_size_method_change(selected_columns):
         )
     return filter_component#.value
 
+def on_filter_model_size_method_change_quality(selected_columns):
+    updated_data = get_baseline_df_quality(selected_columns)
+    #print(updated_data)
+    # columns:
+    selected_columns = [item for item in QUALITY_TAB if item in selected_columns]
+    present_columns = MODEL_INFO_TAB_QUALITY + selected_columns
+    updated_data = updated_data[present_columns]
+    updated_data = updated_data.sort_values(by="Selected Score", ascending=False)
+    updated_data = convert_scores_to_percentage(updated_data)
+    updated_headers = present_columns
+    update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES.index(x)] for x in updated_headers]
+    # print(updated_data,present_columns,update_datatype)
+    filter_component = gr.components.Dataframe(
+        value=updated_data, 
+        headers=updated_headers,
+        type="pandas", 
+        datatype=update_datatype,
+        interactive=False,
+        visible=True,
+        )
+    return filter_component#.value
+
+
 block = gr.Blocks()
 
 
@@ -186,6 +220,7 @@ with block:
         LEADERBORAD_INTRODUCTION
     )
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
+        # Table 0
         with gr.TabItem("📊 VBench", elem_id="vbench-tab-table", id=1):
             with gr.Row():
                 with gr.Accordion("Citation", open=False):
@@ -230,6 +265,28 @@ with block:
             disable_b.click(disable_all, inputs=None, outputs=[checkbox_group]).then(fn=on_filter_model_size_method_change, inputs=[ checkbox_group], outputs=data_component)
             checkbox_group.change(fn=on_filter_model_size_method_change, inputs=[ checkbox_group], outputs=data_component)
 
+        with gr.TabItem("Video Quaity", elem_id="vbench-tab-table", id=1):
+            with gr.Row():
+                with gr.Column(scale=1.0):
+                    # selection for column part:
+                    checkbox_group_quality = gr.CheckboxGroup(
+                        choices=QUALITY_TAB,
+                        value=QUALITY_TAB,
+                        label="Evaluation Quality Dimension",
+                        interactive=True,
+                    )
+
+            data_component_quality = gr.components.Dataframe(
+                value=get_baseline_df_quality, 
+                headers=COLUMN_NAMES_QUALITY,
+                type="pandas", 
+                datatype=DATA_TITILE_TYPE,
+                interactive=False,
+                visible=True,
+                )
+    
+            checkbox_group_quality.change(fn=on_filter_model_size_method_change_quality, inputs=[checkbox_group_quality], outputs=data_component_quality)
+            
         # table 2
         with gr.TabItem("📝 About", elem_id="mvbench-tab-table", id=2):
             gr.Markdown(LEADERBORAD_INFO, elem_classes="markdown-text")