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


class VisualLogger:
    def __init__(self, level, output_folder):
        self.level = level
        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)
    
    def level_is_debug(self):
        return self.level == "DEBUG"


vizlogger = VisualLogger(CONFIG.visual_logging.level, CONFIG.visual_logging.output_folder)