2022-09-03 19:32:45 +03:00
|
|
|
import modules.scripts as scripts
|
|
|
|
import gradio as gr
|
|
|
|
|
|
|
|
from modules.processing import Processed
|
|
|
|
from modules.shared import opts, cmd_opts, state
|
|
|
|
|
|
|
|
class Script(scripts.Script):
|
2022-09-17 00:06:21 +03:00
|
|
|
|
2022-09-03 19:32:45 +03:00
|
|
|
def title(self):
|
|
|
|
return "Custom code"
|
|
|
|
|
2022-09-04 13:08:06 +03:00
|
|
|
|
|
|
|
def show(self, is_img2img):
|
2022-09-03 19:32:45 +03:00
|
|
|
return cmd_opts.allow_code
|
|
|
|
|
|
|
|
def ui(self, is_img2img):
|
2023-01-05 10:27:09 +03:00
|
|
|
elem_prefix = ('img2img' if is_img2img else 'txt2txt') + '_script_custom_code_'
|
2023-01-05 00:03:32 +03:00
|
|
|
|
|
|
|
code = gr.Textbox(label="Python code", lines=1, elem_id=elem_prefix + "code")
|
2022-09-03 19:32:45 +03:00
|
|
|
|
|
|
|
return [code]
|
|
|
|
|
2022-09-17 00:06:21 +03:00
|
|
|
|
2022-09-03 19:32:45 +03:00
|
|
|
def run(self, p, code):
|
2022-09-04 13:08:06 +03:00
|
|
|
assert cmd_opts.allow_code, '--allow-code option must be enabled'
|
2022-09-03 19:32:45 +03:00
|
|
|
|
|
|
|
display_result_data = [[], -1, ""]
|
|
|
|
|
|
|
|
def display(imgs, s=display_result_data[1], i=display_result_data[2]):
|
|
|
|
display_result_data[0] = imgs
|
|
|
|
display_result_data[1] = s
|
|
|
|
display_result_data[2] = i
|
|
|
|
|
|
|
|
from types import ModuleType
|
|
|
|
compiled = compile(code, '', 'exec')
|
|
|
|
module = ModuleType("testmodule")
|
|
|
|
module.__dict__.update(globals())
|
|
|
|
module.p = p
|
|
|
|
module.display = display
|
|
|
|
exec(compiled, module.__dict__)
|
|
|
|
|
|
|
|
return Processed(p, *display_result_data)
|
2022-09-17 00:06:21 +03:00
|
|
|
|
|
|
|
|