import os
from cv_analysis.config import CONFIG
from cv_analysis.utils.display import save_mpl

LEVEL = CONFIG.visual_logging.level
OUTPUT_FOLDER = CONFIG.visual_logging.output_folder


class VisualLogger:
    def __init__(self):
        self.level_is_debug = LEVEL == "DEBUG"
        self.output_folder = OUTPUT_FOLDER
        if not os.path.exists(self.output_folder):
            os.mkdir(self.output_folder)

    def debug(self, img, name):
        if self.level_is_debug:
            output_path = os.path.join(self.output_folder, name)
            save_mpl(img, output_path)


vizlogger = VisualLogger()