import os
import time
import shutil
import train
import preprocessing
import datetime

start_file = False
while True:
    for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
        start_file = True
        for filename in filenames:
            if start_file == True:
                time.sleep(5)
                preprocessing.main(os.path.join('datasets', 'zombobase-current'),'labels', 'images')
                time.sleep(3)
                train.main()
                time.sleep(3)
                shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'images'))
                shutil.rmtree(os.path.join('datasets', 'zombobase-current', 'labels'))

                start_file = False