diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 00000000..1d9d7b3f --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,11 @@ +[project] +name = "ComfyUI" +version = "0.3.10" +readme = "README.md" +license = { file = "LICENSE" } +requires-python = ">=3.9" + +[project.urls] +homepage = "https://www.comfy.org/" +repository = "https://github.com/comfyanonymous/ComfyUI" +documentation = "https://docs.comfy.org/" diff --git a/requirements.txt b/requirements.txt index 4c2c0b2b..d8745105 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,6 +13,7 @@ Pillow scipy tqdm psutil +toml #non essential dependencies: kornia>=0.7.1 diff --git a/server.py b/server.py index ceb5e83b..ccbea28d 100644 --- a/server.py +++ b/server.py @@ -20,6 +20,7 @@ from io import BytesIO import aiohttp from aiohttp import web +import toml import logging import mimetypes @@ -45,19 +46,13 @@ async def send_socket_catch_exception(function, message): logging.warning("send error: {}".format(err)) def get_comfyui_version(): - comfyui_version = "unknown" - repo_path = os.path.dirname(os.path.realpath(__file__)) - try: - import pygit2 - repo = pygit2.Repository(repo_path) - comfyui_version = repo.describe(describe_strategy=pygit2.GIT_DESCRIBE_TAGS) - except Exception: - try: - import subprocess - comfyui_version = subprocess.check_output(["git", "describe", "--tags"], cwd=repo_path).decode('utf-8') - except Exception as e: - logging.warning(f"Failed to get ComfyUI version: {e}") - return comfyui_version.strip() + """ Get the version of ComfyUI from the pyproject.toml file. + + Note: + Use Python's built-in `tomllib` from Python 3.11 or later when available. + """ + with open("pyproject.toml", "r", encoding="utf-8") as f: + return toml.load(f)["project"]["version"] @web.middleware async def cache_control(request: web.Request, handler):