fix the bug in script-info API

This commit is contained in:
AUTOMATIC1111 2023-09-09 08:38:42 +03:00
parent 741e8ecb7d
commit 259768f27f
1 changed files with 5 additions and 1 deletions

View File

@ -491,11 +491,15 @@ class ScriptRunner:
arg_info = api_models.ScriptArg(label=control.label or "")
for field in ("value", "minimum", "maximum", "step", "choices"):
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:
setattr(arg_info, 'choices', [x[0] if isinstance(x, tuple) else x for x in choices])
api_args.append(arg_info)
script.api_info = api_models.ScriptInfo(