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

p = False
while True:
    for foldername, subfolders, filenames in os.walk(os.path.join('datasets', 'zombobase-current')):
        p = True
        for filename in filenames:
            if p == True:
                time.sleep(3)
                print('8' * 8)
                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'))

                p = False