Merge pull request #1 from ia-labo/feat/add_streamlit

feat: add Streamlit app with image display and dropdown selection

front/app.py

@@ -0,0 +1,49 @@
+"""App to display images in a gallery"""
+
+import streamlit as st
+from image_preprocessing import get_image_caption, get_images, resize_image
+
+
+def image_gallery(images: list) -> None:
+    """Display a gallery of images in a streamlit app
+
+    Args:
+        images (list): list of images to display
+    """
+    st.title("Welcome")
+
+    columns = st.columns(3)
+    for index, image in enumerate(images):
+        with columns[index % 3]:
+            resized_image = resize_image(image)
+            image_caption = get_image_caption(image)
+            st.image(
+                resized_image,
+                width=200,
+                use_container_width=True,
+                caption=image_caption,
+            )
+
+
+def sidebar() -> str:
+    """Create a sidebar to select the park
+
+    Returns:
+        selected_park: selected park in the sidebar
+    """
+    selected_park = st.sidebar.selectbox(
+        "Park List", ["Al Khaldiyah Park", "Family Park"]
+    )
+    return selected_park
+
+
+def main() -> None:
+    """Main function to run the app"""
+    path = "../images/"
+    park = sidebar()
+    images = get_images(path + park)
+    image_gallery(images)
+
+
+if __name__ == "__main__":
+    main()

front/image_preprocessing.py

@@ -0,0 +1,52 @@
+"""File to preprocess images"""
+
+import os
+
+from PIL import Image
+
+
+def get_images(folder: str) -> list:
+    """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(img_path: str, target_w: int = 200, target_h: int = 200) -> Image:
+    """Resize an image
+
+    Args:
+        img_path (str): path to the image
+        target_w (int): int to resize the width of the image
+        target_h (int): int to resize the height of the image
+
+    Returns:
+        img: resized images
+    """
+    img = Image.open(img_path)
+    img = img.resize((target_w, target_h))
+    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