diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-11 13:43:34 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-11 13:43:34 +0300 |
| commit | 8ec0442dcd5e51059c2f7ae1572621135f75d858 (patch) | |
| tree | 324581ebde5321d995c184cf152539209d0e725e /modules/api/api.py | |
| parent | 3531a50080e63197752dd4d9b49f0ac34a758e12 (diff) | |
| parent | f85a192f998dcc17d0a59d8755c76100e8e31bde (diff) | |
Merge pull request #8322 from yeataro/master
add: /sdapi/v1/scripts in API
Diffstat (limited to 'modules/api/api.py')
| -rw-r--r-- | modules/api/api.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index fbd50552..376f7f04 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -150,6 +150,7 @@ class Api: self.add_api_route("/sdapi/v1/train/embedding", self.train_embedding, methods=["POST"], response_model=TrainResponse) self.add_api_route("/sdapi/v1/train/hypernetwork", self.train_hypernetwork, methods=["POST"], response_model=TrainResponse) self.add_api_route("/sdapi/v1/memory", self.get_memory, methods=["GET"], response_model=MemoryResponse) + self.add_api_route("/sdapi/v1/scripts", self.get_scripts_list, methods=["GET"], response_model=ScriptsList) def add_api_route(self, path: str, endpoint, **kwargs): if shared.cmd_opts.api_auth: @@ -174,6 +175,12 @@ class Api: script_idx = script_name_to_index(script_name, script_runner.selectable_scripts) script = script_runner.selectable_scripts[script_idx] 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] + + return ScriptsList(txt2img = t2ilist, img2img = i2ilist) def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI): script, script_idx = self.get_script(txt2imgreq.script_name, scripts.scripts_txt2img) |