import argparse
from asyncio.log import logger
from Layoutlm_inference.ocr import prepare_batch_for_inference
from Layoutlm_inference.inference_handler import handle
import logging
import os

if __name__ == "__main__":
    # try:
    parser = argparse.ArgumentParser()
    parser.add_argument("--model_path", type=str)
    parser.add_argument("--images_path", type=str)
    args, _ = parser.parse_known_args()
    images_path = args.images_path
    image_files = os.listdir(images_path)
    images_path = [images_path+f'/{image_file}' for image_file in image_files]
    inference_batch = prepare_batch_for_inference(images_path)
    context = {"model_dir": args.model_path}
    output_ls = handle(inference_batch,context)
    
    print("output_ls",output_ls)
    
    
    # except Exception as err:
        # os.makedirs('log', exist_ok=True)
        # logging.basicConfig(filename='log/error_output.log', level=logging.ERROR,
        #                     format='%(asctime)s %(levelname)s %(name)s %(message)s')
        # logger = logging.getLogger(__name__)
        # logger.error(err)