import glob

from tqdm import tqdm

from generator import TrainingImageGenerator


def main():
    """
    Create training masks for single cells and aggregates for images in 
    directory using TrainingImageGenerator

    Returns
    -------
    None.

    """
    
    names = []
    # Define directory
    for filename in glob.glob(
            r'PATH\TO\IMAGES\*.tiff'):
        names.append(filename)
    for name in tqdm(names):
        TIG = TrainingImageGenerator(name, contour_size_conformity=0.4)
        TIG.run()
  
    
if __name__ == '__main__':
    main()