import sys
from Processor import Processor
from PathWalker import PathWalker
arguments = sys.argv
processor = Processor()
pathWalker = PathWalker()
if arguments[0] == '--cluster':
dir = arguments[1]
clustersNum = arguments[2]
files = pathWalker.getFilesInPath(dir, '*.wav')
print processor.cluster(files, clustersNum)
elif arguments[0] == '--classify':
file = arguments[1]
print processor.classify(file)
else:
print 'Bad arguments'
