from cnn.music_processor import *



if sys.argv[1] == 'train':
    print("preparing all train data processing...")
    serv = f'./{sys.argv[2]:s}/*'
    music_for_train(serv, verbose=True)
    print("all train data processing done!")    

if sys.argv[1] == 'test':
    print("test data proccesing...")
    serv = f'./{sys.argv[2]:s}/*'
    music_for_test(serv)
    print("test data processing done!")