From de2b8fbd88cf245a1ba7dcc62455e2136d51f0c9 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Thu, 2 Jan 2025 03:52:18 +0900
Subject: [PATCH] improved: add aboutBadge

---
 glob/manager_core.py   |  2 +-
 glob/manager_server.py |  5 +++++
 js/comfyui-manager.js  | 14 ++++++++++++++
 pyproject.toml         |  2 +-
 4 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/glob/manager_core.py b/glob/manager_core.py
index 9f976437..f62b217e 100644
--- a/glob/manager_core.py
+++ b/glob/manager_core.py
@@ -36,7 +36,7 @@ import manager_downloader
 from node_package import InstalledNodePackage
 
 
-version_code = [3, 2, 1]
+version_code = [3, 3]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 
diff --git a/glob/manager_server.py b/glob/manager_server.py
index 6118169b..f70246b8 100644
--- a/glob/manager_server.py
+++ b/glob/manager_server.py
@@ -1359,6 +1359,11 @@ async def load_components(request):
         return web.Response(status=400)
 
 
+@routes.get("/manager/version")
+async def get_version(request):
+    return web.Response(text=core.version_str, status=200)
+
+
 async def _confirm_try_install(sender, custom_node_url, msg):
     json_obj = await core.get_data_by_mode('default', 'custom-node-list.json')
 
diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js
index 182573d5..b5a6aa15 100644
--- a/js/comfyui-manager.js
+++ b/js/comfyui-manager.js
@@ -1507,9 +1507,23 @@ class ManagerMenuDialog extends ComfyDialog {
 	}
 }
 
+async function getVersion() {
+	let version = await api.fetchApi(`/manager/version`);
+	return await version.text();
+}
+
 
 app.registerExtension({
 	name: "Comfy.ManagerMenu",
+
+	aboutPageBadges: [
+		{
+			label: `ComfyUI-Manager ${await getVersion()}`,
+			url: 'https://github.com/ltdrdata/ComfyUI-Manager',
+			icon: 'pi pi-th-large'
+		}
+	],
+
 	init() {
 		$el("style", {
 			textContent: style,
diff --git a/pyproject.toml b/pyproject.toml
index 11b6f7bc..ca36a07e 100644
--- a/pyproject.toml
+++ b/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.2.1"
+version = "3.3"
 license = { file = "LICENSE.txt" }
 dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions"]