StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename routes

Browse Source
This commit is contained in:
w-e-w 2023-06-14 18:51:47 +09:00
parent d80962681a
commit 5be6c026f5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/api/api.py
View File

@ -210,9 +210,9 @@ class Api:
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/script-info", self.get_script_info, methods=["GET"], response_model=List[models.ScriptInfo])
if shared.cmd_opts.add_stop_route: if shared.cmd_opts.add_stop_route:
self.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"]) self.add_api_route("/sdapi/v1/server-kill", self.kill_webui, methods=["POST"])
self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"]) self.add_api_route("/sdapi/v1/server-restart", self.restart_webui, methods=["POST"])
self.add_api_route("/_stop", self.stop_route, methods=["POST"]) self.add_api_route("/sdapi/v1/server-terminate", self.terminate_webui, methods=["POST"])
self.default_script_arg_txt2img = [] self.default_script_arg_txt2img = []
self.default_script_arg_img2img = [] self.default_script_arg_img2img = []
@ -721,13 +721,13 @@ class Api:
self.app.include_router(self.router) self.app.include_router(self.router)
uvicorn.run(self.app, host=server_name, port=port, timeout_keep_alive=0) uvicorn.run(self.app, host=server_name, port=port, timeout_keep_alive=0)
def quit_webui(self): def kill_webui(self):
restart.stop_program() restart.stop_program()
def restart_webui(self): def restart_webui(self):
if restart.is_restartable(): if restart.is_restartable():
restart.restart_program() restart.restart_program()
def stop_route(request): def terminate_webui(request):
shared.state.server_command = "stop" shared.state.server_command = "stop"
return Response("Stopping.") return Response("Stopping.")