refactoring

Procfile

@@ -1 +1 @@
-web: sh setup.sh && streamlit run src/main.py
+web: sh setup.sh && streamlit run src/app.py

run.sh

@@ -1 +1 @@
-streamlit run src/main.py
+streamlit run src/app.py

src/{main.py → app.py}

@@ -1,10 +1,9 @@
 import streamlit as st
 import albumentations as A
 
-from control import *
 from utils import (
     load_augmentations_config,
-    generate_executable_string,
+    get_params_string
 )
 from visuals import (
     show_transform_control,
@@ -14,8 +13,10 @@ from visuals import (
 )
 
 
-# main
+# show title
 st.title("Demo of Albumentations transforms")
+
+# select image
 image = select_image(path_to_images="images")
 
 # load the config
@@ -30,13 +31,14 @@ transform_name = st.sidebar.selectbox(
 param_values = show_transform_control(augmentations[transform_name])
 
 # apply the transformation to the image
-executable_string = generate_executable_string(transform_name, param_values)
-st.text(executable_string)
-exec("transform = A." + executable_string)
+transform = getattr(A, transform_name)(**param_values)
 augmented_image = transform(image=image)["image"]
 
-# show the images
+# show the params passed
+st.text("Params passed:" + get_params_string(param_values))
 st.text("Press R to update")
+
+# show the images
 st.image(
     [image, augmented_image],
     caption=["Original image", "Transformed image"],

src/utils.py

@@ -6,15 +6,18 @@ import streamlit as st
 
 
 @st.cache
-def load_image(image_name, path_to_folder="../images"):
+def load_image(
+    image_name: str, path_to_folder: str = "../images", bgr2rgb: bool = True
+):
     path_to_image = os.path.join(path_to_folder, image_name)
-    image = cv2.imread(path_to_image)
-    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    image = cv2.imread(path_to_image,)
+    if bgr2rgb:
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     return image
 
 
 @st.cache
-def get_images_list(path_to_folder):
+def get_images_list(path_to_folder: str) -> list:
     image_names_list = [
         x for x in os.listdir(path_to_folder) if x[-3:] in ["jpg", "peg", "png"]
     ]
@@ -22,15 +25,16 @@ def get_images_list(path_to_folder):
 
 
 @st.cache
-def load_augmentations_config(path_to_config: str = "configs/augmentations.json"):
+def load_augmentations_config(
+    path_to_config: str = "configs/augmentations.json",
+) -> dict:
     with open(path_to_config, "r") as config_file:
         augmentations = json.load(config_file)
     return augmentations
 
 
-def generate_executable_string(transform_name, param_values):
+def get_params_string(param_values: dict) -> str:
     params_string = ", ".join(
         [k + "=" + str(param_values[k]) for k in param_values.keys()] + ["p=1.0"]
     )
-    params_string = "(" + params_string + ")"
-    return transform_name + params_string
+    return params_string

src/visuals.py

@@ -4,6 +4,10 @@ from control import param2func
 from utils import get_images_list, load_image
 
 
+def show_logo():
+    st.image(load_image("logo.png", "../images"), format="PNG")
+
+
 def select_image(path_to_images: str = "images"):
     image_names_list = get_images_list(path_to_images)
     image_name = st.sidebar.selectbox("Select an image:", image_names_list)
@@ -14,10 +18,11 @@ def select_image(path_to_images: str = "images"):
 def show_transform_control(transform_params: dict):
     param_values = {}
     if len(transform_params) == 0:
-        st.sidebar.text(transform_name + " transform has no parameters")
+        st.sidebar.text("Transform has no parameters")
     else:
         for param in transform_params:
-            param_values[param["param_name"]] = param2func[param["type"]](**param)
+            control_function = param2func[param["type"]]
+            param_values[param["param_name"]] = control_function(**param)
     return param_values
 
 
@@ -25,13 +30,11 @@ def show_credentials():
     st.text("")
     st.text("")
     st.subheader("Credentials:")
-    st.text("Source: https://github.com/IliaLarchenko/albumentations-demo")
-    st.text(
-        "Albumentations library: https://github.com/albumentations-team/albumentations"
-    )
-    st.text("Image Source: https://www.pexels.com/royalty-free-images/")
+    st.text("Source: github.com/IliaLarchenko/albumentations-demo")
+    st.text("Albumentations library: github.com/albumentations-team/albumentations")
+    st.text("Image Source: pexels.com/royalty-free-images/")
 
 
-def show_docstring(object):
+def show_docstring(obj_with_ds):
     st.subheader("Docstring:")
-    st.text(str(object.__doc__))
+    st.text(str(obj_with_ds.__doc__))