Merge pull request #6329 from Kryptortio/add_even_more_element_ids

Add additional elem_id/HTML ids (again)

1 files changed, 11 insertions, 5 deletions

diff --git a/scripts/prompts_from_file.py b/scripts/prompts_from_file.py
index e8386ed2..5c84c3e9 100644
--- a/scripts/prompts_from_file.py
+++ b/scripts/prompts_from_file.py
@@ -13,6 +13,7 @@ from modules import sd_samplers
 from modules.processing import Processed, process_images

 from PIL import Image

 from modules.shared import opts, cmd_opts, state

+import re

 

 

 def process_string_tag(tag):

@@ -111,12 +112,17 @@ class Script(scripts.Script):
     def title(self):

         return "Prompts from file or textbox"

 

-    def ui(self, is_img2img):

-        checkbox_iterate = gr.Checkbox(label="Iterate seed every line", value=False)

-        checkbox_iterate_batch = gr.Checkbox(label="Use same random seed for all lines", value=False)

+    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

 

-        prompt_txt = gr.Textbox(label="List of prompt inputs", lines=1)

-        file = gr.File(label="Upload prompt inputs", type='bytes')

+    def ui(self, is_img2img):       

+        checkbox_iterate = gr.Checkbox(label="Iterate seed every line", value=False, elem_id=self.elem_id("checkbox_iterate"))

+        checkbox_iterate_batch = gr.Checkbox(label="Use same random seed for all lines", value=False, elem_id=self.elem_id("checkbox_iterate_batch"))

+

+        prompt_txt = gr.Textbox(label="List of prompt inputs", lines=1, elem_id=self.elem_id("prompt_txt"))

+        file = gr.File(label="Upload prompt inputs", type='bytes', elem_id=self.elem_id("file"))

 

         file.change(fn=load_prompt_file, inputs=[file], outputs=[file, prompt_txt, prompt_txt])