Refactor elem_prefix as function elem_id

1 files changed, 9 insertions, 5 deletions

diff --git a/scripts/prompt_matrix.py b/scripts/prompt_matrix.py
index f610c334..278d2e68 100644
--- a/scripts/prompt_matrix.py
+++ b/scripts/prompt_matrix.py
@@ -10,6 +10,7 @@ from modules import images
 from modules.processing import process_images, Processed

 from modules.shared import opts, cmd_opts, state

 import modules.sd_samplers

+import re

 

 

 def draw_xy_grid(xs, ys, x_label, y_label, cell):

@@ -44,11 +45,14 @@ class Script(scripts.Script):
     def title(self):

         return "Prompt matrix"

 

-    def ui(self, is_img2img):

-        elem_prefix = ('img2img' if is_img2img else 'txt2txt') + '_script_prompt_matrix_'

-        

-        put_at_start = gr.Checkbox(label='Put variable parts at start of prompt', value=False, elem_id=elem_prefix + "put_at_start")

-        different_seeds = gr.Checkbox(label='Use different seed for each picture', value=False, elem_id=elem_prefix + "different_seeds")

+    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 ui(self, is_img2img):        

+        put_at_start = gr.Checkbox(label='Put variable parts at start of prompt', value=False, elem_id=self.elem_id("put_at_start"))

+        different_seeds = gr.Checkbox(label='Use different seed for each picture', value=False, elem_id=self.elem_id("different_seeds"))

 

         return [put_at_start, different_seeds]