Commit b9664ab6 authored by w-e-w's avatar w-e-w
Browse files

move _stop route to api

parent 9142be0a
Loading
Loading
Loading
Loading
+9 −2
Original line number Diff line number Diff line
@@ -208,8 +208,11 @@ 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])

        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/restart-webui", self.restart_webui, methods=["POST"])
            self.add_api_route("/_stop", self.stop_route, methods=["POST"])

        self.default_script_arg_txt2img = []
        self.default_script_arg_img2img = []
@@ -724,3 +727,7 @@ class Api:
    def restart_webui(self):
        if restart.is_restartable():
            restart.restart_program()

    def stop_route(request):
        shared.state.server_command = "stop"
        return Response("Stopping.")
+0 −7
Original line number Diff line number Diff line
@@ -362,11 +362,6 @@ def api_only():
    api.launch(server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1", port=cmd_opts.port if cmd_opts.port else 7861)


def stop_route(request):
    shared.state.server_command = "stop"
    return Response("Stopping.")


def webui():
    launch_api = cmd_opts.api
    initialize()
@@ -404,8 +399,6 @@ def webui():
                "redoc_url": "/redoc",
            },
        )
        if cmd_opts.add_stop_route:
            app.add_route("/_stop", stop_route, methods=["POST"])

        # after initial launch, disable --autolaunch for subsequent restarts
        cmd_opts.autolaunch = False