Merge pull request #6329 from Kryptortio/add_even_more_element_ids

Add additional elem_id/HTML ids (again)

1 files changed, 6 insertions, 1 deletions

diff --git a/scripts/custom_code.py b/scripts/custom_code.py
index 22e7b77a..9ce1f650 100644
--- a/scripts/custom_code.py
+++ b/scripts/custom_code.py
@@ -3,18 +3,23 @@ import gradio as gr
 

 from modules.processing import Processed

 from modules.shared import opts, cmd_opts, state

+import re

 

 class Script(scripts.Script):

 

     def title(self):

         return "Custom code"

 

+    def elem_id(self, item_id):

+        gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id

+        gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)

+        return gen_elem_id

 

     def show(self, is_img2img):

         return cmd_opts.allow_code

 

     def ui(self, is_img2img):

-        code = gr.Textbox(label="Python code", lines=1)

+        code = gr.Textbox(label="Python code", lines=1, elem_id=self.elem_id("code"))

 

         return [code]