add cache video to speed up

app.py

@@ -3,10 +3,15 @@ import os
 import random
 import datetime
 from utils import *
+from pathlib import Path
 
 file_url = "https://storage.googleapis.com/derendering_model/derendering_supp.zip"
 filename = "derendering_supp.zip"
 
+# Cache videos to speed up demo
+video_cache_dir = Path("./cached_videos")
+video_cache_dir.mkdir(exist_ok=True)
+
 download_file(file_url, filename)
 unzip_file(filename)
 print("Downloaded and unzipped the file.")
@@ -86,17 +91,17 @@ def demo(Dataset, Model, Output_Format):
             inkml_file = os.path.join(inkml_path, mode, example_id + ".inkml")
             text_field = parse_inkml_annotations(inkml_file)["textField"]
             output_text = f"{plot_title[mode]}{text_field}"
-            # Text output for three modes
-            # d+t: OCR recognition input to the model
-            # r+d: Recognition from the model
-            # vanilla: None
             text_outputs.append(output_text)
             ink = inkml_to_ink(inkml_file)
 
+            video_filename = f"{Model}_{Dataset}_{mode}_{example_id}.mp4"
+            video_filepath = video_cache_dir / video_filename
+
             if Output_Format == "Image+Video":
-                video_filename = mode + ".mp4"
-                plot_ink_to_video(ink, video_filename, input_image=img)
-                video_outputs.append(video_filename)
+                if not video_filepath.exists():
+                    plot_ink_to_video(ink, str(video_filepath), input_image=img)
+                    print("Cached video at:", video_filepath)
+                video_outputs.append("./" + str(video_filepath))
             else:
                 video_outputs.append(None)