fixed: close dialogs before restart

fixed: visual bug

glob/manager_core.py

@@ -42,7 +42,7 @@ import manager_downloader
 from node_package import InstalledNodePackage
 
 
-version_code = [3, 17]
+version_code = [3, 17, 2]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 
@@ -2432,7 +2432,7 @@ def check_state_of_git_node_pack_single(item, do_fetch=False, do_update_check=Tr
 
 def get_installed_pip_packages():
     # extract pip package infos
-    cmd = manager_util.make_pip_cmd(['pip', 'freeze'])
+    cmd = manager_util.make_pip_cmd(['freeze'])
     pips = subprocess.check_output(cmd, text=True).split('\n')
 
     res = {}

js/custom-nodes-manager.js

@@ -400,6 +400,7 @@ export class CustomNodesManager {
 		this.init();
 
         api.addEventListener("cm-queue-status", this.onQueueStatus);
+        api.addEventListener('reconnected', this.onReconnected);
 	}
 
 	init() {
@@ -762,10 +763,9 @@ export class CustomNodesManager {
 
 			".cn-manager-restart": {
 				click: () => {
-					if(rebootAPI()) {
+					this.close();
-						this.close();
+					this.manager_dialog.close();
-						this.manager_dialog.close();
+					rebootAPI();
-					}
 				}
 			},
 
@@ -1385,19 +1385,14 @@ export class CustomNodesManager {
 
 		this.install_context = {btn: btn, targets: target_items};
 
-		for(let k in target_items) {
-			let item = this.install_context.targets[k];
-			this.grid.updateCell(item, "action");
-		}
-
 		if(errorMsg) {
 			this.showError(errorMsg);
 			show_message("Installation Error:\n"+errorMsg);
 
 			// reset
-			for (const hash of list) {
+			for(let k in target_items) {
-				const item = this.grid.getRowItemBy("hash", hash);
+				let item = this.install_context.targets[k];
-				self.grid.updateCell(item, "action");
+				this.grid.updateCell(item, "action");
 			}
 		}
 		else {
@@ -1406,6 +1401,21 @@ export class CustomNodesManager {
 		}
 	}
 
+	async onReconnected(event) {
+		let self = CustomNodesManager.instance;
+
+		if(self.need_restart) {
+			self.need_restart = false;
+
+			const confirmed = await customConfirm("To apply the changes to the node pack's installation status, you need to refresh the browser. Would you like to refresh?");
+			if (!confirmed) {
+				return;
+			}
+
+			window.location.reload(true);
+		}
+	}
+
 	async onQueueStatus(event) {
 		let self = CustomNodesManager.instance;
 		if(event.detail.status == 'in_progress' && event.detail.ui_target == 'nodepack_manager') {
@@ -1879,6 +1889,7 @@ export class CustomNodesManager {
 
 	showRestart() {
 		this.element.querySelector(".cn-manager-restart").style.display = "block";
+		this.need_restart = true;
 	}
 
 	showStop() {

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-manager"
 description = "ComfyUI-Manager provides features to install and manage custom nodes for ComfyUI, as well as various functionalities to assist with ComfyUI."
-version = "3.17"
+version = "3.17.2"
 license = { file = "LICENSE.txt" }
 dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions"]