Upload webui_comfy.py

webui_comfy.py

@@ -0,0 +1,35 @@
+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()