aboutsummaryrefslogtreecommitdiff
path: root/modules/api/api.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-17 22:43:24 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-05-17 22:43:24 +0300
commitf6fc7916c4615c4f5cac97cab5add9b0536f6efa (patch)
tree3cf665dfdf38200a0ec80e78165ced7005de3935 /modules/api/api.py
parent8fe9ea7f4d8fd76038db61e1fd2b1cc5927ce108 (diff)
add /sdapi/v1/script-info api
Diffstat (limited to 'modules/api/api.py')
-rw-r--r--modules/api/api.py13
1 files changed, 11 insertions, 2 deletions
diff --git a/modules/api/api.py b/modules/api/api.py
index 165985c3..eee99bbb 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -204,6 +204,7 @@ class Api:
self.add_api_route("/sdapi/v1/unload-checkpoint", self.unloadapi, methods=["POST"])
self.add_api_route("/sdapi/v1/reload-checkpoint", self.reloadapi, methods=["POST"])
self.add_api_route("/sdapi/v1/scripts", self.get_scripts_list, methods=["GET"], response_model=models.ScriptsList)
+ self.add_api_route("/sdapi/v1/script-info", self.get_script_info, methods=["GET"], response_model=List[models.ScriptInfo])
self.default_script_arg_txt2img = []
self.default_script_arg_img2img = []
@@ -229,11 +230,19 @@ class Api:
return script, script_idx
def get_scripts_list(self):
- t2ilist = [str(title.lower()) for title in scripts.scripts_txt2img.titles]
- i2ilist = [str(title.lower()) for title in scripts.scripts_img2img.titles]
+ t2ilist = [script.name for script in scripts.scripts_txt2img.scripts if script.name is not None]
+ i2ilist = [script.name for script in scripts.scripts_img2img.scripts if script.name is not None]
return models.ScriptsList(txt2img=t2ilist, img2img=i2ilist)
+ def get_script_info(self):
+ res = []
+
+ for script_list in [scripts.scripts_txt2img.scripts, scripts.scripts_img2img.scripts]:
+ res += [script.api_info for script in script_list if script.api_info is not None]
+
+ return res
+
def get_script(self, script_name, script_runner):
if script_name is None or script_name == "":
return None, None