diff --git a/helper_images.py b/helper.py similarity index 89% rename from helper_images.py rename to helper.py index 7c0b342..fe1da54 100644 --- a/helper_images.py +++ b/helper.py @@ -85,3 +85,12 @@ def cut_images(images_folder, output_folder): # Save the cropped image img.save(f"{output_folder}/{filename}") + + +def create_new_topic_dirs(topic, project_dir): + # /customers/___ + if not os.path.exists(f"{project_dir}/customers/{topic}"): + os.makedirs(f"{project_dir}/customers/{topic}") + # /sources/images/___ + if not os.path.exists(f"{project_dir}/sources/images/{topic}"): + os.makedirs(f"{project_dir}/sources/images/{topic}")