Skip to content
Snippets Groups Projects
cli.py 1.06 KiB
Newer Older
from src.worker import Worker
import argparse

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="anonymizer")
    parser.add_argument("input_path", type=str, help="Path to the input file")
    parser.add_argument("output_path", type=str, help="Path to the output file")
    parser.add_argument(
        "--replace-method",
        type=str,
        default="tag",
        choices=["delete", "tag", "pseudo"],
        help="Method of replacing tokens",
    )
    parser.add_argument(
        "--language",
        type=str,
        default="pl",
        choices=["pl"],
        help="Language of the input text",
    )
    parser.add_argument(
        "--configuration",
        type=str,
        default="ccl",
        choices=["ccl", "wiktorner_jsonl"],
        help="Configuration of the anonymizer",
    )
    args = parser.parse_args()

    worker = Worker(configuration=args.configuration)
    worker.process(
        args.input_path,
        {"method": args.replace_method, "language": args.language},
        args.output_path,
    )
    print("Done")