diff --git a/main.py b/main.py index 626358d..f692179 100755 --- a/main.py +++ b/main.py @@ -74,8 +74,11 @@ def main(): else: newname = file if args.classify: + print(f"Existing categories: {os.listdir(outputFolder)}") category: str = input("Category: ") - if not os.path.exists(outputFolder + category) or not os.path.isdir( + if category == "delete": + os.remove(inputFolder + file) + elif not os.path.exists(outputFolder + category) or not os.path.isdir( outputFolder + category ): os.mkdir(outputFolder + category)