import argparse
import json

from image_prediction.pipeline import Pipeline


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("pdf")

    args = parser.parse_args()

    return args


def main(args):

    pipeline = Pipeline()

    with open(args.pdf, "rb") as f:
        predictions = pipeline(f.read())

    for prd in predictions:
        print(prd)


if __name__ == "__main__":
    args = parse_args()
    main(args)