'''
Eye Mask - Stable Diffusion Web UI extension (embedded version) for mark and redraw eyes/faces.
Core logic is in eyemask.script_embedded.

Author: ilian.iliev
Since: 09.01.2023
'''

import os
import sys

from modules import scripts
sys.path.append(os.path.join(scripts.basedir(), 'scripts'))

from eyemask import constants, ui, script_embedded as eye_mask_script
import modules.shared as shared


class EyeMaskEmbeddedScript(scripts.Script):

    def __init__(self, *k, **kw):
        self.eye_mask_core = eye_mask_script.EyeMasksEmbeddedCore()
        self.eye_mask_ui = ui.EyeMaskUI(self)
        super().__init__()

    def title(self):
        return constants.script_name

    def show(self, is_img2img):
        try:
            return scripts.AlwaysVisible if shared.opts.em_show_embedded_version else False
        except Exception as e:
            return False

    def ui(self, is_img2img):
        return self.eye_mask_ui.render(is_img2img)

    def process(self, p, *args):
        return self.eye_mask_core.execute_process(p, *args)

    def postprocess(self, p, processed, *args):
        return self.eye_mask_core.execute_postprocess(p, processed, *args)