pregenerate some samples

app.py

@@ -51,6 +51,41 @@ sketches_base64_strings = {
     name: get_base64_encoded_gif(f"sketches/{name}") for name in sketches
 }
 
+datasets = ["IAM", "IMGUR5K", "HierText"]
+models = ["Small-i", "Large-i", "Small-p"]
+query_modes = ["d+t", "r+d", "vanilla"]
+
+
+def pregenerate_videos():
+    for Dataset in datasets:
+        for Model in models:
+            inkml_path_base = f"./derendering_supp/{Model.lower()}_{Dataset}_inkml"
+            for mode in query_modes:
+                path = f"./derendering_supp/{Dataset}/images_sample"
+                if not os.path.exists(path):
+                    continue
+                samples = os.listdir(path)
+                selected_samples = random.sample(samples, len(samples) // 3)
+                for name in tqdm(
+                    selected_samples, desc=f"Generating {Model}-{Dataset}-{mode} videos"
+                ):
+                    example_id = name.strip(".png")
+                    inkml_file = os.path.join(
+                        inkml_path_base, mode, f"{example_id}.inkml"
+                    )
+                    if not os.path.exists(inkml_file):
+                        continue
+                    video_filename = f"{Model}_{Dataset}_{mode}_{example_id}.mp4"
+                    video_filepath = video_cache_dir / video_filename
+                    if not video_filepath.exists():
+                        img_path = os.path.join(path, name)
+                        img = load_and_pad_img_dir(img_path)
+                        ink = inkml_to_ink(inkml_file)
+                        plot_ink_to_video(ink, str(video_filepath), input_image=img)
+
+
+pregenerate_videos()
+
 
 def demo(Dataset, Model, Output_Format):
     if Model == "Small-i":