Update app.py

app.py

@@ -11,12 +11,7 @@ from datasets import config
 hf_token = os.environ['hf_token']  # 确保环境变量中有你的令牌
 
 local_dir = "VBench_sampled_video"  # 本地文件夹路径
-# dataset = load_dataset("Vchitect/VBench_sampled_video")
-# print(os.listdir("~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"))
-# root = "~/.cache/huggingface/datasets/Vchitect___VBench_sampled_video/"
-# print(config.HF_DATASETS_CACHE)
-# root = config.HF_DATASETS_CACHE
-# print(root)
+
 def print_directory_contents(path, indent=0):
     # 打印当前目录的内容
     try:
@@ -32,10 +27,10 @@ def print_directory_contents(path, indent=0):
 os.makedirs(local_dir, exist_ok=True)
 hf_api = HfApi(endpoint="https://huggingface.co", token=hf_token)
 hf_api = HfApi(token=hf_token)
-repo_id = "Vchitect/VBench_sampled_video"
+repo_id = "Vchitect/VBench-2.0_sampled_videos"
 
 model_names=[]
-for i in hf_api.list_repo_tree('Vchitect/VBench_sampled_video',repo_type='dataset'):
+for i in hf_api.list_repo_tree('Vchitect/VBench-2.0_sampled_videos',repo_type='dataset'):
     model_name = i.path
     if '.git' not in model_name and '.md' not in model_name:
         model_names.append(model_name)
@@ -51,8 +46,24 @@ with open("videos_by_dimension.json") as f:
 # with open("all_videos.json") as f:
     # all_videos = json.load(f)
 
-types = ['appearance_style', 'color', 'temporal_style', 'spatial_relationship', 'temporal_flickering', 'scene', 'multiple_objects', 'object_class', 'human_action', 'overall_consistency', 'subject_consistency']
-
+types =['Material',
+ 'Human_Anatomy',
+ 'Complex_Plot',
+ 'Multi-View_Consistency',
+ 'Motion_Rationality',
+ 'Human_Clothes',
+ 'Instance_Preservation',
+ 'Motion_Order_Understanding',
+ 'Dynamic_Attribute',
+ 'Dynamic_Spatial_Relationship',
+ 'Thermotics',
+ 'Human_Interaction',
+ 'Camera_Motion',
+ 'Diversity',
+ 'Composition',
+ 'Complex_Landscape',
+ 'Human_Identity',
+ 'Mechanics']
 def get_video_path_local(model_name, type, prompt):
     if 'Show-1' in model_name:
         video_path_subfolder = os.path.join(model_name, type, 'super2')
@@ -61,8 +72,8 @@ def get_video_path_local(model_name, type, prompt):
     else:
         video_path_subfolder = os.path.join(model_name, type)
 
-    if model_name == 'cogvideo':
-        prompt = prompt.replace(".mp4",".gif")
+    # if model_name == 'cogvideo':
+    #     prompt = prompt.replace(".mp4",".gif")
     
     try:
         return hf_api.hf_hub_download(
@@ -121,15 +132,15 @@ def record_user_feedback_a(model_name1, model_name2, type, prompt):
     # 0 means model A  better, 1 means model B better,  -1 means tie; 
     hf_api.hf_hub_download(
             repo_id = "Vchitect/VBench_human_annotation",
-            filename = "arena_feedback.csv",
+            filename = "VBench2_arena_feedback.csv",
             repo_type = "dataset",
             local_dir = './'
         )
-    with open("arena_feedback.csv",'a') as f:
+    with open("VBench2_arena_feedback.csv",'a') as f:
         f.write(f"{model_name1}\t{model_name2}\t{type}\t{prompt}\t{0}\n")
     hf_api.upload_file(
-        path_or_fileobj="arena_feedback.csv",
-        path_in_repo="arena_feedback.csv",
+        path_or_fileobj="VBench2_arena_feedback.csv",
+        path_in_repo="VBench2_arena_feedback.csv",
         repo_id="Vchitect/VBench_human_annotation",
         token=hf_token,
         repo_type="dataset",
@@ -141,15 +152,15 @@ def record_user_feedback_b(model_name1, model_name2, type, prompt):
     # 0 means model A  better, 1 means model B better , -1 means tie;
     hf_api.hf_hub_download(
             repo_id = "Vchitect/VBench_human_annotation",
-            filename = "arena_feedback.csv",
+            filename = "VBench2_arena_feedback.csv",
             repo_type = "dataset",
             local_dir = './'
         )
-    with open("arena_feedback.csv",'a') as f:
+    with open("VBench2_arena_feedback.csv",'a') as f:
         f.write(f"{model_name1}\t{model_name2}\t{type}\t{prompt}\t{1}\n")  
     hf_api.upload_file(
-        path_or_fileobj="arena_feedback.csv",
-        path_in_repo="arena_feedback.csv",
+        path_or_fileobj="VBench2_arena_feedback.csv",
+        path_in_repo="VBench2_arena_feedback.csv",
         repo_id="Vchitect/VBench_human_annotation",
         token=hf_token,
         repo_type="dataset",
@@ -161,15 +172,15 @@ def record_user_feedback_tie(model_name1, model_name2, type, prompt):
     # 0 means model A  better, 1 means model B better , -1 means tie;
     hf_api.hf_hub_download(
             repo_id = "Vchitect/VBench_human_annotation",
-            filename = "arena_feedback.csv",
+            filename = "VBench2_arena_feedback.csv",
             repo_type = "dataset",
             local_dir = './'
         )
-    with open("arena_feedback.csv",'a') as f:
+    with open("VBench2_arena_feedback.csv",'a') as f:
         f.write(f"{model_name1}\t{model_name2}\t{type}\t{prompt}\t{-1}\n")  
     hf_api.upload_file(
-        path_or_fileobj="arena_feedback.csv",
-        path_in_repo="arena_feedback.csv",
+        path_or_fileobj="VBench2_arena_feedback.csv",
+        path_in_repo="VBench2_arena_feedback.csv",
         repo_id="Vchitect/VBench_human_annotation",
         token=hf_token,
         repo_type="dataset",