File size: 1,306 Bytes
6c411e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import subprocess
import threading
import time
import socket
import urllib.request
import os
def install_cloudflared():
if not os.path.exists("cloudflared-linux-amd64.deb"):
subprocess.run(["wget", "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64.deb"])
subprocess.run(["dpkg", "-i", "cloudflared-linux-amd64.deb"])
def iframe_thread(port):
while True:
time.sleep(0.5)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result = sock.connect_ex(('127.0.0.1', port))
if result == 0:
break
sock.close()
print("\nComfyUI finished loading, trying to launch cloudflared (if it gets stuck here cloudflared is having issues)\n")
p = subprocess.Popen(["cloudflared", "tunnel", "--url", "http://127.0.0.1:{}".format(port)], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
for line in p.stderr:
l = line.decode()
if "trycloudflare.com " in l:
print("This is the URL to access ComfyUI:", l[l.find("http"):], end='')
def main():
install_cloudflared()
threading.Thread(target=iframe_thread, daemon=True, args=(8188,)).start()
subprocess.run(["python", "main.py", "--port", "8188", "--auto-launch"])
if __name__ == "__main__":
main()
|