diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-11-27 05:00:17 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-11-27 05:18:17 +0000 |
| commit | 8a6e4bda21dddef3ab2e70a05d71b587b6c8b04b (patch) | |
| tree | b5a99f78067e53218b2ee4fbb0292d805b01c55a | |
| parent | f0f100e67b78f686dc73cf3c8cad422e45cc9b8a (diff) | |
| download | stable-diffusion-webui-gfx803-8a6e4bda21dddef3ab2e70a05d71b587b6c8b04b.tar.gz stable-diffusion-webui-gfx803-8a6e4bda21dddef3ab2e70a05d71b587b6c8b04b.tar.bz2 stable-diffusion-webui-gfx803-8a6e4bda21dddef3ab2e70a05d71b587b6c8b04b.zip | |
catch uncaught exception with ui creation scripts
prevent total webui crash
| -rw-r--r-- | modules/scripts.py | 54 |
1 files changed, 29 insertions, 25 deletions
diff --git a/modules/scripts.py b/modules/scripts.py index b0689a23..961d032c 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -570,40 +570,44 @@ class ScriptRunner: if controls is None:
return
- script.name = wrap_call(script.title, script.filename, "title", default=script.filename).lower()
- api_args = []
+ try:
+ script.name = wrap_call(script.title, script.filename, "title", default=script.filename).lower()
+ api_args = []
- for control in controls:
- control.custom_script_source = os.path.basename(script.filename)
+ for control in controls:
+ control.custom_script_source = os.path.basename(script.filename)
- arg_info = api_models.ScriptArg(label=control.label or "")
+ arg_info = api_models.ScriptArg(label=control.label or "")
- for field in ("value", "minimum", "maximum", "step"):
- v = getattr(control, field, None)
- if v is not None:
- setattr(arg_info, field, v)
+ for field in ("value", "minimum", "maximum", "step"):
+ v = getattr(control, field, None)
+ if v is not None:
+ setattr(arg_info, field, v)
- choices = getattr(control, 'choices', None) # as of gradio 3.41, some items in choices are strings, and some are tuples where the first elem is the string
- if choices is not None:
- arg_info.choices = [x[0] if isinstance(x, tuple) else x for x in choices]
+ choices = getattr(control, 'choices', None) # as of gradio 3.41, some items in choices are strings, and some are tuples where the first elem is the string
+ if choices is not None:
+ arg_info.choices = [x[0] if isinstance(x, tuple) else x for x in choices]
- api_args.append(arg_info)
+ api_args.append(arg_info)
- script.api_info = api_models.ScriptInfo(
- name=script.name,
- is_img2img=script.is_img2img,
- is_alwayson=script.alwayson,
- args=api_args,
- )
+ script.api_info = api_models.ScriptInfo(
+ name=script.name,
+ is_img2img=script.is_img2img,
+ is_alwayson=script.alwayson,
+ args=api_args,
+ )
- if script.infotext_fields is not None:
- self.infotext_fields += script.infotext_fields
+ if script.infotext_fields is not None:
+ self.infotext_fields += script.infotext_fields
- if script.paste_field_names is not None:
- self.paste_field_names += script.paste_field_names
+ if script.paste_field_names is not None:
+ self.paste_field_names += script.paste_field_names
- self.inputs += controls
- script.args_to = len(self.inputs)
+ self.inputs += controls
+ script.args_to = len(self.inputs)
+
+ except Exception:
+ errors.report(f"Error creating UI for {script.name}: ", exc_info=True)
def setup_ui_for_section(self, section, scriptlist=None):
if scriptlist is None:
|