diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 16:32:45 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-03 16:32:45 +0000 |
| commit | 592334f322d403679a125225afb5ff0114935edd (patch) | |
| tree | 3f8bc6963488eba5d4a1421e98de6a0f49adc439 /scripts/custom_code.py | |
| parent | 595c827bd31773cc98eb6e87b11090960a32b2a2 (diff) | |
| download | stable-diffusion-webui-gfx803-592334f322d403679a125225afb5ff0114935edd.tar.gz stable-diffusion-webui-gfx803-592334f322d403679a125225afb5ff0114935edd.tar.bz2 stable-diffusion-webui-gfx803-592334f322d403679a125225afb5ff0114935edd.zip | |
scripts
Diffstat (limited to 'scripts/custom_code.py')
| -rw-r--r-- | scripts/custom_code.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/scripts/custom_code.py b/scripts/custom_code.py new file mode 100644 index 00000000..b359050a --- /dev/null +++ b/scripts/custom_code.py @@ -0,0 +1,40 @@ +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):
+ def title(self):
+ return "Custom code"
+
+ def enabled(self):
+ return cmd_opts.allow_code
+
+ def ui(self, is_img2img):
+ code = gr.Textbox(label="Python code", visible=False, lines=1)
+
+ return [code]
+
+ def run(self, p, code):
+ if not cmd_opts.allow_code:
+ return
+
+ 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)
+
|