Commit 259768f2 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

fix the bug in script-info API

parent 741e8ecb
Loading
Loading
Loading
Loading
+5 −1
Original line number Diff line number Diff line
@@ -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(