import argparse
import json

from image_prediction.config import CONFIG
from image_prediction.pipeline import load_pipeline
from image_prediction.utils import get_logger
import logging

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

    args = parser.parse_args()

    return args


def main(args):
    pipeline = load_pipeline(verbose=True)

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

    for prd in predictions:
        print(json.dumps(prd, indent=2))


if __name__ == "__main__":
    # logging.basicConfig(level=logging.DEBUG)
    logger = get_logger()

    logger.info("111111111111111111111111111111111")
    print(logger.level)
    print(logging.DEBUG)
    args = parse_args()
    # main(args)