diff --git a/methods/bulk_methods.py b/methods/bulk_methods.py index 5fb12bd..01deb97 100644 --- a/methods/bulk_methods.py +++ b/methods/bulk_methods.py @@ -2,8 +2,6 @@ from PIL import Image, ImageFilter from skimage import feature import numpy as np -methods = ["Sharpen", "Edge detection", "Canny edge detection"] - def edge(file_name): image = Image.open(file_name) @@ -31,3 +29,5 @@ methods_funcs = { "Edge detection": edge, "Canny edge detection": canny_edge, } + +methods = list(methods_funcs.keys()) diff --git a/methods/stack_methods.py b/methods/stack_methods.py index 54526ae..b6550dc 100644 --- a/methods/stack_methods.py +++ b/methods/stack_methods.py @@ -2,8 +2,6 @@ from PIL import Image, ImageChops from tqdm import tqdm import numpy as np -methods = ["Denoise", "StarTracks", "Noise extractor", "Untrack"] - def denoise(files): images = [np.asarray(Image.open(file)) for file in tqdm(files)] @@ -46,3 +44,5 @@ methods_funcs = { "Noise extractor": noise_extractor, "Untrack": untrack, } + +methods = list(methods_funcs.keys()) diff --git a/tabs/Bulk processing/app.py b/tabs/Bulk processing/app.py index 7948d6a..ca32247 100644 --- a/tabs/Bulk processing/app.py +++ b/tabs/Bulk processing/app.py @@ -3,9 +3,7 @@ from processing.bulk import bulk_processing from methods.bulk_methods import methods with gr.Blocks() as app: - gr.Markdown( - "Mass processing of images one at a time and saving to video if needed. # **WIP, not working**" - ) + gr.Markdown("Mass processing of images one at a time.") with gr.Row(): with gr.Column(): directory = gr.Text(