Update app.py

app.py

@@ -237,6 +237,8 @@ def get_final_score_quality(df, selected_columns):
         df.insert(1, 'Selected Score', selected_score)
     return df
 
+
+
 def get_baseline_df():
     submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
     submission_repo.git_pull()
@@ -270,6 +272,17 @@ def get_baseline_df_i2v():
     df = convert_scores_to_percentage(df)
     return df
 
+def get_baseline_df_long():
+    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(LONG_DIR)
+    df = get_final_score(df, checkbox_group.value)
+    df = df.sort_values(by="Selected Score", ascending=False)
+    present_columns = MODEL_INFO + checkbox_group.value
+    df = df[present_columns]
+    df = convert_scores_to_percentage(df)
+    return df
+
 def get_all_df(selected_columns, dir=CSV_DIR):
     submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
     submission_repo.git_pull()
@@ -294,6 +307,15 @@ def get_all_df_i2v(selected_columns, dir=I2V_DIR):
     df = df.sort_values(by="Selected Score", ascending=False)
     return df
 
+def get_all_df_long(selected_columns, dir=LONG_DIR):
+    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(dir)
+    df = get_final_score(df, selected_columns)
+    df = df.sort_values(by="Selected Score", ascending=False)
+    return df
+
+
 def convert_scores_to_percentage(df):
     # 对DataFrame中的每一列（除了'name'列）进行操作
 
@@ -318,6 +340,7 @@ def disable_all():
 def enable_all():
     return gr.update(value=TASK_INFO)
 
+# select function
 def on_filter_model_size_method_change(selected_columns):
     updated_data = get_all_df(selected_columns, CSV_DIR)
     #print(updated_data)
@@ -382,6 +405,25 @@ def on_filter_model_size_method_change_i2v(selected_columns):
         )
     return filter_component#.value
 
+def on_filter_model_size_method_change_long(selected_columns):
+    updated_data = get_all_df_long(selected_columns, LONG_DIR)
+    selected_columns = [item for item in TASK_INFO if item in selected_columns]
+    present_columns = MODEL_INFO + 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]
+    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()
 
 
@@ -436,6 +478,7 @@ 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)
 
+        # Table 1
         with gr.TabItem("Video Quaity", elem_id="vbench-tab-table", id=2):
             with gr.Accordion("INSTRUCTION", open=False):
                     citation_button = gr.Textbox(
@@ -464,7 +507,8 @@ with block:
                 )
     
             checkbox_group_quality.change(fn=on_filter_model_size_method_change_quality, inputs=[checkbox_group_quality], outputs=data_component_quality)
-            
+        
+        # Table i2v
         with gr.TabItem("VBench-I2V", elem_id="vbench-tab-table", id=3):
             with gr.Accordion("NOTE", open=False):
                     i2v_note_button = gr.Textbox(
@@ -493,13 +537,57 @@ with block:
                 )
     
             checkbox_group_i2v.change(fn=on_filter_model_size_method_change_i2v, inputs=[checkbox_group_i2v], outputs=data_component_i2v)
+        
+        with gr.TabItem("📊 VBench-Long", elem_id="vbench-tab-table", id=4):
+            with gr.Row():
+                with gr.Accordion("INSTRUCTION", open=False):
+                    citation_button = gr.Textbox(
+                        value=LONG_CLAIM_TEXT,
+                        label="",
+                        elem_id="long-ins-button",
+                        lines=2,
+                    )
+    
+            gr.Markdown(
+                TABLE_INTRODUCTION
+            )
+            with gr.Row():
+                with gr.Column(scale=0.2):
+                    choosen_q_long = gr.Button("Select Quality Dimensions")
+                    choosen_s_long = gr.Button("Select Semantic Dimensions")
+                    enable_b_long = gr.Button("Select All")
+                    disable_b_long = gr.Button("Deselect All")
+
+                with gr.Column(scale=0.8):
+                    checkbox_group_long = gr.CheckboxGroup(
+                        choices=TASK_INFO,
+                        value=DEFAULT_INFO,
+                        label="Evaluation Dimension",
+                        interactive=True,
+                    )
+
+            data_component = gr.components.Dataframe(
+                value=get_baseline_df_long, 
+                headers=COLUMN_NAMES,
+                type="pandas", 
+                datatype=DATA_TITILE_TYPE,
+                interactive=False,
+                visible=True,
+                height=700,
+                )
+    
+            choosen_q_long.click(choose_all_quailty, inputs=None, outputs=[checkbox_group_long]).then(fn=on_filter_model_size_method_change_long, inputs=[ checkbox_group_long], outputs=data_component)
+            choosen_s_long.click(choose_all_semantic, inputs=None, outputs=[checkbox_group_long]).then(fn=on_filter_model_size_method_change_long, inputs=[ checkbox_group_long], outputs=data_component)
+            enable_b_long.click(enable_all, inputs=None, outputs=[checkbox_group_long]).then(fn=on_filter_model_size_method_change_long, inputs=[ checkbox_group_long], outputs=data_component)
+            disable_b_long.click(disable_all, inputs=None, outputs=[checkbox_group_long]).then(fn=on_filter_model_size_method_change_long, inputs=[ checkbox_group_long], outputs=data_component)
+            checkbox_group_long.change(fn=on_filter_model_size_method_change_long, inputs=[checkbox_group_long], outputs=data_component)
             
-        # table 2
-        with gr.TabItem("📝 About", elem_id="mvbench-tab-table", id=4):
+        # table info
+        with gr.TabItem("📝 About", elem_id="mvbench-tab-table", id=5):
             gr.Markdown(LEADERBORAD_INFO, elem_classes="markdown-text")
         
-        # table 3 
-        with gr.TabItem("🚀 Submit here! ", elem_id="mvbench-tab-table", id=4):
+        # table submission 
+        with gr.TabItem("🚀 Submit here! ", elem_id="mvbench-tab-table", id=6):
             gr.Markdown(LEADERBORAD_INTRODUCTION, elem_classes="markdown-text")
 
             with gr.Row():