File size: 1,175 Bytes
194b4ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
'''
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)
|