Fix animation

visual_anagrams/animate.py

@@ -118,21 +118,23 @@ def animate_two_view(
     frames = frames[-hold_duration//2:] + frames[:-hold_duration//2]
     images = frames
 
-    processed_frames = [images[0]]
+    # processed_frames = [images[0]]
 
-    for i in range(1, len(images)):
-        # Calculate the difference between current and previous frame
-        diff = ImageChops.difference(images[i], images[i - 1])
-        # Create a mask to isolate changes
-        mask = diff.convert("L").point(lambda x: 0 if x < 5 else 255, "1")
-        # Apply the mask to the current frame
-        new_frame = ImageChops.composite(images[i], processed_frames[-1], mask)
-        processed_frames.append(new_frame)
+    # for i in range(1, len(images)):
+    #     # Calculate the difference between current and previous frame
+    #     diff = ImageChops.difference(images[i], images[i - 1])
+    #     # Create a mask to isolate changes
+    #     mask = diff.convert("L").point(lambda x: 0 if x < 5 else 255, "1")
+    #     # Apply the mask to the current frame
+    #     new_frame = ImageChops.composite(images[i], processed_frames[-1], mask)
+    #     processed_frames.append(new_frame)
 
     # Save the frames as a GIF
-    imageio.mimsave(save_video_path,
-                    [np.array(frame) for frame in processed_frames],
-                    fps=30)
+    image_array = [imageio.core.asarray(frame) for frame in frames]
+
+    # Save as video
+    print('Making video...')
+    imageio.mimsave(save_video_path, image_array, fps=30)