Commit 259768f2 authored by AUTOMATIC1111's avatar AUTOMATIC1111

fix the bug in script-info API

parent 741e8ecb
......@@ -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(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment