From 521aca76eef3f103cdda329207cd60dfc374c56f Mon Sep 17 00:00:00 2001
From: Michael <32886427+turtlebasket@users.noreply.github.com>
Date: Mon, 13 Oct 2025 18:12:04 -0700
Subject: [PATCH] feat: `COMFYUI_PATH` setting for externally-managed ComfyUI
 (#79)

* configurable COMFYUI_PATH

* Revert "configurable COMFYUI_PATH"

This reverts commit 43628ed3ceee9dd39f249bcde283b2b65994be49.

* move `COMFYUI_PATH` override to bento constructor
---
 src/comfy_pack/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/comfy_pack/service.py b/src/comfy_pack/service.py
index 34d75f5..b1a3e30 100644
--- a/src/comfy_pack/service.py
+++ b/src/comfy_pack/service.py
@@ -80,8 +80,9 @@ class ComfyService:
         logger = logging.getLogger("comfy_pack")
         logger.setLevel(logging.INFO)
         if not EXISTING_COMFYUI_SERVER:
+            workspace_override = os.environ.get("COMFYUI_PATH", None)
             self.server = comfy_pack.run.ComfyUIServer(
-                str(_get_workspace()),
+                workspace_override if workspace_override else str(_get_workspace()),
                 str(INPUT_DIR),
                 verbose=int("BENTOML_DEBUG" in os.environ),
             )