Added the ability to disable port (#210)

* Added the ability to disable port from being added so that users that have server redirects via nginx can use it * restrict canonicalize_url to webui process --------- Co-authored-by: ljleb <[email protected]>
ModelSurge · Feb 1, 2024 · 10e95a4 · 10e95a4
1 parent ac6ec6a
commit 10e95a4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib_comfyui/webui/settings.py b/lib_comfyui/webui/settings.py
@@ -41,6 +41,8 @@ def create_section():
     shared.opts.onchange("comfyui_reverse_proxy_enabled", update_reverse_proxy_enabled)
     update_reverse_proxy_enabled()
 
+    shared.opts.add_option("comfyui_reverse_proxy_disable_port", shared.OptionInfo(False,'Disable using the port when using reverse_proxy. Useful if using a nginx server',section=section))
+
 
 @ipc.restrict_to_process('webui')
 def update_enabled():
@@ -159,9 +161,10 @@ def get_comfyui_client_url():
     return client_url
 
 
+@ipc.restrict_to_process('webui')
 def canonicalize_url(input_url: str, default_port: int = 8189) -> str:
     from urllib.parse import urlparse, urlunparse
-
+    from modules import shared
     # Step 1: Prepend 'http://' if scheme is missing
     if not input_url.startswith(('http://', 'https://')):
         input_url = 'http://' + input_url
@@ -173,8 +176,9 @@ def canonicalize_url(input_url: str, default_port: int = 8189) -> str:
     scheme = parsed.scheme if parsed.scheme else 'http'
 
     # Step 4: Add the missing port
+    disable_port = shared.opts.data.get('comfyui_reverse_proxy_disable_port',False)
     netloc = parsed.netloc
-    if ':' not in netloc:  # Check if port is missing
+    if ':' not in netloc and not disable_port:  # Check if port is missing
         netloc += f":{default_port}"
     elif netloc.count(':') == 1 and parsed.scheme:  # If port exists but scheme was present in input
         host, port = netloc.split(':')