Adding console handler for logging

utils.py

@@ -16,6 +16,17 @@ current_directory = Path(__file__).resolve().parent
 logger = logging.getLogger("test-classifier")
 logger.setLevel(level=logging.DEBUG)
 
+# create console handler and set level to debug
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.DEBUG)
+# formatter for console handler
+console_formatter = logging.Formatter(
+    "%(asctime)s - %(name)s - %(levelname)s: %(message)s"
+)
+console_handler.setFormatter(console_formatter)
+# add console handler to logger
+logger.addHandler(console_handler)
+
 run_name_prompt = """
 Create a random name that sounds german or dutch
 The parts should be separated by underscores and contain only lowercase.
@@ -33,10 +44,12 @@ def initialize_run_directory(model: OpenAI | Llama2):
     run_directory = current_directory / f"runs/{run_name}"
     # TODO what if name exists?
     run_directory.mkdir(parents=True, exist_ok=False)
+    # create file handler and set level to debug
     file_handler = logging.FileHandler(run_directory / "output.log")
     file_handler.setLevel(logging.DEBUG)
     formatter = logging.Formatter("%(asctime)s %(levelname)s %(message)s")
     file_handler.setFormatter(formatter)
+    # add file handler to logger
     logger.addHandler(file_handler)
 
     logger.info(f"Hello my name is {run_name} and I live in {run_directory}")