From 05d5a75cdcb749286c9ce9e034bb37a2f6195c37 Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
Date: Fri, 18 Apr 2025 02:25:33 +0800
Subject: [PATCH] Update frontend to 1.16 (Install templates as pip package)
 (#7623)

* install templates as pip package

* Update requirements.txt

* bump templates version to include hidream

---------

Co-authored-by: Chenlei Hu <hcl@comfy.org>
---
 app/frontend_management.py | 21 +++++++++++++++++++++
 requirements.txt           |  3 ++-
 server.py                  |  6 ++++++
 3 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/app/frontend_management.py b/app/frontend_management.py
index c56ea86e0..7b7923b79 100644
--- a/app/frontend_management.py
+++ b/app/frontend_management.py
@@ -184,6 +184,27 @@ comfyui-frontend-package is not installed.
             )
             sys.exit(-1)
 
+    @classmethod
+    def templates_path(cls) -> str:
+        try:
+            import comfyui_workflow_templates
+
+            return str(
+                importlib.resources.files(comfyui_workflow_templates) / "templates"
+            )
+        except ImportError:
+            logging.error(
+                f"""
+********** ERROR ***********
+
+comfyui-workflow-templates is not installed.
+
+{frontend_install_warning_message()}
+
+********** ERROR ***********
+""".strip()
+            )
+
     @classmethod
     def parse_version_string(cls, value: str) -> tuple[str, str, str]:
         """
diff --git a/requirements.txt b/requirements.txt
index 851db23bd..278e3eaa8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
-comfyui-frontend-package==1.15.13
+comfyui-frontend-package==1.16.8
+comfyui-workflow-templates==0.1.1
 torch
 torchsde
 torchvision
diff --git a/server.py b/server.py
index 62667ce18..0cc97b248 100644
--- a/server.py
+++ b/server.py
@@ -736,6 +736,12 @@ class PromptServer():
         for name, dir in nodes.EXTENSION_WEB_DIRS.items():
             self.app.add_routes([web.static('/extensions/' + name, dir)])
 
+        workflow_templates_path = FrontendManager.templates_path()
+        if workflow_templates_path:
+            self.app.add_routes([
+                web.static('/templates', workflow_templates_path)
+            ])
+
         self.app.add_routes([
             web.static('/', self.web_root),
         ])