fix: resize gifs and copy, add pytest

src/evaluation_system/api/plugin_manager.py

@@ -43,7 +43,7 @@
 from typing import Any, Dict, Iterator, Optional, Sequence, Tuple, TypeVar, Union, cast
 
 from django.db.models.query import QuerySet
-from PIL import Image
+from PIL import Image, ImageSequence
 from rich import print as pprint
 from typing_extensions import TypedDict
 
@@ -609,11 +609,33 @@ def _preview_convert(source_path: str, dest_path: str, width: int = 800) -> None
         so height is calculated automatically. Default is 800
     """
 
-    with Image.open(source_path) as img:
+    def resizer(img, width):
         x, y = img.size
         height = 2 * (int(y / (x / width)) // 2)
-        im_resize = img.resize((width, height))
-        im_resize.save(dest_path)
+        return img.resize((width, height))
+
+    with Image.open(source_path) as img:
+        if img.format == "GIF":
+            frames = []
+            duration = []
+            for frame in ImageSequence.Iterator(img):
+                im_resize = resizer(frame, width)
+                frames.append(im_resize)
+                duration.append(frame.info.get("duration", 0))
+            frames[0].save(
+                dest_path,
+                format="GIF",
+                save_all=True,
+                append_images=frames[1:],
+                loop=0,
+                duration=duration,
+                optimize=True,
+                quality=95,
+            )
+        else:
+            im_resize = resizer(img, width)
+            im_resize.save(dest_path)
+
     os.chmod(dest_path, 509)
 
 

src/evaluation_system/tests/plugin_manager_test.py

@@ -436,6 +436,11 @@ def test_preview_generation(dummy_env):
         s = os.path.dirname(__file__) + "/test_output/test_image.png"
         pm._preview_convert(s, d)
         assert os.path.isfile(d)
+    with tempfile.TemporaryDirectory() as td:
+        d = str(Path(td) / "tmp.gif")
+        s = os.path.dirname(__file__) + "/test_output/test_animation.gif"
+        pm._preview_convert(s, d)
+        assert os.path.isfile(d)
 
     r = pm._preview_generate_name("dummy", {})
     assert "dummy" in r

src/evaluation_system/tests/plugin_test.py

@@ -725,7 +725,7 @@ def test_prepare_output(dummy_plugin):
         assert meta_data["type"] == check["type"]
         assert meta_data["caption"] == "Manually added result"
     meta_data = plugin._run_tool({"folder": str(fn.parent)})
-    assert len(meta_data) == 2
+    assert len(meta_data) == 3
 
 
 @mock.patch("os.getpid", lambda: 12345)