diff --git a/src/generate_images.py b/src/generate_images.py index 8926477..f1b69b3 100644 --- a/src/generate_images.py +++ b/src/generate_images.py @@ -68,6 +68,8 @@ def generate_concat(images, dest): for img in images: content += f"\nfile {img}\nduration 8" + content += "\nfile "+images[-1] + with open(dest, "w") as f: f.write(content) @@ -103,12 +105,12 @@ if __name__ == "__main__": generate_images(sys.argv[1], sys.argv[2]) img_dir = os.path.dirname(sys.argv[2]) - images = [ + images = sorted([ img for img in os.listdir(img_dir) if img.startswith(os.path.basename(sys.argv[2])) and img.endswith(".png") and "short" not in img - ] - short_images = [ + ]) + short_images = sorted([ img for img in os.listdir(img_dir) if img.startswith(os.path.basename(sys.argv[2])) and img.endswith(".png") and "short" in img - ] + ]) adjust_images([ os.path.join(img_dir, img) for img in images ])