import argparse
import json
import requests

from vidocp.utils.preprocessing import open_pdf


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--pdf_path", required=True, help="path to PDF file")
    parser.add_argument("--first_page", type=int, required=True, help="path to PDF file")
    parser.add_argument("--last_page", type=int, required=False, default=None, help="path to PDF file")
    args = parser.parse_args()

    return args


def main(args):
    
    #data = open_pdf(args.pdf_path, args.first_page, args.last_page)
    # params = json.dumps({
    #     "pdf_path": "a",#args.pdf_path, 
    #     "first_page": 4,#args.first_page, 
    #     "last_page": 6#args.last_page
    # })
    response = requests.post("http://127.0.0.1:5000", data=open(args.pdf_path, "rb"))#, json=params)
    response.raise_for_status()
    predictions = response.json()

    print(json.dumps(predictions, indent=2))


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