allow to survey with particular methods

app.py

@@ -1,5 +1,5 @@
 from argparse import ArgumentParser
-import os, threading, time
+import os, threading, time, yaml
 from dataclasses import dataclass
 from datetime import datetime, timezone
 from pprint import pprint
@@ -58,7 +58,8 @@ class SurveyEngine:
             radius_size=gr.themes.sizes.radius_md,
         )
 
-        self._start_periodic_sync()
+        if not args.no_sync:
+            self._start_periodic_sync()
 
     def _start_periodic_sync(self):
         def _upload_periodically(path_local, path_remote):
@@ -101,7 +102,14 @@ class SurveyEngine:
 
     def _get_all_video_paths(self):
         video_dir = f"{self.local_dir}/video"
+
         method_list = sorted(os.listdir(video_dir))
+        
+        # filter methods
+        if len(self.args.method_filter) > 0:
+            method_filter = np.array(self.args.method_filter)
+            method_list = np.intersect1d(method_list, method_filter)
+
         video_name_list = sorted(os.listdir(f"{video_dir}/{method_list[0]}"))
 
         N_prompts = len(video_name_list)
@@ -549,23 +557,29 @@ class SurveyEngine:
         demo.launch(share=self.args.share, show_api=False)
 
 
-if __name__ == "__main__":
+def parse_args():
     parser = ArgumentParser()
+    
+    # use this config as HF space does not take args
+    parser.add_argument("--config", type=str, default="config.yaml")
+    
+    # these args are useful for local debugging
     parser.add_argument("--debug", action="store_true")
     parser.add_argument("--share", action="store_true")
-    parser.add_argument(
-        "--period_upload",
-        type=int,
-        default=300,
-        help="Period (sec) to upload response files in seconds, 300sec=5min",
-    )
-    parser.add_argument(
-        "--period_squash",
-        type=int,
-        default=43200,
-        help="Period (sec) to squash commits in seconds, 43200sec=12h",
-    )
+    parser.add_argument("--no_sync", action="store_true")
     args = parser.parse_args()
+    
+    with open(args.config, "r") as f:
+        config = yaml.safe_load(f)
+    
+    for key, value in config.items():
+        setattr(args, key, value)
+    pprint(vars(args))
+    return args
+
+
+if __name__ == "__main__":
+    args = parse_args()
 
     survey_engine = SurveyEngine(args)
     survey_engine.main()

config.yaml

@@ -0,0 +1,5 @@
+method_filter:
+  - method_A
+  - method_B
+period_upload: 300  #  Period (sec) to upload response files in seconds, 300sec=5min
+period_squash: 43200  # Period (sec) to squash commits in seconds, 43200sec=12h