diff --git a/__init__.py b/__init__.py index b27de0ec..c09018fd 100644 --- a/__init__.py +++ b/__init__.py @@ -17,7 +17,7 @@ import http.client import re import signal -version = "V1.6.2" +version = "V1.6.3" print(f"### Loading: ComfyUI-Manager ({version})") diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js index 700d63f2..1e167aa4 100644 --- a/js/comfyui-manager.js +++ b/js/comfyui-manager.js @@ -574,7 +574,7 @@ class ManagerMenuDialog extends ComfyDialog { var url = prompt("Please enter the URL of the Git repository to install", ""); if (url !== null) { - install_via_git_url(url); + install_via_git_url(url, self); } } }), diff --git a/js/common.js b/js/common.js index 024008f0..1cbd837e 100644 --- a/js/common.js +++ b/js/common.js @@ -80,7 +80,7 @@ function isValidURL(url) { return pattern.test(url); } -export async function install_via_git_url(url) { +export async function install_via_git_url(url, manager_dialog) { if(!url) { return; } @@ -97,7 +97,16 @@ export async function install_via_git_url(url) { const res = await api.fetchApi(`/customnode/install/git_url?url=${url}`); if(res.status == 200) { - app.ui.dialog.show(`'${url}' is installed
To apply the installed/disabled/enabled custom node, please restart ComfyUI.`); + app.ui.dialog.show(`'${url}' is installed
To apply the installed custom node, please ComfyUI.`); + + const rebootButton = document.getElementById('cm-reboot-button'); + const self = this; + rebootButton.onclick = function() { + if(rebootAPI()) { + manager_dialog.close(); + } + }; + app.ui.dialog.element.style.zIndex = 10010; } else {