From f185baeb28f348e4ec97cd7070ed219b5f74a48e Mon Sep 17 00:00:00 2001
From: me <25877290+Kryptortio@users.noreply.github.com>
Date: Thu, 5 Jan 2023 09:29:07 +0100
Subject: Refactor elem_prefix as function elem_id

---
 scripts/custom_code.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'scripts/custom_code.py')

diff --git a/scripts/custom_code.py b/scripts/custom_code.py
index b3bbee03..9ce1f650 100644
--- a/scripts/custom_code.py
+++ b/scripts/custom_code.py
@@ -3,20 +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):
-        elem_prefix = ('img2img' if is_img2img else 'txt2txt') + '_script_custom_code_'
-        
-        code = gr.Textbox(label="Python code", lines=1, elem_id=elem_prefix + "code")
+        code = gr.Textbox(label="Python code", lines=1, elem_id=self.elem_id("code"))
 
         return [code]
 
-- 
cgit v1.2.3