Update helpers

editor/ploting/display_images.py

@@ -0,0 +1,25 @@
+import matplotlib.pyplot as plt
+from typing import List
+from PIL.Image import Image
+from math import ceil
+
+
+def display_images(images: List[Image], titles: List[str], images_per_row: int = 3):
+    fig, axes = plt.subplots(
+        nrows=ceil(len(images) / images_per_row),
+        ncols=min(images_per_row, len(images)),
+        figsize=(12, 8),
+    )
+
+    axes = axes.flatten()
+
+    for i, (title, image) in enumerate(zip(titles, images)):
+        axes[i].imshow(image)
+        axes[i].axis("off")
+        axes[i].set_title(title)
+
+    for i in range(len(images), len(axes)):
+        axes[i].axis("off")
+
+    plt.tight_layout()
+    plt.show()