diff --git a/image_prediction/utils/logger.py b/image_prediction/utils/logger.py index 1fbb456..7fd8473 100644 --- a/image_prediction/utils/logger.py +++ b/image_prediction/utils/logger.py @@ -12,8 +12,8 @@ def make_logger_getter(): handler = logging.StreamHandler() handler.setLevel(CONFIG.service.logging_level) - log_format = "[%(levelname)s]: %(message)s" - formatter = logging.Formatter(log_format) + log_format = "%(asctime)s %(levelname)-8s %(message)s" + formatter = logging.Formatter(log_format, datefmt="%Y-%m-%d %H:%M:%S") handler.setFormatter(formatter) logger.addHandler(handler) diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py index 4d74413..81fa140 100644 --- a/scripts/run_pipeline.py +++ b/scripts/run_pipeline.py @@ -4,8 +4,11 @@ import os from glob import glob from image_prediction.pipeline import load_pipeline +from image_prediction.utils import get_logger from image_prediction.utils.pdf_annotation import annotate_pdf +logger = get_logger() + def parse_args(): parser = argparse.ArgumentParser() @@ -21,6 +24,7 @@ def parse_args(): def process_pdf(pipeline, pdf_path): with open(pdf_path, "rb") as f: + logger.info(f"Processing {pdf_path}") predictions = list(pipeline(f.read())) annotate_pdf(