add a1111 compatibility items

This commit is contained in:
Vladimir Mandic
2023-10-20 08:19:37 -04:00
parent 52078069d2
commit 3794af95c5
7 changed files with 201 additions and 4 deletions
+17 -1
View File
@@ -35,6 +35,7 @@ class Script:
group = None
infotext_fields = None
paste_field_names = None
section = None
def title(self):
"""this function should return the title of the script. This is what will be displayed in the dropdown menu."""
@@ -332,7 +333,6 @@ class ScriptRunner:
self.paste_field_names.clear()
self.script_load_ctr = 0
self.is_img2img = is_img2img
self.scripts.clear()
self.alwayson_scripts.clear()
self.selectable_scripts.clear()
@@ -355,6 +355,22 @@ class ScriptRunner:
except Exception as e:
log.error(f'Script initialize: {path} {e}')
def setup_ui_for_section(self, section, scriptlist=None):
if scriptlist is None:
scriptlist = self.alwayson_scripts
for script in scriptlist:
if script.alwayson and script.section != section:
continue
if script.create_group:
with gr.Group(visible=script.alwayson) as group:
self.create_script_ui(script)
script.group = group
else:
self.create_script_ui(script)
def prepare_ui(self):
self.inputs = [None]
def setup_ui(self):
import modules.api.models as api_models
self.titles = [wrap_call(script.title, script.filename, "title") or f"{script.filename} [error]" for script in self.selectable_scripts]