From fdb7ebe6180a74d5348ea02639a53ef802967811 Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Mon, 4 Apr 2022 23:37:49 +0200
Subject: [PATCH] logging change

---
 image_prediction/utils/logger.py | 4 ++--
 scripts/run_pipeline.py          | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

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(