main.py

"""Implementation of text error correction service."""
import argparse
import nlp_ws
from src.speller2_worker import Speller2Worker


def get_args():
    """Gets command line arguments."""
    parser = argparse.ArgumentParser(description="speller2")

    subparsers = parser.add_subparsers(dest="mode")
    subparsers.required = True

    subparsers.add_parser(
        "service",
        help="Run as a service")

    return parser.parse_args()


def main():
    """Runs the program."""
    args = get_args()

    generators = {
        "service": lambda: nlp_ws.NLPService.main(Speller2Worker),
    }

    gen_fn = generators.get(args.mode, lambda: None)
    gen_fn()


if __name__ == "__main__":
    main()