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

put all code for /docs in same place and make it work properly with UI reloads

Browse Source
This commit is contained in:
AUTOMATIC 2023-05-08 07:57:17 +03:00
parent 064eda930c
commit 160780283a
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
webui.py
View File

@ -280,12 +280,6 @@ def api_only():
print(f"Startup time: {startup_timer.summary()}.") print(f"Startup time: {startup_timer.summary()}.")
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) 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)
# patch in url for api docs
def my_setup(self):
self.docs_url = "/docs"
self.redoc_url = "/redoc"
self.orig_setup()
def webui(): def webui():
launch_api = cmd_opts.api launch_api = cmd_opts.api
initialize() initialize()
@ -312,9 +306,16 @@ def webui():
for line in file.readlines(): for line in file.readlines():
gradio_auth_creds += [x.strip() for x in line.split(',') if x.strip()] gradio_auth_creds += [x.strip() for x in line.split(',') if x.strip()]
if launch_api: # this restores the missing /docs endpoint
FastAPI.orig_setup = FastAPI.setup if launch_api and not hasattr(FastAPI, 'original_setup'):
setattr(FastAPI, "setup", my_setup) def fastapi_setup(self):
self.docs_url = "/docs"
self.redoc_url = "/redoc"
self.original_setup()
FastAPI.original_setup = FastAPI.setup
FastAPI.setup = fastapi_setup
app, local_url, share_url = shared.demo.launch( app, local_url, share_url = shared.demo.launch(
share=cmd_opts.share, share=cmd_opts.share,
server_name=server_name, server_name=server_name,