dsh-image-gen 0.2.1 → 0.2.2

package/README.en.md

@@ -6,7 +6,7 @@
 
 # 🎨 dsh-image-gen
 
-**Bring ChatGPT-like conversational image generation and image editing to DeepSeek Harness, with support for text-to-image, image-to-image, continuous editing, and multiple image providers.**
+**Complete conversational image capabilities for DeepSeek Harness: text-to-image, image-to-image, multi-image references, continuous editing, local ComfyUI, and gallery management.**
 
 [![npm version](https://img.shields.io/npm/v/dsh-image-gen.svg?style=flat-square&color=blue)](https://www.npmjs.com/package/dsh-image-gen)
 [![DSH Plugin](https://img.shields.io/badge/Plugin%20For-DeepSeek%20Harness-6366f1?style=flat-square)](https://github.com/deepseek-ai)
@@ -27,7 +27,7 @@ Install the image generation plugin by running: pnpm dsh plugin --profile web ad
 
 <br />
 
-<p align="center">After installation, enter your API Key in DSH Settings, then tell the Agent:</p>
+<p align="center">After installation, enter your API key or configure local ComfyUI in DSH Settings, then tell the Agent:</p>
 
 ```text
 Draw a cyberpunk cat on a neon street in a rainy night.
@@ -56,7 +56,7 @@ DeepSeek Harness empowers agents to use tools for various tasks. This project ad
 graph LR
     A[User Prompt] --> B[DeepSeek Harness Agent]
     B --> C[generate_image / edit_image]
-    C --> D[Gemini / OpenAI / Seedream / DashScope]
+    C --> D[Gemini / OpenAI / Seedream / DashScope / ComfyUI]
     D --> E[Image Data]
     E --> F[In-chat Conversation Stream]
 ```
@@ -91,12 +91,12 @@ pnpm dsh plugin --profile web add ./dsh-image-gen
 
 </details>
 
-### 2. Configure API Key & Workspace Settings
+### 2. Configure Provider & Workspace Settings
 
 Open DSH Web (`http://localhost:3080`):
 
 1. Navigate to **Settings → Plugins → Image generation**.
-2. Select Provider, input API Key.
+2. Select a Provider. Enter an API key for cloud providers; for local ComfyUI, enter its URL and import an API Format Workflow JSON using `{{prompt}}` for the prompt and optional `{{seed}}` for the seed.
 3. Optionally enable **Save to workspace** (enabled by default) and specify subfolder, then click **Save**.
 
 <div align="center">
@@ -128,7 +128,7 @@ Click the **`[Gallery]`** Tab in the top navigation bar to browse and search all
 - 💬 **In-Chat Image Generation**: Tell your Agent what you want to draw without switching tabs or tools.
 - 🖼️ **Native Image Gallery**: Dedicated "Gallery" tab automatically collecting all generated images with keyword search, provider filtering, single-item deletion (with confirmation and persistent tombstones), and quick copy/download.
 - 🔍 **Interactive Image Tools**: High-res fullscreen preview, one-click copy to clipboard, local download, and open in new tab.
-- 🎨 **Multi-Provider Support**: Supports Google Gemini, OpenAI Images, OpenAI Compatible API, ByteDance Seedream / Volcengine Ark, and Aliyun DashScope (Wanx / Qwen-Image).
+- 🎨 **Multi-Provider Support**: Supports Google Gemini, OpenAI Images, OpenAI Compatible API, ByteDance Seedream / Volcengine Ark, Aliyun DashScope (Wanx / Qwen-Image), and local ComfyUI workflows.
 - 🔑 **BYOK (Bring Your Own Key)**: Uses your own API keys managed securely by DSH credentials service with write-only protection.
 - 🖼️ **Durable Session Persistence**: Images integrate with DSH Attachment and conversation lifecycle, preserved across reloads.
 - 💾 **Workspace File Output**: By default each generated image is also written as a file into the current session workspace; the tool result carries the absolute file path. Can be disabled or re-pointed in settings.
@@ -145,6 +145,7 @@ Click the **`[Gallery]`** Tab in the top navigation bar to browse and search all
 | **OpenAI Compatible** | Custom | Custom Base URL |
 | **ByteDance Seedream / Volcengine Ark** | `doubao-seedream-5-0-260128` | `https://ark.cn-beijing.volces.com/api/v3` |
 | **Aliyun DashScope / Wanx** | `wanx2.1-t2i-turbo` | `https://dashscope.aliyuncs.com/api/v1` |
+| **Local ComfyUI (text-to-image only)** | Imported API Format Workflow | `http://127.0.0.1:8188` |
 
 ---
 

package/README.md

@@ -6,7 +6,7 @@
 
 # 🎨 dsh-image-gen
 
-**让 DeepSeek Harness 拥有类似 ChatGPT 的原生对话生图与图片编辑能力，支持文生图、图生图、连续编辑，并兼容多家图像模型提供商。**
+**为 DeepSeek Harness 提供完整的对话图像能力：文生图、图生图、多图参考、连续编辑、本地 ComfyUI 与画廊管理。**
 
 [![npm version](https://img.shields.io/npm/v/dsh-image-gen.svg?style=flat-square&color=blue)](https://www.npmjs.com/package/dsh-image-gen)
 [![DSH Plugin](https://img.shields.io/badge/Plugin%20For-DeepSeek%20Harness-6366f1?style=flat-square)](https://github.com/deepseek-ai)
@@ -28,7 +28,7 @@
 
 <br />
 
-<p align="center">安装完成后，在 DSH 设置中填入自己的 API Key，就可以直接对 Agent 说：</p>
+<p align="center">安装完成后，在 DSH 设置中填入自己的 API Key 或配置本地 ComfyUI，就可以直接对 Agent 说：</p>
 
 ```text
 帮我画一张雨夜霓虹街头的赛博朋克猫咪。
@@ -57,7 +57,7 @@ DeepSeek Harness 已经可以让 Agent 调用不同工具完成任务，本项
 graph LR
     A[用户 Prompt] --> B[DeepSeek Harness Agent]
     B --> C[generate_image / edit_image]
-    C --> D[Gemini / OpenAI / Seedream / DashScope]
+    C --> D[Gemini / OpenAI / Seedream / DashScope / ComfyUI]
     D --> E[图片数据]
     E --> F[当前 Conversation 对话流]
 ```
@@ -92,12 +92,12 @@ pnpm dsh plugin --profile web add ./dsh-image-gen
 
 </details>
 
-### 2. 配置 API Key 与工作区设置
+### 2. 配置 Provider 与工作区设置
 
 打开 DSH Web 页面（默认 `http://localhost:3080`）：
 
 1. 进入 **Settings → Plugins → Image generation**。
-2. 选择 Provider，填写 API Key。
+2. 选择 Provider；云端 Provider 填写 API Key，本地 ComfyUI 填写地址并导入 API Format Workflow JSON（提示词位置使用 `{{prompt}}`，种子可选用 `{{seed}}`）。
 3. 可按需开启 **保存到工作区**（默认开启）并自定义子目录，点击 **保存** 即可。
 
 <div align="center">
@@ -137,7 +137,7 @@ Agent 会调用 `edit_image`，复用当前会话中的图片继续修改。
 - 💬 **原生对话生图与编辑**：直接在 DeepSeek Harness 对话中生成图片，也可以基于已有图片继续修改。
 - 🔁 **连续多轮编辑**：支持复用当前会话中的上传图片、历史生成图和上一轮编辑结果继续迭代。
 - 🖼️ **画廊与图片工具**：自动汇总历史图片，支持搜索、筛选、全屏预览、复制、下载与删除。
-- 🎨 **多 Provider 支持**：兼容 Google Gemini、OpenAI Images / Compatible、Seedream、DashScope Qwen Image。
+- 🎨 **多 Provider 支持**：兼容 Google Gemini、OpenAI Images / Compatible、Seedream、DashScope Qwen Image，以及用户本地的 ComfyUI 工作流。
 - 🔑 **BYOK + 原生设置**：API Key、Provider、模型和 Endpoint 都可以直接在 DSH 设置中配置。
 - 💾 **会话与工作区保存**：图片接入 DSH Attachment / Conversation，并可自动保存到当前工作区。
 
@@ -152,6 +152,7 @@ Agent 会调用 `edit_image`，复用当前会话中的图片继续修改。
 | **OpenAI Compatible** | 自定义 | 自定义 Base URL |
 | **ByteDance Seedream / 火山方舟** | `doubao-seedream-5-0-260128` | `https://ark.cn-beijing.volces.com/api/v3` |
 | **Aliyun DashScope / Qwen Image** | `qwen-image-3.0` | `https://dashscope.aliyuncs.com/api/v1` |
+| **Local ComfyUI（仅文生图）** | 用户导入的 API Format Workflow | `http://127.0.0.1:8188` |
 
 ---
 

package/README.zh-CN.md

@@ -6,7 +6,7 @@
 
 # 🎨 dsh-image-gen
 
-**让 DeepSeek Harness 像 ChatGPT 一样在对话中直接生成图片，支持画廊汇总,全屏预览、快捷复制与一键下载。**
+**为 DeepSeek Harness 提供完整的对话图像能力：文生图、图生图、多图参考、连续编辑、本地 ComfyUI 与画廊管理。**
 
 [![npm version](https://img.shields.io/npm/v/dsh-image-gen.svg?style=flat-square&color=blue)](https://www.npmjs.com/package/dsh-image-gen)
 [![DSH Plugin](https://img.shields.io/badge/Plugin%20For-DeepSeek%20Harness-6366f1?style=flat-square)](https://github.com/deepseek-ai)
@@ -27,7 +27,7 @@
 
 <br />
 
-<p align="center">安装完成后，在 DSH 设置中填入自己的 API Key，就可以直接对 Agent 说：</p>
+<p align="center">安装完成后，在 DSH 设置中填入自己的 API Key 或配置本地 ComfyUI，就可以直接对 Agent 说：</p>
 
 ```text
 帮我画一张雨夜霓虹街头的赛博朋克猫咪。
@@ -53,7 +53,7 @@ DeepSeek Harness 已经可以让 Agent 调用不同工具完成任务，本项
 graph LR
     A[用户 Prompt] --> B[DeepSeek Harness Agent]
     B --> C[generate_image 工具]
-    C --> D[Gemini / OpenAI / Seedream / DashScope]
+    C --> D[Gemini / OpenAI / Seedream / DashScope / ComfyUI]
     D --> E[图片数据]
     E --> F[当前 Conversation 对话流]
 ```
@@ -88,12 +88,12 @@ pnpm dsh plugin --profile web add ./dsh-image-gen
 
 </details>
 
-### 2. 配置 API Key 与工作区设置
+### 2. 配置 Provider 与工作区设置
 
 打开 DSH Web 页面（默认 `http://localhost:3080`）：
 
 1. 进入 **Settings → Plugins → Image generation**。
-2. 选择 Provider，填写 API Key。
+2. 选择 Provider；云端 Provider 填写 API Key，本地 ComfyUI 填写地址并导入 API Format Workflow JSON（提示词位置使用 `{{prompt}}`，种子可选用 `{{seed}}`）。
 3. 可按需开启 **保存到工作区**（默认开启）并自定义子目录，点击 **保存** 即可。
 
 <div align="center">
@@ -125,7 +125,7 @@ pnpm dsh plugin --profile web add ./dsh-image-gen
 - 💬 **对话中直接生图**：不需要切换到其他网站，也不需要手动复制 Prompt，直接告诉 Agent 你想画什么即可。
 - 🖼️ **历史生图画廊**：顶栏自带「画廊」Tab，自动汇总所有历史生成的图片，支持关键词搜索、厂商筛选、单张删除（带防误触确认与持久化墓碑）与一键复制/下载。
 - 🔍 **交互式图片工具**：支持点击全屏大图预览、一键复制图片到剪贴板、本地下载与新标签页打开。
-- 🎨 **多 Provider 支持**：支持 Google Gemini、OpenAI Images、OpenAI Compatible API、ByteDance Seedream / 火山方舟以及阿里云 DashScope（通义万相 / Qwen-Image）。Provider、模型和 Endpoint 均可在设置中自由定制。
+- 🎨 **多 Provider 支持**：支持 Google Gemini、OpenAI Images、OpenAI Compatible API、ByteDance Seedream / 火山方舟、阿里云 DashScope（通义万相 / Qwen-Image）以及用户本地的 ComfyUI 工作流。
 - 🔑 **BYOK (自带 Key)**：插件使用你自己的 API Key。API Key 通过 DeepSeek Harness 的 `credentials` 服务管理，采用写保护隔离，不需要写进项目源码或配置文件，前端不存明文。
 - 🖼️ **图片跟随会话保存**：生成结果会接入 DeepSeek Harness 的 Attachment / Conversation 体系，重新打开历史会话后，仍然可以看到之前生成的图片。
 - 💾 **生成图片自动落盘到工作区**：默认在每次生成后把图片文件保存到当前会话工作区，工具结果会返回文件的绝对路径；可在设置中关闭或修改保存目录。
@@ -142,6 +142,7 @@ pnpm dsh plugin --profile web add ./dsh-image-gen
 | **OpenAI Compatible** | 自定义 | 自定义 Base URL |
 | **ByteDance Seedream / 火山方舟** | `doubao-seedream-5-0-260128` | `https://ark.cn-beijing.volces.com/api/v3` |
 | **Aliyun DashScope / 通义万相** | `wanx2.1-t2i-turbo` | `https://dashscope.aliyuncs.com/api/v1` |
+| **Local ComfyUI（仅文生图）** | 用户导入的 API Format Workflow | `http://127.0.0.1:8188` |
 
 ---
 

package/lib/client.js

@@ -17,19 +17,58 @@ window.__ModuleLoader__.load({
 		const DEFAULT_OPENAI_BASE_URL = "https://api.openai.com/v1";
 		const DEFAULT_SEEDREAM_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
 		const DEFAULT_DASHSCOPE_ENDPOINT = "https://dashscope.aliyuncs.com/api/v1";
+		const DEFAULT_COMFYUI_BASE_URL = "http://127.0.0.1:8188";
+		const DEFAULT_COMFYUI_TIMEOUT_MS = 3e5;
 		const DEFAULT_MODELS = {
 			google: "gemini-3.1-flash-image",
 			openai: "gpt-image-2",
 			seedream: "doubao-seedream-5-0-260128",
-			dashscope: "qwen-image-3.0"
+			dashscope: "qwen-image-3.0",
+			comfyui: "API workflow"
 		};
 		const DEFAULT_BASE_URLS = {
 			google: DEFAULT_GOOGLE_ENDPOINT,
 			openai: DEFAULT_OPENAI_BASE_URL,
 			seedream: DEFAULT_SEEDREAM_BASE_URL,
-			dashscope: DEFAULT_DASHSCOPE_ENDPOINT
+			dashscope: DEFAULT_DASHSCOPE_ENDPOINT,
+			comfyui: DEFAULT_COMFYUI_BASE_URL
 		};
 		//#endregion
+		//#region lib/types/comfyui-workflow.js
+		/** Pure ComfyUI API-workflow validation and placeholder injection. */
+		const COMFYUI_PROMPT_PLACEHOLDER = "{{prompt}}";
+		/** Validate imported JSON without exposing workflow graph details to callers. */
+		function validateComfyUIWorkflowJson(workflowJson) {
+			parseWorkflow(workflowJson);
+		}
+		function parseWorkflow(workflowJson) {
+			if (workflowJson.trim().length === 0) throw new Error("Import a ComfyUI API workflow JSON file in Settings before generating");
+			if (new TextEncoder().encode(workflowJson).byteLength > 5242880) throw new Error("ComfyUI workflow file must be no larger than 5 MB");
+			let value;
+			try {
+				value = JSON.parse(workflowJson);
+			} catch {
+				throw new Error("ComfyUI workflow file is not valid JSON");
+			}
+			if (!isRecord(value) || Object.keys(value).length === 0) throw new Error("ComfyUI workflow must be a non-empty API-format JSON object");
+			if (!Object.values(value).some((node) => {
+				const inputs = record(node)?.inputs;
+				return inputs !== void 0 && containsPromptPlaceholder(inputs);
+			})) throw new Error(`ComfyUI workflow must contain ${COMFYUI_PROMPT_PLACEHOLDER} in a text input`);
+			return value;
+		}
+		function containsPromptPlaceholder(value) {
+			if (typeof value === "string") return value.includes(COMFYUI_PROMPT_PLACEHOLDER);
+			if (Array.isArray(value)) return value.some(containsPromptPlaceholder);
+			return isRecord(value) && Object.values(value).some(containsPromptPlaceholder);
+		}
+		function isRecord(value) {
+			return typeof value === "object" && value !== null && !Array.isArray(value);
+		}
+		function record(value) {
+			return isRecord(value) ? value : void 0;
+		}
+		//#endregion
 		//#region lib/types/client/gallery-store.js
 		const DB_NAME = "dsh_image_gen_db";
 		const DB_VERSION = 2;
@@ -906,6 +945,7 @@ window.__ModuleLoader__.load({
 				providerOpenAI: "OpenAI / 中转站",
 				providerSeedream: "字节 Seedream",
 				providerDashScope: "阿里 DashScope (通义万相 / Qwen)",
+				providerComfyUI: "本地 ComfyUI",
 				apiKeyLabel: "{provider} API Key",
 				apiKeyPlaceholder: "留空即可保留已配置的 Key",
 				apiKeyHint: "安全保存为 {key}；页面不会读回明文。",
@@ -916,7 +956,17 @@ window.__ModuleLoader__.load({
 				endpointHintOpenAI: "中转站请填其 OpenAI 兼容的 /v1 地址。",
 				endpointHintSeedream: "火山方舟兼容的 /api/v3 地址。",
 				endpointHintDashScope: "阿里云百炼 DashScope 官方接口地址。",
+				endpointHintComfyUI: "正在运行且 DSH Host 可以访问的 ComfyUI 地址，默认使用本机 8188 端口。",
 				model: "模型",
+				workflow: "API Workflow",
+				workflowImport: "导入 JSON 文件",
+				workflowReplace: "替换 JSON 文件",
+				workflowMissing: "尚未导入工作流",
+				workflowImported: "已导入 {name}",
+				workflowHint: "请从 ComfyUI 导出 API Format JSON，并在提示词输入位置写入 {{prompt}}；可用 {{seed}} 作为随机种子。",
+				workflowTooLarge: "工作流文件不能超过 5 MB。",
+				timeout: "生成超时（秒）",
+				timeoutHint: "包括提交、等待和下载图片；默认 300 秒。",
 				saveToWorkspace: "保存到工作区",
 				saveToWorkspaceHint: "每次生成后，把图片文件保存到当前会话工作区。",
 				folder: "工作区文件夹",
@@ -946,6 +996,7 @@ window.__ModuleLoader__.load({
 				providerOpenAI: "OpenAI / Relay",
 				providerSeedream: "ByteDance Seedream",
 				providerDashScope: "Aliyun DashScope (Wanx / Qwen)",
+				providerComfyUI: "Local ComfyUI",
 				apiKeyLabel: "{provider} API Key",
 				apiKeyPlaceholder: "Leave empty to keep configured key",
 				apiKeyHint: "Securely saved as {key}; never read back in plaintext.",
@@ -956,7 +1007,17 @@ window.__ModuleLoader__.load({
 				endpointHintOpenAI: "OpenAI-compatible /v1 base URL for relays.",
 				endpointHintSeedream: "Volcengine Ark compatible /api/v3 base URL.",
 				endpointHintDashScope: "Official Aliyun DashScope endpoint.",
+				endpointHintComfyUI: "A running ComfyUI server reachable by the DSH Host; the default points to port 8188 on this computer.",
 				model: "Model",
+				workflow: "API Workflow",
+				workflowImport: "Import JSON file",
+				workflowReplace: "Replace JSON file",
+				workflowMissing: "No workflow imported",
+				workflowImported: "Imported {name}",
+				workflowHint: "Export an API Format JSON from ComfyUI and place {{prompt}} in its prompt input. {{seed}} is available for a random seed.",
+				workflowTooLarge: "Workflow files must be no larger than 5 MB.",
+				timeout: "Generation timeout (seconds)",
+				timeoutHint: "Covers submission, waiting, and image download; defaults to 300 seconds.",
 				saveToWorkspace: "Save to workspace",
 				saveToWorkspaceHint: "Write each generated image as a file into the session workspace.",
 				folder: "Workspace folder",
@@ -996,6 +1057,12 @@ window.__ModuleLoader__.load({
 .dsh-ig-input{box-sizing:border-box;width:100%;padding:8px 12px;font-size:13px;border:1px solid var(--dsw-alias-border-l2,#d7dbe0);border-radius:8px;background:var(--dsw-alias-bg-layer-3,transparent);color:inherit;outline:none;transition:border-color .15s}
 .dsh-ig-input:focus{border-color:var(--dsw-alias-brand-primary,#4c78ff)}
 .dsh-ig-input-group{display:flex;gap:8px;align-items:center}
+.dsh-ig-file-row{display:flex;align-items:center;gap:10px;min-width:0}
+.dsh-ig-file-input{position:absolute;width:1px;height:1px;overflow:hidden;clip:rect(0 0 0 0);white-space:nowrap;clip-path:inset(50%)}
+.dsh-ig-file-button{appearance:none;flex:none;border:1px solid var(--dsw-alias-border-l2,#d7dbe0);border-radius:8px;padding:7px 12px;background:var(--dsw-alias-bg-layer-3,#f9fafb);color:var(--dsw-alias-label-secondary,inherit);font-size:13px;cursor:pointer;transition:background .15s,border-color .15s}
+.dsh-ig-file-button:hover{background:var(--dsw-alias-bg-layer-2,#edf0f3);border-color:var(--dsw-alias-label-dimmed,#9ca3af)}
+.dsh-ig-file-button:focus-within{outline:2px solid var(--dsw-alias-brand-primary,#4c78ff);outline-offset:2px}
+.dsh-ig-file-name{min-width:0;overflow:hidden;text-overflow:ellipsis;white-space:nowrap;color:var(--dsw-alias-label-secondary,inherit);font-size:12px}
 .dsh-ig-btn-reset{appearance:none;border:1px solid var(--dsw-alias-border-l2,#d7dbe0);border-radius:8px;padding:7px 12px;background:var(--dsw-alias-bg-layer-3,#f9fafb);color:var(--dsw-alias-label-secondary,inherit);font:inherit;font-size:13px;cursor:pointer;white-space:nowrap;transition:background .15s,border-color .15s}
 .dsh-ig-btn-reset:hover{background:var(--dsw-alias-bg-layer-2,#edf0f3);border-color:var(--dsw-alias-label-dimmed,#9ca3af)}
 .dsh-ig-hint,.dsh-ig-status{margin:0;color:var(--dsw-alias-label-tertiary,#7b818b);font-size:12px;line-height:1.4}
@@ -1078,16 +1145,16 @@ window.__ModuleLoader__.load({
 		/** Required browser services. */
 		const inject = [
 			"slots",
-			"connection",
 			"remote",
+			"remote.credentials",
 			"settingsScope",
 			"locale"
 		];
 		/** Mount the settings card, generated-image card, and native conversation gallery view. */
 		function apply(ctx) {
-			const { api } = ctx.get("connection");
 			const scope = ctx.settingsScope.bind({ namespace: IMAGE_GENERATION_NAMESPACE });
 			const locale = ctx.get("locale");
+			const credentials = ctx.remote.credentials;
 			ctx.effect(() => {
 				const style = document.createElement("style");
 				style.dataset.plugin = "dsh-image-gen";
@@ -1103,7 +1170,7 @@ window.__ModuleLoader__.load({
 				key: IMAGE_GENERATION_NAMESPACE,
 				inject: () => ({
 					scope,
-					credentials: api.credentials,
+					credentials,
 					locale
 				})
 			}, ImageGenerationSettingsCard));
@@ -1135,6 +1202,9 @@ window.__ModuleLoader__.load({
 			const [provider, setProvider] = (0, react.useState)("google");
 			const [model, setModel] = (0, react.useState)("");
 			const [baseURL, setBaseURL] = (0, react.useState)("");
+			const [workflowJson, setWorkflowJson] = (0, react.useState)("");
+			const [workflowName, setWorkflowName] = (0, react.useState)("");
+			const [timeoutSeconds, setTimeoutSeconds] = (0, react.useState)(DEFAULT_COMFYUI_TIMEOUT_MS / 1e3);
 			const [saveToWorkspace, setSaveToWorkspace] = (0, react.useState)(true);
 			const [workspaceFolder, setWorkspaceFolder] = (0, react.useState)("dsh-image-gen");
 			const [key, setKey] = (0, react.useState)("");
@@ -1158,7 +1228,8 @@ window.__ModuleLoader__.load({
 				google: t("providerGoogle"),
 				openai: t("providerOpenAI"),
 				seedream: t("providerSeedream"),
-				dashscope: t("providerDashScope")
+				dashscope: t("providerDashScope"),
+				comfyui: t("providerComfyUI")
 			};
 			(0, react.useEffect)(() => {
 				const value = snapshot.value;
@@ -1166,13 +1237,21 @@ window.__ModuleLoader__.load({
 				setProvider(next);
 				setModel(modelOf(next, value));
 				setBaseURL(baseURLOf(next, value));
+				setWorkflowJson(value?.comfyuiWorkflowJson ?? "");
+				setWorkflowName(value?.comfyuiWorkflowName ?? "");
+				setTimeoutSeconds(Math.max(1, Math.round((value?.comfyuiTimeoutMs ?? 3e5) / 1e3)));
 				setSaveToWorkspace(value?.saveToWorkspace ?? true);
 				setWorkspaceFolder(value?.workspaceFolder ?? "dsh-image-gen");
 			}, [snapshot]);
 			(0, react.useEffect)(() => {
+				const keyRef = KEY_REF[provider];
+				if (keyRef === void 0) {
+					setConfigured(void 0);
+					return;
+				}
 				let active = true;
-				props.credentials.describe({ refs: [KEY_REF[provider]] }).then((response) => {
-					if (active) setConfigured(response.result.ok ? response.result.value.credentials[KEY_REF[provider]]?.configured ?? false : void 0);
+				props.credentials.describe([keyRef]).then((response) => {
+					if (active) setConfigured(response.ok ? response.value?.[keyRef]?.configured ?? false : void 0);
 				}).catch(() => {
 					if (active) setConfigured(void 0);
 				});
@@ -1186,16 +1265,23 @@ window.__ModuleLoader__.load({
 				setMessage("");
 				try {
 					await props.scope.set("provider", provider);
-					await props.scope.set(provider === "google" ? "googleModel" : provider === "openai" ? "openaiModel" : provider === "seedream" ? "seedreamModel" : "dashscopeModel", model);
-					await props.scope.set(provider === "google" ? "googleEndpoint" : provider === "openai" ? "openaiBaseURL" : provider === "seedream" ? "seedreamBaseURL" : "dashscopeEndpoint", baseURL);
+					if (provider === "comfyui") {
+						validateComfyUIWorkflowJson(workflowJson);
+						await props.scope.set("comfyuiBaseURL", baseURL);
+						await props.scope.set("comfyuiWorkflowJson", workflowJson);
+						await props.scope.set("comfyuiWorkflowName", workflowName);
+						await props.scope.set("comfyuiTimeoutMs", Math.max(1, Math.round(timeoutSeconds)) * 1e3);
+					} else {
+						await props.scope.set(provider === "google" ? "googleModel" : provider === "openai" ? "openaiModel" : provider === "seedream" ? "seedreamModel" : "dashscopeModel", model);
+						await props.scope.set(provider === "google" ? "googleEndpoint" : provider === "openai" ? "openaiBaseURL" : provider === "seedream" ? "seedreamBaseURL" : "dashscopeEndpoint", baseURL);
+					}
 					await props.scope.set("saveToWorkspace", saveToWorkspace);
 					await props.scope.set("workspaceFolder", workspaceFolder.trim());
 					if (key.trim().length > 0) {
-						const response = await props.credentials.set({
-							ref: KEY_REF[provider],
-							value: key.trim()
-						});
-						if (!response.result.ok) throw new Error(response.result.error.message);
+						const keyRef = KEY_REF[provider];
+						if (keyRef === void 0) throw new Error("ComfyUI does not use an API key in this version");
+						const response = await props.credentials.set(keyRef, key.trim());
+						if (!response.ok) throw new Error(response.error?.message ?? "Failed to save API key");
 						setKey("");
 						setConfigured(true);
 					}
@@ -1207,6 +1293,23 @@ window.__ModuleLoader__.load({
 				}
 			};
 			const keyStatus = configured === void 0 ? t("checkingKey") : configured ? t("keyConfigured") : t("keyNotConfigured");
+			const workflowStatus = workflowName.length > 0 ? t("workflowImported", { name: workflowName }) : t("workflowMissing");
+			const importWorkflow = async (event) => {
+				const file = event.target.files?.[0];
+				event.target.value = "";
+				if (file === void 0) return;
+				setMessage("");
+				try {
+					if (file.size > 5242880) throw new Error(t("workflowTooLarge"));
+					const json = await file.text();
+					validateComfyUIWorkflowJson(json);
+					setWorkflowJson(json);
+					setWorkflowName(file.name);
+					setMessage(t("workflowImported", { name: file.name }));
+				} catch (cause) {
+					setMessage(cause instanceof Error ? cause.message : String(cause));
+				}
+			};
 			return (0, react_jsx_runtime.jsxs)("li", {
 				className: `dsh-ig-card ${open ? "dsh-ig-card-open" : ""}`,
 				children: [(0, react_jsx_runtime.jsxs)("button", {
@@ -1279,6 +1382,10 @@ window.__ModuleLoader__.load({
 										(0, react_jsx_runtime.jsx)("option", {
 											value: "dashscope",
 											children: t("providerDashScope")
+										}),
+										(0, react_jsx_runtime.jsx)("option", {
+											value: "comfyui",
+											children: t("providerComfyUI")
 										})
 									]
 								}),
@@ -1288,7 +1395,7 @@ window.__ModuleLoader__.load({
 								})
 							]
 						}),
-						(0, react_jsx_runtime.jsxs)("label", {
+						provider !== "comfyui" ? (0, react_jsx_runtime.jsxs)("label", {
 							className: "dsh-ig-field",
 							children: [
 								(0, react_jsx_runtime.jsx)("span", {
@@ -1307,10 +1414,10 @@ window.__ModuleLoader__.load({
 								}),
 								(0, react_jsx_runtime.jsx)("span", {
 									className: "dsh-ig-hint",
-									children: t("apiKeyHint", { key: KEY_REF[provider] })
+									children: t("apiKeyHint", { key: KEY_REF[provider] ?? "" })
 								})
 							]
-						}),
+						}) : null,
 						(0, react_jsx_runtime.jsxs)("label", {
 							className: "dsh-ig-field",
 							children: [
@@ -1340,11 +1447,11 @@ window.__ModuleLoader__.load({
 								}),
 								(0, react_jsx_runtime.jsx)("span", {
 									className: "dsh-ig-hint",
-									children: provider === "google" ? t("endpointHintGoogle") : provider === "openai" ? t("endpointHintOpenAI") : provider === "seedream" ? t("endpointHintSeedream") : t("endpointHintDashScope")
+									children: provider === "google" ? t("endpointHintGoogle") : provider === "openai" ? t("endpointHintOpenAI") : provider === "seedream" ? t("endpointHintSeedream") : provider === "dashscope" ? t("endpointHintDashScope") : t("endpointHintComfyUI")
 								})
 							]
 						}),
-						(0, react_jsx_runtime.jsxs)("label", {
+						provider !== "comfyui" ? (0, react_jsx_runtime.jsxs)("label", {
 							className: "dsh-ig-field",
 							children: [(0, react_jsx_runtime.jsx)("span", {
 								className: "dsh-ig-label",
@@ -1357,7 +1464,61 @@ window.__ModuleLoader__.load({
 								},
 								required: true
 							})]
-						}),
+						}) : (0, react_jsx_runtime.jsxs)(react_jsx_runtime.Fragment, { children: [(0, react_jsx_runtime.jsxs)("div", {
+							className: "dsh-ig-field",
+							children: [
+								(0, react_jsx_runtime.jsx)("span", {
+									className: "dsh-ig-label",
+									children: t("workflow")
+								}),
+								(0, react_jsx_runtime.jsxs)("div", {
+									className: "dsh-ig-file-row",
+									children: [(0, react_jsx_runtime.jsxs)("label", {
+										className: "dsh-ig-file-button",
+										children: [(0, react_jsx_runtime.jsx)("input", {
+											className: "dsh-ig-file-input",
+											type: "file",
+											accept: ".json,application/json",
+											onChange: (event) => {
+												importWorkflow(event);
+											}
+										}), workflowName.length > 0 ? t("workflowReplace") : t("workflowImport")]
+									}), (0, react_jsx_runtime.jsx)("span", {
+										className: "dsh-ig-file-name",
+										title: workflowName,
+										children: workflowName || t("workflowMissing")
+									})]
+								}),
+								(0, react_jsx_runtime.jsx)("span", {
+									className: "dsh-ig-hint",
+									children: t("workflowHint")
+								})
+							]
+						}), (0, react_jsx_runtime.jsxs)("label", {
+							className: "dsh-ig-field",
+							children: [
+								(0, react_jsx_runtime.jsx)("span", {
+									className: "dsh-ig-label",
+									children: t("timeout")
+								}),
+								(0, react_jsx_runtime.jsx)("input", {
+									className: "dsh-ig-input",
+									type: "number",
+									min: "1",
+									max: "3600",
+									step: "1",
+									value: timeoutSeconds,
+									onChange: (event) => {
+										setTimeoutSeconds(Number(event.target.value));
+									},
+									required: true
+								}),
+								(0, react_jsx_runtime.jsx)("span", {
+									className: "dsh-ig-hint",
+									children: t("timeoutHint")
+								})
+							]
+						})] }),
 						(0, react_jsx_runtime.jsxs)("div", {
 							className: "dsh-ig-field",
 							children: [(0, react_jsx_runtime.jsxs)("label", {
@@ -1403,11 +1564,11 @@ window.__ModuleLoader__.load({
 							children: [(0, react_jsx_runtime.jsx)("p", {
 								className: "dsh-ig-status",
 								role: "status",
-								children: message || keyStatus
+								children: message || (provider === "comfyui" ? workflowStatus : keyStatus)
 							}), (0, react_jsx_runtime.jsx)("button", {
 								className: "dsh-ig-save",
 								type: "submit",
-								disabled: saving || !snapshot.writable,
+								disabled: saving || !snapshot.writable || provider === "comfyui" && workflowJson.length === 0,
 								children: saving ? t("saving") : t("save")
 							})]
 						})
@@ -1656,11 +1817,11 @@ window.__ModuleLoader__.load({
 			});
 		}
 		function modelOf(provider, value) {
-			const stored = provider === "google" ? value?.googleModel : provider === "openai" ? value?.openaiModel : provider === "seedream" ? value?.seedreamModel : value?.dashscopeModel;
+			const stored = provider === "google" ? value?.googleModel : provider === "openai" ? value?.openaiModel : provider === "seedream" ? value?.seedreamModel : provider === "dashscope" ? value?.dashscopeModel : value?.comfyuiWorkflowName;
 			return typeof stored === "string" && stored.length > 0 ? stored : DEFAULT_MODELS[provider];
 		}
 		function baseURLOf(provider, value) {
-			const stored = provider === "google" ? value?.googleEndpoint : provider === "openai" ? value?.openaiBaseURL : provider === "seedream" ? value?.seedreamBaseURL : value?.dashscopeEndpoint;
+			const stored = provider === "google" ? value?.googleEndpoint : provider === "openai" ? value?.openaiBaseURL : provider === "seedream" ? value?.seedreamBaseURL : provider === "dashscope" ? value?.dashscopeEndpoint : value?.comfyuiBaseURL;
 			return typeof stored === "string" && stored.length > 0 ? stored : DEFAULT_BASE_URLS[provider];
 		}
 		function imageRef(block) {