mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-09 07:10:16 +08:00
27 lines
1.0 KiB
Python
27 lines
1.0 KiB
Python
from pyngrok import ngrok, conf, exception
|
|
|
|
def connect(token, port, region):
|
|
account = None
|
|
if token is None:
|
|
token = 'None'
|
|
else:
|
|
if ':' in token:
|
|
# token = authtoken:username:password
|
|
account = token.split(':')[1] + ':' + token.split(':')[-1]
|
|
token = token.split(':')[0]
|
|
|
|
config = conf.PyngrokConfig(
|
|
auth_token=token, region=region
|
|
)
|
|
try:
|
|
if account is None:
|
|
public_url = ngrok.connect(port, pyngrok_config=config, bind_tls=True).public_url
|
|
else:
|
|
public_url = ngrok.connect(port, pyngrok_config=config, bind_tls=True, auth=account).public_url
|
|
except exception.PyngrokNgrokError:
|
|
print(f'Invalid ngrok authtoken, ngrok connection aborted.\n'
|
|
f'Your token: {token}, get the right one on https://dashboard.ngrok.com/get-started/your-authtoken')
|
|
else:
|
|
print(f'ngrok connected to localhost:{port}! URL: {public_url}\n'
|
|
'You can use this link after the launch is complete.')
|