#!/usr/bin/env python

import os
import argparse
import tkinter as tk

from simplabel import label_data_manual

#Setup parser
ap = argparse.ArgumentParser()
ap.add_argument("-d", "--directory", default=os.getcwd(), help="Path of the directory")
ap.add_argument("-c", "--categories", nargs='*', help="List of categories")

args = ap.parse_args()

#TODO: implement persistance of categories when the same directory is passed again
#TODO: implement a reset option to delete persistent categories and data

root = tk.Tk() 
rawDirectory = args.directory
categories = args.categories
MyApp = label_data_manual.ImageClassifier(root, rawDirectory, categories)
tk.mainloop()