feat: add image preprocessing to get image path and resize images

front/image_preprocessing.py

@@ -0,0 +1,49 @@
+import os
+
+from PIL import Image
+
+
+def get_images(folder):
+    """Get images from a folder
+
+    Args:
+        folder (str): path to the folder
+
+    Returns:
+        images: list of images in the folder
+    """
+    images = []
+    extetntions = [".png", ".jpeg", ".jpg"]
+    for filename in os.listdir(folder):
+        if any(filename.endswith(ext) for ext in extetntions):
+            images.append(os.path.join(folder, filename))
+    return images
+
+
+def resize_image(image_path, target_height):
+    """Resize an image
+
+    Args:
+        image_path (str): path to the image
+        target_height (int): int to resize the height of the image
+
+    Returns:
+        img: resized images
+    """
+    img = Image.open(image_path)
+    img = img.resize((200, target_height))
+    return img
+
+
+def get_image_caption(image_path):
+    """Get the caption of an image
+
+    Args:
+        image_path (str): path to the image
+
+    Returns:
+        image_path: caption of the image
+    """
+    image_path = image_path.split("/")[-1]
+    image_path = image_path.split(".")[0]
+    return image_path