diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-06-10 14:36:34 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-06-10 14:36:34 +0000 |
| commit | 9142be0a0d8cea37cf1ae86c17fc7dcb161d9a42 (patch) | |
| tree | 061122f8b682531ff9448fe8fa2344d11228cea0 /modules/api | |
| parent | 59419bd64a1581caccaac04dceb66c1c069a2db1 (diff) | |
| download | stable-diffusion-webui-gfx803-9142be0a0d8cea37cf1ae86c17fc7dcb161d9a42.tar.gz stable-diffusion-webui-gfx803-9142be0a0d8cea37cf1ae86c17fc7dcb161d9a42.tar.bz2 stable-diffusion-webui-gfx803-9142be0a0d8cea37cf1ae86c17fc7dcb161d9a42.zip | |
quit restart
Diffstat (limited to 'modules/api')
| -rw-r--r-- | modules/api/api.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 2e49526e..317c809f 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -14,7 +14,7 @@ from fastapi.encoders import jsonable_encoder from secrets import compare_digest import modules.shared as shared -from modules import sd_samplers, deepbooru, sd_hijack, images, scripts, ui, postprocessing, errors +from modules import sd_samplers, deepbooru, sd_hijack, images, scripts, ui, postprocessing, errors, restart from modules.api import models from modules.shared import opts from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img, process_images @@ -208,6 +208,8 @@ class Api: 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.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"]) + self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"]) self.default_script_arg_txt2img = [] self.default_script_arg_img2img = [] @@ -715,3 +717,10 @@ class Api: def launch(self, server_name, port): self.app.include_router(self.router) uvicorn.run(self.app, host=server_name, port=port, timeout_keep_alive=0) + + def quit_webui(self): + restart.stop_program() + + def restart_webui(self): + if restart.is_restartable(): + restart.restart_program() |