OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-04 03:29:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add static domain arg for ngrok

Browse Source
This commit is contained in:
drakce-dev 2025-03-11 16:24:07 +01:00 committed by GitHub
parent 82a973c043
commit 082ca98e16
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/cmd_args.py
View File

@ -48,6 +48,7 @@ parser.add_argument("--share", action='store_true', help="use share=True for gra
parser.add_argument("--ngrok", type=str, help="ngrok authtoken, alternative to gradio --share", default=None)
parser.add_argument("--ngrok-region", type=str, help="does not do anything.", default="")
parser.add_argument("--ngrok-options", type=json.loads, help='The options to pass to ngrok in JSON format, e.g.: \'{"authtoken_from_env":true, "basic_auth":"user:password", "oauth_provider":"google", "oauth_allow_emails":"user@asdf.com"}\'', default=dict())
parser.add_argument("--ngrok-static-domain", type=str, help="Add a static domain for the ngrok tunnel", default=None)
parser.add_argument("--enable-insecure-extension-access", action='store_true', help="enable extensions tab regardless of other options")
parser.add_argument("--codeformer-models-path", type=normalized_filepath, help="Path to directory with codeformer model file(s).", default=os.path.join(models_path, 'Codeformer'))
parser.add_argument("--gfpgan-models-path", type=normalized_filepath, help="Path to directory with GFPGAN model file(s).", default=os.path.join(models_path, 'GFPGAN'))

5
modules/ngrok.py
View File

@ -1,7 +1,7 @@
import ngrok
# Connect to ngrok for ingress
def connect(token, port, options):
def connect(token, port, options, domain):
account = None
if token is None:
token = 'None'
@ -18,7 +18,8 @@ def connect(token, port, options):
options['basic_auth'] = account
if not options.get('session_metadata'):
options['session_metadata'] = 'stable-diffusion-webui'
if domain:
options['domain'] = domain
try:
public_url = ngrok.connect(f"127.0.0.1:{port}", **options).url()

3
modules/ui.py
View File

@ -55,7 +55,8 @@ if cmd_opts.ngrok is not None:
ngrok.connect(
cmd_opts.ngrok,
cmd_opts.port if cmd_opts.port is not None else 7860,
cmd_opts.ngrok_options
cmd_opts.ngrok_options,
cmd_opts.ngrok_static_domain
)