added image_width CLI param

src/app.py

@@ -3,7 +3,7 @@ import streamlit as st
 import albumentations as A
 
 
-from utils import load_augmentations_config, get_path_to_the_image
+from utils import load_augmentations_config, get_arguments
 from visuals import (
     show_transform_control,
     select_image,
@@ -11,11 +11,12 @@ from visuals import (
     show_docstring,
 )
 
-# get the path to images
-path_to_images = get_path_to_the_image()
+# get CLI params: the path to images and image width
+path_to_images, width_original = get_arguments()
 if not os.path.isdir(path_to_images):
     st.title("There is no directory: " + path_to_images)
 else:
+    # select image
     status, image = select_image(path_to_images)
     if status == 0:
         st.title("Can't load image from: " + path_to_images)
@@ -23,8 +24,6 @@ else:
         # show title
         st.title("Demo of Albumentations")
 
-        # select image
-
         placeholder_params = {
             "image_width": image.shape[1],
             "image_height": image.shape[0],
@@ -58,7 +57,6 @@ else:
         # st.write(data["replay"])
 
         # show the images
-        width_original = 400
         width_transformed = int(
             width_original / image.shape[1] * augmented_image.shape[1]
         )

src/utils.py

@@ -7,11 +7,12 @@ import streamlit as st
 
 
 @st.cache
-def get_path_to_the_image():
+def get_arguments():
     parser = argparse.ArgumentParser()
     parser.add_argument("--image_folder", default="images")
+    parser.add_argument("--image_width", default=400, type=int)
     args = parser.parse_args()
-    return getattr(args, "image_folder")
+    return getattr(args, "image_folder"), getattr(args, "image_width")
 
 
 @st.cache