@mindstudio-ai/local-model-tunnel 0.3.1 → 0.3.2

package/dist/{chunk-PTK4SJQK.js → chunk-V3RKCMCQ.js}

@@ -47,10 +47,20 @@ function getProviderBaseUrl(name, defaultUrl) {
   const urls = config.get("providerBaseUrls");
   return urls[name] ?? defaultUrl;
 }
+function setProviderBaseUrl(name, url) {
+  const urls = config.get("providerBaseUrls");
+  urls[name] = url;
+  config.set("providerBaseUrls", urls);
+}
 function getProviderInstallPath(name) {
   const paths = config.get("providerInstallPaths");
   return paths[name];
 }
+function setProviderInstallPath(name, installPath) {
+  const paths = config.get("providerInstallPaths");
+  paths[name] = installPath;
+  config.set("providerInstallPaths", paths);
+}
 
 // src/api.ts
 function getHeaders() {
@@ -83,39 +93,20 @@ async function pollForRequest(models) {
   const data = await response.json();
   return data.request;
 }
-async function submitProgress(requestId, content) {
+async function submitProgress(requestId, content, type = "chunk") {
   const baseUrl = getApiBaseUrl();
   const response = await fetch(
     `${baseUrl}/v1/local-models/requests/${requestId}/progress`,
     {
       method: "POST",
       headers: getHeaders(),
-      body: JSON.stringify({ content })
+      body: JSON.stringify({ type, content })
     }
   );
   if (!response.ok) {
     console.warn(`Progress update failed: ${response.status}`);
   }
 }
-async function submitGenerationProgress(requestId, step, totalSteps, preview) {
-  const baseUrl = getApiBaseUrl();
-  const response = await fetch(
-    `${baseUrl}/v1/local-models/requests/${requestId}/progress`,
-    {
-      method: "POST",
-      headers: getHeaders(),
-      body: JSON.stringify({
-        type: "generation",
-        step,
-        totalSteps,
-        preview
-      })
-    }
-  );
-  if (!response.ok) {
-    console.warn(`Generation progress update failed: ${response.status}`);
-  }
-}
 async function submitResult(requestId, success, result, error) {
   const baseUrl = getApiBaseUrl();
   const response = await fetch(
@@ -145,7 +136,7 @@ async function verifyApiKey() {
     return false;
   }
 }
-async function registerLocalModel(modelNameOrOptions, provider = "ollama", modelType = "llm_chat") {
+async function syncLocalModel(modelNameOrOptions, provider = "ollama", modelType = "llm_chat") {
   const baseUrl = getApiBaseUrl();
   let payload;
   if (typeof modelNameOrOptions === "string") {
@@ -169,10 +160,29 @@ async function registerLocalModel(modelNameOrOptions, provider = "ollama", model
   });
   if (!response.ok) {
     const errorText = await response.text();
-    throw new Error(`Register failed: ${response.status} ${errorText}`);
+    throw new Error(`Sync failed: ${response.status} ${errorText}`);
   }
 }
-async function getRegisteredModels() {
+async function updateLocalModel(options) {
+  const baseUrl = getApiBaseUrl();
+  const payload = {
+    modelId: options.modelId,
+    modelName: options.modelName,
+    provider: options.provider,
+    modelType: options.modelType || "llm_chat",
+    parameters: options.parameters
+  };
+  const response = await fetch(`${baseUrl}/v1/local-models/models/update`, {
+    method: "POST",
+    headers: getHeaders(),
+    body: JSON.stringify(payload)
+  });
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`Update failed: ${response.status} ${errorText}`);
+  }
+}
+async function getSyncedModels() {
   const baseUrl = getApiBaseUrl();
   const response = await fetch(`${baseUrl}/v1/local-models/models`, {
     method: "GET",
@@ -181,7 +191,7 @@ async function getRegisteredModels() {
   if (!response.ok) {
     const errorText = await response.text();
     throw new Error(
-      `Failed to fetch registered models: ${response.status} ${errorText}`
+      `Failed to fetch synced models: ${response.status} ${errorText}`
     );
   }
   const data = await response.json();
@@ -567,7 +577,7 @@ var StableDiffusionProvider = class {
       throw new Error(`Failed to switch model: ${error}`);
     }
   }
-  async generateImage(model, prompt, options) {
+  async generateImage(model, prompt, options, onProgress) {
     const currentModel = await this.getCurrentModel();
     if (currentModel && !currentModel.includes(model)) {
       await this.setModel(model);
@@ -582,44 +592,20 @@ var StableDiffusionProvider = class {
       seed: options?.seed ?? -1,
       sampler_name: options?.sampler || "Euler a"
     };
-    const response = await fetch(`${this.getBaseUrl()}/sdapi/v1/txt2img`, {
+    const generatePromise = fetch(`${this.getBaseUrl()}/sdapi/v1/txt2img`, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
       body: JSON.stringify(payload)
     });
-    if (!response.ok) {
-      const error = await response.text();
-      throw new Error(`Image generation failed: ${response.status} ${error}`);
-    }
-    const result = await response.json();
-    if (!result.images || result.images.length === 0) {
-      throw new Error("No images returned from Stable Diffusion");
-    }
-    let info = {};
-    let seed;
-    try {
-      info = JSON.parse(result.info);
-      seed = typeof info.seed === "number" ? info.seed : void 0;
-    } catch {
-    }
-    return {
-      imageBase64: result.images[0],
-      mimeType: "image/png",
-      seed,
-      info
-    };
-  }
-  async generateImageWithProgress(model, prompt, options, onProgress) {
-    const generatePromise = this.generateImage(model, prompt, options);
     if (onProgress) {
       const pollProgress = async () => {
         while (true) {
           try {
-            const response = await fetch(
+            const response2 = await fetch(
               `${this.getBaseUrl()}/sdapi/v1/progress`
             );
-            if (!response.ok) break;
-            const progress = await response.json();
+            if (!response2.ok) break;
+            const progress = await response2.json();
             onProgress({
               step: progress.state.sampling_step,
               totalSteps: progress.state.sampling_steps,
@@ -635,7 +621,28 @@ var StableDiffusionProvider = class {
       pollProgress().catch(() => {
       });
     }
-    return generatePromise;
+    const response = await generatePromise;
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Image generation failed: ${response.status} ${error}`);
+    }
+    const result = await response.json();
+    if (!result.images || result.images.length === 0) {
+      throw new Error("No images returned from Stable Diffusion");
+    }
+    let info = {};
+    let seed;
+    try {
+      info = JSON.parse(result.info);
+      seed = typeof info.seed === "number" ? info.seed : void 0;
+    } catch {
+    }
+    return {
+      imageBase64: result.images[0],
+      mimeType: "image/png",
+      seed,
+      info
+    };
   }
   /**
    * Fetch available samplers from the backend
@@ -718,7 +725,7 @@ var StableDiffusionProvider = class {
         label: "Steps",
         variable: "steps",
         helpText: "Number of denoising steps. More steps = higher quality but slower.",
-        defaultValue: 20,
+        defaultValue: "20",
         numberOptions: {
           min: 1,
           max: 150,
@@ -730,7 +737,7 @@ var StableDiffusionProvider = class {
         label: "CFG Scale",
         variable: "cfgScale",
         helpText: "How strongly the image should follow the prompt. Higher = more literal.",
-        defaultValue: 7,
+        defaultValue: "7",
         numberOptions: {
           min: 1,
           max: 30,
@@ -738,15 +745,11 @@ var StableDiffusionProvider = class {
         }
       },
       {
-        type: "number",
+        type: "seed",
         label: "Seed",
         variable: "seed",
         helpText: "A specific value used to guide the 'randomness' of generation. Use -1 for random.",
-        defaultValue: -1,
-        numberOptions: {
-          min: -1,
-          max: 2147483647
-        }
+        defaultValue: "-1"
       },
       {
         type: "text",
@@ -761,665 +764,537 @@ var StableDiffusionProvider = class {
 var stable_diffusion_default = new StableDiffusionProvider();
 
 // src/providers/comfyui/index.ts
+import * as path7 from "path";
+
+// src/providers/comfyui/workflow-discovery.ts
+import * as fs4 from "fs";
+import * as path5 from "path";
+
+// src/providers/comfyui/converter-install.ts
 import * as fs3 from "fs";
 import * as path4 from "path";
-import * as os4 from "os";
-
-// src/providers/comfyui/workflows/ltx-video.ts
-var LTX_VIDEO_DEFAULTS = {
-  model: "ltx-video-2b-v0.9.5.safetensors",
-  textEncoder: "t5xxl_fp16.safetensors",
-  prompt: "",
-  negativePrompt: "worst quality, blurry, distorted, disfigured, motion smear, motion artifacts",
-  width: 512,
-  height: 320,
-  numFrames: 41,
-  fps: 8,
-  steps: 20,
-  cfgScale: 3,
-  seed: -1
-};
-function buildLtxVideoWorkflow(params) {
-  const p = { ...LTX_VIDEO_DEFAULTS, ...params };
-  const seed = p.seed === -1 ? Math.floor(Math.random() * 2 ** 32) : p.seed;
-  return {
-    // Node 1: Load checkpoint (MODEL + VAE, CLIP output unused)
-    "1": {
-      class_type: "CheckpointLoaderSimple",
-      inputs: {
-        ckpt_name: p.model
-      }
-    },
-    // Node 2: Load text encoder (T5-XXL) separately
-    "2": {
-      class_type: "CLIPLoader",
-      inputs: {
-        clip_name: p.textEncoder,
-        type: "ltxv"
-      }
-    },
-    // Node 3: Positive prompt encoding (CLIP from CLIPLoader, NOT from checkpoint)
-    "3": {
-      class_type: "CLIPTextEncode",
-      inputs: {
-        text: p.prompt,
-        clip: ["2", 0]
-      }
-    },
-    // Node 4: Negative prompt encoding
-    "4": {
-      class_type: "CLIPTextEncode",
-      inputs: {
-        text: p.negativePrompt,
-        clip: ["2", 0]
-      }
-    },
-    // Node 5: Empty latent video
-    "5": {
-      class_type: "EmptyLTXVLatentVideo",
-      inputs: {
-        width: p.width,
-        height: p.height,
-        length: p.numFrames,
-        batch_size: 1
-      }
-    },
-    // Node 6: KSampler
-    "6": {
-      class_type: "KSampler",
-      inputs: {
-        model: ["1", 0],
-        positive: ["3", 0],
-        negative: ["4", 0],
-        latent_image: ["5", 0],
-        seed,
-        steps: p.steps,
-        cfg: p.cfgScale,
-        sampler_name: "euler",
-        scheduler: "normal",
-        denoise: 1
-      }
-    },
-    // Node 7: VAE Decode (VAE from checkpoint, slot 2)
-    "7": {
-      class_type: "VAEDecode",
-      inputs: {
-        samples: ["6", 0],
-        vae: ["1", 2]
-      }
-    },
-    // Node 8: Save as MP4 via VideoHelperSuite
-    "8": {
-      class_type: "VHS_VideoCombine",
-      inputs: {
-        images: ["7", 0],
-        frame_rate: p.fps,
-        loop_count: 0,
-        filename_prefix: "ltxv_output",
-        format: "video/h264-mp4",
-        pingpong: false,
-        save_output: true
+var CONVERTER_DIR = "comfyui-workflow-to-api-converter-endpoint";
+var GITHUB_RAW_BASE = "https://raw.githubusercontent.com/SethRobinson/comfyui-workflow-to-api-converter-endpoint/main";
+var FILES_TO_DOWNLOAD = ["__init__.py", "workflow_converter.py"];
+async function ensureConverterInstalled(installPath) {
+  const customNodesDir = path4.join(installPath, "custom_nodes");
+  const converterDir = path4.join(customNodesDir, CONVERTER_DIR);
+  const allFilesExist = FILES_TO_DOWNLOAD.every(
+    (f) => fs3.existsSync(path4.join(converterDir, f))
+  );
+  if (allFilesExist) {
+    return true;
+  }
+  if (!fs3.existsSync(customNodesDir)) {
+    return false;
+  }
+  try {
+    if (!fs3.existsSync(converterDir)) {
+      fs3.mkdirSync(converterDir, { recursive: true });
+    }
+    for (const filename of FILES_TO_DOWNLOAD) {
+      const url = `${GITHUB_RAW_BASE}/${filename}`;
+      const response = await fetch(url, {
+        signal: AbortSignal.timeout(15e3)
+      });
+      if (!response.ok) {
+        throw new Error(`Failed to download ${filename}: ${response.status}`);
       }
+      const content = await response.text();
+      fs3.writeFileSync(path4.join(converterDir, filename), content, "utf-8");
     }
-  };
+    return true;
+  } catch {
+    return false;
+  }
+}
+var converterAvailableCache = null;
+async function isConverterEndpointAvailable(baseUrl) {
+  if (converterAvailableCache !== null) {
+    return converterAvailableCache;
+  }
+  try {
+    const response = await fetch(`${baseUrl}/workflow/convert`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ nodes: [], links: [] }),
+      signal: AbortSignal.timeout(5e3)
+    });
+    converterAvailableCache = response.ok;
+    return converterAvailableCache;
+  } catch {
+    converterAvailableCache = false;
+    return false;
+  }
+}
+function resetConverterCache() {
+  converterAvailableCache = null;
+}
+async function convertWorkflow(baseUrl, uiWorkflow) {
+  const response = await fetch(`${baseUrl}/workflow/convert`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify(uiWorkflow),
+    signal: AbortSignal.timeout(1e4)
+  });
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`Workflow conversion failed: ${response.status} ${errorText}`);
+  }
+  return await response.json();
 }
-var LTX_VIDEO_OUTPUT_NODE = "8";
 
-// src/providers/comfyui/workflows/wan2.1.ts
-var WAN21_DEFAULTS = {
-  model: "wan2.1_t2v_1.3B_fp16.safetensors",
-  textEncoder: "umt5_xxl_fp8_e4m3fn_scaled.safetensors",
-  vae: "wan_2.1_vae.safetensors",
-  prompt: "",
-  negativePrompt: "worst quality, blurry, distorted",
-  width: 480,
-  height: 320,
-  numFrames: 25,
-  fps: 8,
-  steps: 20,
-  cfgScale: 5,
-  seed: -1
-};
-function buildWan21Workflow(params) {
-  const p = { ...WAN21_DEFAULTS, ...params };
-  const seed = p.seed === -1 ? Math.floor(Math.random() * 2 ** 32) : p.seed;
-  return {
-    // Node 1: Load diffusion model (UNET)
-    "1": {
-      class_type: "UNETLoader",
-      inputs: {
-        unet_name: p.model,
-        weight_dtype: "default"
-      }
-    },
-    // Node 2: Load text encoder (UMT5-XXL)
-    "2": {
-      class_type: "CLIPLoader",
-      inputs: {
-        clip_name: p.textEncoder,
-        type: "wan"
-      }
-    },
-    // Node 3: Load VAE
-    "3": {
-      class_type: "VAELoader",
-      inputs: {
-        vae_name: p.vae
-      }
-    },
-    // Node 4: Positive prompt encoding
-    "4": {
-      class_type: "CLIPTextEncode",
-      inputs: {
-        text: p.prompt,
-        clip: ["2", 0]
-      }
-    },
-    // Node 5: Negative prompt encoding
-    "5": {
-      class_type: "CLIPTextEncode",
-      inputs: {
-        text: p.negativePrompt,
-        clip: ["2", 0]
-      }
-    },
-    // Node 6: Empty latent image (for video frames)
-    "6": {
-      class_type: "EmptySD3LatentImage",
-      inputs: {
-        width: p.width,
-        height: p.height,
-        batch_size: p.numFrames
-      }
-    },
-    // Node 7: KSampler
-    "7": {
-      class_type: "KSampler",
-      inputs: {
-        model: ["1", 0],
-        positive: ["4", 0],
-        negative: ["5", 0],
-        latent_image: ["6", 0],
-        seed,
-        steps: p.steps,
-        cfg: p.cfgScale,
-        sampler_name: "euler",
-        scheduler: "normal",
-        denoise: 1
-      }
-    },
-    // Node 8: VAE Decode
-    "8": {
-      class_type: "VAEDecode",
-      inputs: {
-        samples: ["7", 0],
-        vae: ["3", 0]
+// src/providers/comfyui/workflow-discovery.ts
+var VIDEO_OUTPUT_NODES = ["VHS_VideoCombine", "SaveVideo"];
+var IMAGE_OUTPUT_NODES = ["SaveImage", "PreviewImage"];
+async function discoverWorkflows(baseUrl, installPath) {
+  resetConverterCache();
+  let converterJustInstalled = false;
+  if (installPath) {
+    const wasInstalled = await ensureConverterInstalled(installPath);
+    if (wasInstalled) {
+      converterJustInstalled = true;
+    }
+  }
+  const converterAvailable = await isConverterEndpointAvailable(baseUrl);
+  const needsRestart = converterJustInstalled && !converterAvailable;
+  const workflowFiles = await listWorkflowFiles(baseUrl, installPath);
+  const converted = { image: [], video: [] };
+  const unconvertedCapabilities = /* @__PURE__ */ new Set();
+  for (const file of workflowFiles) {
+    try {
+      const workflowJson = await fetchWorkflowJson(baseUrl, installPath, file);
+      if (!workflowJson) continue;
+      let apiWorkflow;
+      if (isApiFormat(workflowJson)) {
+        apiWorkflow = workflowJson;
+      } else if (converterAvailable) {
+        try {
+          apiWorkflow = await convertWorkflow(baseUrl, workflowJson);
+        } catch {
+          continue;
+        }
+      } else {
+        unconvertedCapabilities.add("image");
+        continue;
       }
-    },
-    // Node 9: Save as MP4 via VideoHelperSuite
-    "9": {
-      class_type: "VHS_VideoCombine",
-      inputs: {
-        images: ["8", 0],
-        frame_rate: p.fps,
-        loop_count: 0,
-        filename_prefix: "wan21_output",
-        format: "video/h264-mp4",
-        pingpong: false,
-        save_output: true
+      const capability = detectCapability(apiWorkflow);
+      const name = path5.basename(file, path5.extname(file));
+      converted[capability].push({ name, workflow: apiWorkflow });
+    } catch {
+    }
+  }
+  const models = [];
+  for (const capability of ["image", "video"]) {
+    if (converted[capability].length > 0) {
+      const displayName = capability === "image" ? "ComfyUI Image Generation" : "ComfyUI Video Generation";
+      const workflowParam = {
+        type: "comfyWorkflow",
+        variable: "workflow",
+        label: "Workflow",
+        comfyWorkflowOptions: { availableWorkflows: converted[capability] }
+      };
+      models.push({
+        name: displayName,
+        provider: "comfyui",
+        capability,
+        parameters: [workflowParam]
+      });
+    }
+  }
+  for (const capability of unconvertedCapabilities) {
+    if (converted[capability].length > 0) continue;
+    const displayName = capability === "image" ? "ComfyUI Image Generation" : "ComfyUI Video Generation";
+    models.push({
+      name: displayName,
+      provider: "comfyui",
+      capability,
+      statusHint: needsRestart ? "Restart ComfyUI to enable" : "Workflow converter not available"
+    });
+  }
+  return models;
+}
+async function listWorkflowFiles(baseUrl, installPath) {
+  try {
+    const response = await fetch(
+      `${baseUrl}/userdata?dir=workflows/&recurse=true&full_info=true`,
+      { signal: AbortSignal.timeout(5e3) }
+    );
+    if (response.ok) {
+      const data = await response.json();
+      return data.map((entry) => typeof entry === "string" ? entry : entry.path).filter((p) => p.endsWith(".json"));
+    }
+  } catch {
+  }
+  if (installPath) {
+    const workflowsDir = path5.join(installPath, "user", "default", "workflows");
+    return scanDirectory(workflowsDir);
+  }
+  return [];
+}
+function scanDirectory(dir) {
+  if (!fs4.existsSync(dir)) return [];
+  const results = [];
+  try {
+    const entries = fs4.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      const fullPath = path5.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        results.push(
+          ...scanDirectory(fullPath).map((f) => path5.join(entry.name, f))
+        );
+      } else if (entry.name.endsWith(".json")) {
+        results.push(entry.name);
       }
     }
-  };
+  } catch {
+  }
+  return results;
 }
-var WAN21_OUTPUT_NODE = "9";
-
-// src/providers/comfyui/workflows/index.ts
-var MODEL_REGISTRY = [
-  // LTX-Video models
-  {
-    pattern: /ltx[_-]?video/i,
-    config: {
-      family: "ltx-video",
-      displayName: "LTX-Video",
-      buildWorkflow: (params) => buildLtxVideoWorkflow({
-        model: params.model,
-        prompt: params.prompt,
-        negativePrompt: params.negativePrompt,
-        width: params.width,
-        height: params.height,
-        numFrames: params.numFrames,
-        fps: params.fps,
-        steps: params.steps,
-        cfgScale: params.cfgScale,
-        seed: params.seed
-      }),
-      outputNodeId: LTX_VIDEO_OUTPUT_NODE,
-      defaults: {
-        width: LTX_VIDEO_DEFAULTS.width,
-        height: LTX_VIDEO_DEFAULTS.height,
-        numFrames: LTX_VIDEO_DEFAULTS.numFrames,
-        fps: LTX_VIDEO_DEFAULTS.fps,
-        steps: LTX_VIDEO_DEFAULTS.steps,
-        cfgScale: LTX_VIDEO_DEFAULTS.cfgScale
+async function fetchWorkflowJson(baseUrl, installPath, filePath) {
+  try {
+    const userdataPath = `workflows/${filePath}`;
+    const response = await fetch(
+      `${baseUrl}/userdata/${encodeURIComponent(userdataPath)}`,
+      { signal: AbortSignal.timeout(5e3) }
+    );
+    if (response.ok) {
+      return await response.json();
+    }
+  } catch {
+  }
+  if (installPath) {
+    const fullPath = path5.join(
+      installPath,
+      "user",
+      "default",
+      "workflows",
+      filePath
+    );
+    try {
+      const content = fs4.readFileSync(fullPath, "utf-8");
+      return JSON.parse(content);
+    } catch {
+      return null;
+    }
+  }
+  return null;
+}
+function isApiFormat(json) {
+  const keys = Object.keys(json);
+  if (keys.length === 0) return false;
+  return keys.some((key) => {
+    const node = json[key];
+    return /^\d+$/.test(key) && typeof node === "object" && node !== null && "class_type" in node;
+  });
+}
+function detectCapability(apiWorkflow) {
+  for (const node of Object.values(apiWorkflow)) {
+    if (typeof node === "object" && node !== null && "class_type" in node) {
+      const classType = node.class_type;
+      if (VIDEO_OUTPUT_NODES.includes(classType)) {
+        return "video";
       }
     }
-  },
-  // Wan 2.1 models
-  {
-    pattern: /wan2[\._]?1/i,
-    config: {
-      family: "wan2.1",
-      displayName: "Wan 2.1",
-      buildWorkflow: (params) => buildWan21Workflow({
-        model: params.model,
-        prompt: params.prompt,
-        negativePrompt: params.negativePrompt,
-        width: params.width,
-        height: params.height,
-        numFrames: params.numFrames,
-        fps: params.fps,
-        steps: params.steps,
-        cfgScale: params.cfgScale,
-        seed: params.seed
-      }),
-      outputNodeId: WAN21_OUTPUT_NODE,
-      defaults: {
-        width: WAN21_DEFAULTS.width,
-        height: WAN21_DEFAULTS.height,
-        numFrames: WAN21_DEFAULTS.numFrames,
-        fps: WAN21_DEFAULTS.fps,
-        steps: WAN21_DEFAULTS.steps,
-        cfgScale: WAN21_DEFAULTS.cfgScale
+  }
+  for (const node of Object.values(apiWorkflow)) {
+    if (typeof node === "object" && node !== null && "class_type" in node) {
+      const classType = node.class_type;
+      if (IMAGE_OUTPUT_NODES.includes(classType)) {
+        return "image";
       }
     }
   }
-];
-function getWorkflowForModel(modelFilename) {
-  for (const entry of MODEL_REGISTRY) {
-    if (entry.pattern.test(modelFilename)) {
-      return entry.config;
+  return "image";
+}
+
+// src/providers/comfyui/workflow-executor.ts
+import * as path6 from "path";
+async function executeWorkflow(options) {
+  const { baseUrl, workflow, onProgress } = options;
+  const clientId = `mindstudio_${Date.now()}_${Math.random().toString(36).slice(2)}`;
+  const wsUrl = baseUrl.replace(/^http/, "ws") + `/ws?clientId=${clientId}`;
+  const submitResponse = await fetch(`${baseUrl}/prompt`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({
+      prompt: workflow,
+      client_id: clientId
+    })
+  });
+  if (!submitResponse.ok) {
+    const errorText = await submitResponse.text();
+    throw new Error(
+      `ComfyUI prompt submission failed: ${submitResponse.status} ${errorText}`
+    );
+  }
+  const submitResult2 = await submitResponse.json();
+  if (submitResult2.node_errors && Object.keys(submitResult2.node_errors).length > 0) {
+    throw new Error(
+      `ComfyUI workflow validation failed: ${JSON.stringify(submitResult2.node_errors)}`
+    );
+  }
+  const promptId = submitResult2.prompt_id;
+  await waitForCompletion(wsUrl, promptId, onProgress);
+  const historyResponse = await fetch(`${baseUrl}/history/${promptId}`, {
+    signal: AbortSignal.timeout(3e4)
+  });
+  if (!historyResponse.ok) {
+    throw new Error(
+      `Failed to fetch result history: ${historyResponse.status}`
+    );
+  }
+  const history = await historyResponse.json();
+  const promptHistory = history[promptId];
+  if (!promptHistory) {
+    throw new Error("No result found in ComfyUI history");
+  }
+  let outputFile = null;
+  for (const nodeOutputs of Object.values(promptHistory.outputs)) {
+    if (nodeOutputs.gifs && nodeOutputs.gifs.length > 0) {
+      outputFile = nodeOutputs.gifs[0];
+      break;
+    }
+    if (!outputFile && nodeOutputs.images && nodeOutputs.images.length > 0) {
+      outputFile = nodeOutputs.images[0];
     }
   }
-  return null;
+  if (!outputFile) {
+    throw new Error("No output files found in ComfyUI result");
+  }
+  const fileUrl = new URL(`${baseUrl}/view`);
+  fileUrl.searchParams.set("filename", outputFile.filename);
+  fileUrl.searchParams.set("subfolder", outputFile.subfolder || "");
+  fileUrl.searchParams.set("type", outputFile.type || "output");
+  const fileResponse = await fetch(fileUrl.toString(), {
+    signal: AbortSignal.timeout(6e4)
+  });
+  if (!fileResponse.ok) {
+    throw new Error(`Failed to download output file: ${fileResponse.status}`);
+  }
+  const fileBuffer = await fileResponse.arrayBuffer();
+  const dataBase64 = Buffer.from(fileBuffer).toString("base64");
+  const ext = path6.extname(outputFile.filename).toLowerCase();
+  const mimeType = getMimeType(ext);
+  return { dataBase64, mimeType, filename: outputFile.filename };
 }
-function isKnownVideoModel(modelFilename) {
-  return getWorkflowForModel(modelFilename) !== null;
+function getMimeType(ext) {
+  switch (ext) {
+    case ".mp4":
+      return "video/mp4";
+    case ".webm":
+      return "video/webm";
+    case ".webp":
+      return "image/webp";
+    case ".gif":
+      return "image/gif";
+    case ".png":
+      return "image/png";
+    case ".jpg":
+    case ".jpeg":
+      return "image/jpeg";
+    default:
+      return "application/octet-stream";
+  }
+}
+function waitForCompletion(wsUrl, promptId, onProgress) {
+  return new Promise((resolve, reject) => {
+    const timeoutMs = 30 * 60 * 1e3;
+    let ws;
+    const timeout = setTimeout(() => {
+      try {
+        ws?.close();
+      } catch {
+      }
+      reject(new Error("Workflow execution timed out after 30 minutes"));
+    }, timeoutMs);
+    try {
+      ws = new WebSocket(wsUrl);
+    } catch (err) {
+      clearTimeout(timeout);
+      reject(
+        new Error(
+          `Failed to connect to ComfyUI WebSocket: ${err instanceof Error ? err.message : err}`
+        )
+      );
+      return;
+    }
+    ws.onmessage = (event) => {
+      try {
+        const message = JSON.parse(
+          typeof event.data === "string" ? event.data : ""
+        );
+        if (message.type === "progress") {
+          const data = message.data;
+          if (!data.prompt_id || data.prompt_id === promptId) {
+            onProgress?.({
+              step: data.value,
+              totalSteps: data.max,
+              currentNode: data.node
+            });
+          }
+        }
+        if (message.type === "execution_success") {
+          const data = message.data;
+          if (data.prompt_id === promptId) {
+            clearTimeout(timeout);
+            ws.close();
+            resolve();
+          }
+        }
+        if (message.type === "execution_error") {
+          const data = message.data;
+          if (data.prompt_id === promptId) {
+            clearTimeout(timeout);
+            ws.close();
+            reject(
+              new Error(
+                `ComfyUI execution error${data.node_type ? ` in ${data.node_type}` : ""}: ${data.exception_message || "Unknown error"}`
+              )
+            );
+          }
+        }
+      } catch {
+      }
+    };
+    ws.onerror = () => {
+      clearTimeout(timeout);
+      reject(new Error("ComfyUI WebSocket error: connection failed"));
+    };
+    ws.onclose = (event) => {
+      if (!event.wasClean) {
+        clearTimeout(timeout);
+        reject(new Error("ComfyUI WebSocket connection closed unexpectedly"));
+      }
+    };
+  });
 }
 
 // src/providers/comfyui/readme.md
-var readme_default4 = '# ComfyUI\n\nComfyUI runs video generation models (LTX-Video, Wan2.1) locally. MindStudio handles all the workflow complexity for you -- you just need to install ComfyUI and download a model.\n\n**Default port:** 8188\n**Website:** https://www.comfy.org\n**GitHub:** https://github.com/comfyanonymous/ComfyUI\n\n## What You\'ll Need\n\n- **Python 3.10 or newer** -- Check by opening a terminal and typing `python3 --version`. If you don\'t have it, download from https://www.python.org/downloads/\n\n- **Git** -- Check by typing `git --version`. If you don\'t have it, download from https://git-scm.com/downloads\n\n- **A GPU with 8+ GB of VRAM** -- Video generation is demanding. Without enough GPU memory, generation will fail or be extremely slow.\n\n## Step 1: Install ComfyUI\n\nOpen a terminal and run these commands one at a time, waiting for each to finish before running the next.\n\nDownload ComfyUI:\n\n```\ngit clone https://github.com/comfyanonymous/ComfyUI.git ~/ComfyUI\n```\n\nGo into the folder:\n\n```\ncd ~/ComfyUI\n```\n\nCreate an isolated Python environment:\n\n```\npython3 -m venv venv\n```\n\nActivate the environment:\n\n```\nsource venv/bin/activate\n```\n\n**Windows users:** use `venv\\Scripts\\activate` instead.\n\nInstall dependencies (may take a few minutes):\n\n```\npip install -r requirements.txt\n```\n\n## Step 2: Download a Video Model\n\nYou need at least one video model for MindStudio to use.\n\n### LTX-Video (recommended to start)\n\nFastest option, good for getting up and running quickly.\n\n1. Go to https://huggingface.co/Lightricks/LTX-Video\n2. Download `ltx-video-2b-v0.9.5.safetensors`\n3. Move the file into:\n\n```\n~/ComfyUI/models/checkpoints/\n```\n\n### Wan2.1\n\nHigher quality but slower and needs more VRAM. Requires multiple files -- make sure you download all of them or it won\'t work.\n\n1. Go to https://huggingface.co/Comfy-Org/Wan2.1_ComfyUI_repackaged\n2. Place UNET files in `~/ComfyUI/models/diffusion_models/`\n3. Place text encoder files in `~/ComfyUI/models/text_encoders/`\n4. Place VAE files in `~/ComfyUI/models/vae/`\n\n## Step 3: Start the Server\n\nEvery time you want to use ComfyUI with MindStudio, open a terminal and run:\n\n```\ncd ~/ComfyUI && source venv/bin/activate && python main.py --listen\n```\n\n**Windows users:**\n\n```\ncd %USERPROFILE%\\ComfyUI && venv\\Scripts\\activate && python main.py --listen\n```\n\n**Important:** The `--listen` flag is required. Without it, MindStudio cannot connect to the server.\n\nLeave this terminal window open. When you see "To see the GUI go to: http://0.0.0.0:8188", the server is ready. Go back to the tunnel and select **Refresh Providers** -- your models should appear.\n\n## Troubleshooting\n\n- **MindStudio says ComfyUI is "not running"** -- Make sure you started with the `--listen` flag. Without it, the server won\'t accept connections from MindStudio.\n\n- **Server is running but no models show up** -- Check that your model files are in the right folders under `~/ComfyUI/models/`. Checkpoint files go in `checkpoints/`, UNET files go in `diffusion_models/`.\n\n- **Generation fails with workflow errors** -- For Wan2.1, you need all three files (UNET, text encoder, VAE). If any are missing, generation will fail.\n\n- **"CUDA out of memory"** -- Video generation needs a lot of GPU memory. Try reducing the resolution or number of frames in your generation settings, or use LTX-Video which is lighter.\n\n- **Server crashes mid-generation** -- Press Ctrl+C in the terminal and run the start command again.\n';
+var readme_default4 = "# ComfyUI\n\nComfyUI is a node-based workflow tool for running image and video generation models locally. MindStudio automatically discovers your saved workflows and makes them available as models -- any workflow you build or download in ComfyUI can be used through MindStudio.\n\n**Default port:** 8188\n**Website:** https://www.comfy.org\n\n## What You'll Need\n\n- **A GPU with 8+ GB of VRAM** -- Image and video generation is demanding. Without enough GPU memory, generation will fail or be extremely slow.\n\n## Step 1: Install ComfyUI\n\nDownload and install ComfyUI Desktop from the official website:\n\nhttps://www.comfy.org/download\n\nThe installer handles Python, dependencies, and everything else automatically. Follow the on-screen prompts to complete setup.\n\n## Step 2: Save a Workflow\n\nMindStudio discovers workflows you've saved in ComfyUI. If you don't have any saved workflows yet, open the ComfyUI interface in your browser (http://127.0.0.1:8188), build or load a workflow, and save it using the menu. Downloaded workflow files placed in ComfyUI's workflows folder will also be discovered.\n\nAny workflow that produces image or video output will work. MindStudio detects the output type automatically based on the nodes in your workflow.\n\n## Step 3: Start the Server\n\nOpen ComfyUI Desktop. Once it's running, go back to the tunnel and select **Refresh Providers** -- your saved workflows should appear as models.\n\nIf you're running ComfyUI from the command line instead, start it with:\n\n```\ncd ~/ComfyUI && python main.py --listen\n```\n\n**Important:** The `--listen` flag is required when running from the command line. Without it, MindStudio cannot connect to the server.\n\n## Tip: Workflow Converter\n\nMindStudio automatically installs a custom node called `comfyui-workflow-to-api-converter-endpoint` into ComfyUI's `custom_nodes/` folder. This converts workflows saved in ComfyUI's UI format into the API format needed for execution. **After the first run, you'll need to restart ComfyUI once** so it picks up the new node \u2014 after that, it works automatically. If the auto-install doesn't work (e.g. permissions issues), you can install it manually by cloning https://github.com/SethRobinson/comfyui-workflow-to-api-converter-endpoint into your ComfyUI `custom_nodes/` directory and restarting ComfyUI. Without this node, only workflows already saved in API format will be discovered.\n\n## Troubleshooting\n\n- **MindStudio says ComfyUI is \"not running\"** -- Make sure ComfyUI Desktop is open, or if running from the terminal, that you started with the `--listen` flag.\n\n- **Server is running but no workflows show up** -- Make sure you have at least one saved workflow in ComfyUI. Open the ComfyUI interface, load or build a workflow, and save it.\n\n- **\"CUDA out of memory\"** -- Your GPU doesn't have enough memory for the workflow you're running. Try a lighter model or reduce resolution in your workflow.\n\n- **Server crashes mid-generation** -- Restart ComfyUI Desktop, or press Ctrl+C in the terminal and run the start command again.\n";
 
 // src/providers/comfyui/index.ts
+var COMFYUI_PORTS = [8e3, 8188];
 var ComfyUIProvider = class {
   name = "comfyui";
   displayName = "ComfyUI";
-  description = "Generate videos locally using node-based workflows. Supports LTX-Video and Wan2.1.";
-  capabilities = ["video"];
+  description = "Run any saved ComfyUI workflow \u2014 images, video, and more.";
+  capabilities = ["image", "video"];
   readme = readme_default4;
-  defaultBaseUrl = "http://127.0.0.1:8188";
+  defaultBaseUrl = "http://127.0.0.1:8000";
   get baseUrl() {
     return getProviderBaseUrl(this.name, this.defaultBaseUrl);
   }
   getBaseUrl() {
     return this.baseUrl;
   }
-  async isRunning() {
-    try {
-      const response = await fetch(`${this.getBaseUrl()}/system_stats`, {
-        method: "GET",
-        signal: AbortSignal.timeout(5e3)
-      });
-      return response.ok;
-    } catch {
-      return false;
-    }
-  }
-  async detect() {
-    const savedPath = getProviderInstallPath(this.name);
-    const possiblePaths = [
-      ...savedPath ? [savedPath] : [],
-      path4.join(os4.homedir(), "ComfyUI"),
-      path4.join(os4.homedir(), "comfyui"),
-      path4.join(os4.homedir(), "Projects", "ComfyUI"),
-      path4.join(os4.homedir(), "Code", "ComfyUI")
-    ];
-    let installed = false;
-    for (const p of possiblePaths) {
-      if (fs3.existsSync(path4.join(p, "main.py")) && fs3.existsSync(path4.join(p, "requirements.txt"))) {
-        installed = true;
-        break;
+  /**
+   * Try to reach ComfyUI on the configured URL, then fall back to known ports.
+   * Persists whichever URL responds so future calls go direct.
+   */
+  async findRunningUrl() {
+    const configured = this.getBaseUrl();
+    if (await this.checkUrl(configured)) return configured;
+    for (const port of COMFYUI_PORTS) {
+      const url = `http://127.0.0.1:${port}`;
+      if (url === configured) continue;
+      if (await this.checkUrl(url)) {
+        setProviderBaseUrl(this.name, url);
+        return url;
       }
     }
-    let running = false;
+    return null;
+  }
+  async checkUrl(url) {
     try {
-      const response = await fetch("http://127.0.0.1:8188/system_stats", {
+      const response = await fetch(`${url}/system_stats`, {
         signal: AbortSignal.timeout(1e3)
       });
-      running = response.ok;
-      if (running) installed = true;
+      return response.ok;
     } catch {
-      running = false;
+      return false;
     }
-    return { installed, running };
   }
   /**
-   * Discover video models by scanning ComfyUI's model directories.
+   * Query the running ComfyUI server for its install path via /internal/folder_paths.
+   * Derives the root from the custom_nodes path and persists it for future use.
    */
-  async discoverModels() {
-    const models = [];
-    try {
-      const response = await fetch(
-        `${this.getBaseUrl()}/object_info/CheckpointLoaderSimple`,
-        { signal: AbortSignal.timeout(5e3) }
-      );
-      if (response.ok) {
-        const data = await response.json();
-        const nodeInfo = data.CheckpointLoaderSimple;
-        const checkpoints = nodeInfo?.input?.required?.ckpt_name?.[0] || [];
-        for (const name of checkpoints) {
-          if (isKnownVideoModel(name)) {
-            const workflow = getWorkflowForModel(name);
-            models.push({
-              name,
-              provider: this.name,
-              capability: "video",
-              parameterSize: workflow?.displayName
-            });
-          }
-        }
-      }
-    } catch {
-    }
+  async queryInstallPath(baseUrl) {
     try {
-      const response = await fetch(
-        `${this.getBaseUrl()}/object_info/UNETLoader`,
-        { signal: AbortSignal.timeout(5e3) }
-      );
-      if (response.ok) {
-        const data = await response.json();
-        const nodeInfo = data.UNETLoader;
-        const unetModels = nodeInfo?.input?.required?.unet_name?.[0] || [];
-        for (const name of unetModels) {
-          if (isKnownVideoModel(name) && !models.some((m) => m.name === name)) {
-            const workflow = getWorkflowForModel(name);
-            models.push({
-              name,
-              provider: this.name,
-              capability: "video",
-              parameterSize: workflow?.displayName
-            });
-          }
-        }
-      }
+      const response = await fetch(`${baseUrl}/internal/folder_paths`, {
+        signal: AbortSignal.timeout(3e3)
+      });
+      if (!response.ok) return null;
+      const data = await response.json();
+      const customNodesPaths = data.custom_nodes;
+      if (!customNodesPaths || customNodesPaths.length === 0) return null;
+      const installPath = path7.dirname(customNodesPaths[0]);
+      setProviderInstallPath(this.name, installPath);
+      return installPath;
     } catch {
+      return null;
     }
-    if (models.length === 0) {
-      const installPath = getProviderInstallPath(this.name);
-      if (installPath) {
-        const dirs = [
-          path4.join(installPath, "models", "checkpoints"),
-          path4.join(installPath, "models", "diffusion_models")
-        ];
-        for (const dir of dirs) {
-          if (fs3.existsSync(dir)) {
-            try {
-              const files = fs3.readdirSync(dir);
-              for (const file of files) {
-                if (isKnownVideoModel(file) && !models.some((m) => m.name === file)) {
-                  const workflow = getWorkflowForModel(file);
-                  models.push({
-                    name: file,
-                    provider: this.name,
-                    capability: "video",
-                    parameterSize: workflow?.displayName
-                  });
-                }
-              }
-            } catch {
-            }
-          }
-        }
+  }
+  async isRunning() {
+    return await this.findRunningUrl() !== null;
+  }
+  async detect() {
+    const runningUrl = await this.findRunningUrl();
+    if (runningUrl) {
+      const queriedPath = await this.queryInstallPath(runningUrl);
+      if (queriedPath) {
+        ensureConverterInstalled(queriedPath).catch(() => {
+        });
       }
+      return { installed: true, running: true };
     }
-    return models;
+    const savedPath = getProviderInstallPath(this.name);
+    return { installed: !!savedPath, running: false };
   }
   /**
-   * Generate a video using ComfyUI.
+   * Discover workflow-based models from user-saved ComfyUI workflows.
    */
-  async generateVideo(model, prompt, options, onProgress) {
-    const baseUrl = this.getBaseUrl();
-    const workflowConfig = getWorkflowForModel(model);
-    if (!workflowConfig) {
-      throw new Error(
-        `No workflow template found for model: ${model}. Supported families: LTX-Video, Wan2.1`
-      );
-    }
-    const defaults = workflowConfig.defaults;
-    const seed = options?.seed !== void 0 && options.seed !== -1 ? options.seed : Math.floor(Math.random() * 2 ** 32);
-    const workflow = workflowConfig.buildWorkflow({
-      model,
-      prompt,
-      negativePrompt: options?.negativePrompt || "worst quality, blurry, distorted",
-      width: options?.width || defaults.width,
-      height: options?.height || defaults.height,
-      numFrames: options?.numFrames || defaults.numFrames,
-      fps: options?.fps || defaults.fps,
-      steps: options?.steps || defaults.steps,
-      cfgScale: options?.cfgScale || defaults.cfgScale,
-      seed
-    });
-    const clientId = `mindstudio_${Date.now()}_${Math.random().toString(36).slice(2)}`;
-    const wsUrl = baseUrl.replace(/^http/, "ws") + `/ws?clientId=${clientId}`;
-    const submitResponse = await fetch(`${baseUrl}/prompt`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({
-        prompt: workflow,
-        client_id: clientId
-      })
-    });
-    if (!submitResponse.ok) {
-      const errorText = await submitResponse.text();
-      throw new Error(
-        `ComfyUI prompt submission failed: ${submitResponse.status} ${errorText}`
-      );
-    }
-    const submitResult2 = await submitResponse.json();
-    if (submitResult2.node_errors && Object.keys(submitResult2.node_errors).length > 0) {
-      throw new Error(
-        `ComfyUI workflow validation failed: ${JSON.stringify(submitResult2.node_errors)}`
-      );
-    }
-    const promptId = submitResult2.prompt_id;
-    await this.waitForCompletion(wsUrl, promptId, onProgress);
-    const historyResponse = await fetch(`${baseUrl}/history/${promptId}`, {
-      signal: AbortSignal.timeout(3e4)
-    });
-    if (!historyResponse.ok) {
-      throw new Error(
-        `Failed to fetch result history: ${historyResponse.status}`
-      );
-    }
-    const history = await historyResponse.json();
-    const promptHistory = history[promptId];
-    if (!promptHistory) {
-      throw new Error("No result found in ComfyUI history");
-    }
-    const outputNodeId = workflowConfig.outputNodeId;
-    const outputData = promptHistory.outputs[outputNodeId];
-    const outputFiles = outputData?.gifs || outputData?.images;
-    if (!outputFiles || outputFiles.length === 0) {
-      throw new Error("No output files found in ComfyUI result");
+  async discoverModels() {
+    const installPath = getProviderInstallPath(this.name) ?? null;
+    return discoverWorkflows(this.getBaseUrl(), installPath);
+  }
+  /**
+   * Generate an image using a ComfyUI workflow, with progress tracking.
+   */
+  async generateImage(_model, _prompt, options, onProgress) {
+    if (!options?.workflow) {
+      throw new Error("ComfyUI image generation requires a workflow");
     }
-    const outputFile = outputFiles[0];
-    const fileUrl = new URL(`${baseUrl}/view`);
-    fileUrl.searchParams.set("filename", outputFile.filename);
-    fileUrl.searchParams.set("subfolder", outputFile.subfolder || "");
-    fileUrl.searchParams.set("type", outputFile.type || "output");
-    const fileResponse = await fetch(fileUrl.toString(), {
-      signal: AbortSignal.timeout(6e4)
+    const result = await executeWorkflow({
+      baseUrl: this.getBaseUrl(),
+      workflow: options.workflow,
+      onProgress: onProgress ? (p) => onProgress({ step: p.step, totalSteps: p.totalSteps }) : void 0
     });
-    if (!fileResponse.ok) {
-      throw new Error(`Failed to download output file: ${fileResponse.status}`);
-    }
-    const fileBuffer = await fileResponse.arrayBuffer();
-    const videoBase64 = Buffer.from(fileBuffer).toString("base64");
-    const ext = path4.extname(outputFile.filename).toLowerCase();
-    const mimeType = ext === ".mp4" ? "video/mp4" : ext === ".webm" ? "video/webm" : ext === ".webp" ? "image/webp" : ext === ".gif" ? "image/gif" : "video/mp4";
-    const fps = options?.fps || defaults.fps;
-    const numFrames = options?.numFrames || defaults.numFrames;
     return {
-      videoBase64,
-      mimeType,
-      duration: numFrames / fps,
-      fps,
-      seed
+      imageBase64: result.dataBase64,
+      mimeType: result.mimeType
     };
   }
   /**
-   * Wait for a ComfyUI prompt to finish execution via WebSocket.
+   * Generate a video using a ComfyUI workflow.
    */
-  waitForCompletion(wsUrl, promptId, onProgress) {
-    return new Promise((resolve, reject) => {
-      const timeoutMs = 30 * 60 * 1e3;
-      let ws;
-      const timeout = setTimeout(() => {
-        try {
-          ws?.close();
-        } catch {
-        }
-        reject(new Error("Video generation timed out after 30 minutes"));
-      }, timeoutMs);
-      try {
-        ws = new WebSocket(wsUrl);
-      } catch (err) {
-        clearTimeout(timeout);
-        reject(
-          new Error(
-            `Failed to connect to ComfyUI WebSocket: ${err instanceof Error ? err.message : err}`
-          )
-        );
-        return;
-      }
-      ws.onmessage = (event) => {
-        try {
-          const message = JSON.parse(
-            typeof event.data === "string" ? event.data : ""
-          );
-          if (message.type === "progress") {
-            const data = message.data;
-            if (!data.prompt_id || data.prompt_id === promptId) {
-              onProgress?.({
-                step: data.value,
-                totalSteps: data.max,
-                currentNode: data.node
-              });
-            }
-          }
-          if (message.type === "execution_success") {
-            const data = message.data;
-            if (data.prompt_id === promptId) {
-              clearTimeout(timeout);
-              ws.close();
-              resolve();
-            }
-          }
-          if (message.type === "execution_error") {
-            const data = message.data;
-            if (data.prompt_id === promptId) {
-              clearTimeout(timeout);
-              ws.close();
-              reject(
-                new Error(
-                  `ComfyUI execution error${data.node_type ? ` in ${data.node_type}` : ""}: ${data.exception_message || "Unknown error"}`
-                )
-              );
-            }
-          }
-        } catch {
-        }
-      };
-      ws.onerror = () => {
-        clearTimeout(timeout);
-        reject(new Error("ComfyUI WebSocket error: connection failed"));
-      };
-      ws.onclose = (event) => {
-        if (!event.wasClean) {
-          clearTimeout(timeout);
-          reject(new Error("ComfyUI WebSocket connection closed unexpectedly"));
-        }
-      };
+  async generateVideo(_model, _prompt, options, onProgress) {
+    if (!options?.workflow) {
+      throw new Error("ComfyUI video generation requires a workflow");
+    }
+    const result = await executeWorkflow({
+      baseUrl: this.getBaseUrl(),
+      workflow: options.workflow,
+      onProgress: onProgress ? (p) => onProgress({
+        step: p.step,
+        totalSteps: p.totalSteps,
+        currentNode: p.currentNode
+      }) : void 0
     });
-  }
-  /**
-   * Get parameter schemas for video generation UI configuration.
-   */
-  async getParameterSchemas() {
-    return [
-      {
-        type: "number",
-        label: "Width",
-        variable: "width",
-        helpText: "Video width in pixels. Larger = better quality but bigger file.",
-        defaultValue: 512,
-        numberOptions: { min: 256, max: 1280, step: 64 }
-      },
-      {
-        type: "number",
-        label: "Height",
-        variable: "height",
-        helpText: "Video height in pixels. Larger = better quality but bigger file.",
-        defaultValue: 320,
-        numberOptions: { min: 256, max: 1280, step: 64 }
-      },
-      {
-        type: "number",
-        label: "Frames",
-        variable: "numFrames",
-        helpText: "Number of frames to generate. More frames = longer video but bigger file. Keep low to avoid upload limits.",
-        defaultValue: 41,
-        numberOptions: { min: 9, max: 97, step: 8 }
-      },
-      {
-        type: "number",
-        label: "FPS",
-        variable: "fps",
-        helpText: "Frames per second for the output video.",
-        defaultValue: 8,
-        numberOptions: { min: 4, max: 30, step: 1 }
-      },
-      {
-        type: "number",
-        label: "Steps",
-        variable: "steps",
-        helpText: "Number of denoising steps. More steps = higher quality but slower.",
-        defaultValue: 20,
-        numberOptions: { min: 10, max: 100, step: 1 }
-      },
-      {
-        type: "number",
-        label: "CFG Scale",
-        variable: "cfgScale",
-        helpText: "How strongly the video should follow the prompt. Higher = more literal.",
-        defaultValue: 7,
-        numberOptions: { min: 1, max: 20, step: 0.5 }
-      },
-      {
-        type: "number",
-        label: "Seed",
-        variable: "seed",
-        helpText: "A specific value used to guide randomness. Use -1 for random.",
-        defaultValue: -1,
-        numberOptions: { min: -1, max: 2147483647 }
-      },
-      {
-        type: "text",
-        label: "Negative Prompt",
-        variable: "negativePrompt",
-        helpText: "Things you don't want in the video",
-        placeholder: "worst quality, blurry, distorted"
-      }
-    ];
+    return {
+      videoBase64: result.dataBase64,
+      mimeType: result.mimeType
+    };
   }
 };
 var comfyui_default = new ComfyUIProvider();
@@ -1471,14 +1346,15 @@ async function discoverAllModelsWithParameters() {
   const modelsWithParams = await Promise.all(
     runningProviders.map(async (provider) => {
       const models = await provider.discoverModels();
+      const realModels = models.filter((m) => !m.statusHint);
       if (typeof provider.getParameterSchemas === "function") {
         const parameters = await provider.getParameterSchemas();
-        return models.map((model) => ({
+        return realModels.map((model) => ({
           ...model,
-          parameters
+          parameters: model.parameters ?? parameters
         }));
       }
-      return models;
+      return realModels;
     })
   );
   return modelsWithParams.flat();
@@ -1644,45 +1520,32 @@ var TunnelRunner = class {
     }
     const prompt = request.payload.prompt || "";
     const config2 = request.payload.config || {};
-    let result;
-    if (provider.generateImageWithProgress) {
-      result = await provider.generateImageWithProgress(
-        request.modelId,
-        prompt,
-        {
-          negativePrompt: config2.negativePrompt,
-          width: config2.width,
-          height: config2.height,
-          steps: config2.steps,
-          cfgScale: config2.cfgScale,
-          seed: config2.seed,
-          sampler: config2.sampler
-        },
-        async (progress) => {
-          await submitGenerationProgress(
-            request.id,
-            progress.step,
-            progress.totalSteps,
-            progress.preview
-          );
-          requestEvents.emitProgress({
-            id: request.id,
-            step: progress.step,
-            totalSteps: progress.totalSteps
-          });
-        }
-      );
-    } else {
-      result = await provider.generateImage(request.modelId, prompt, {
+    const result = await provider.generateImage(
+      request.modelId,
+      prompt,
+      {
         negativePrompt: config2.negativePrompt,
         width: config2.width,
         height: config2.height,
         steps: config2.steps,
         cfgScale: config2.cfgScale,
         seed: config2.seed,
-        sampler: config2.sampler
-      });
-    }
+        sampler: config2.sampler,
+        workflow: config2.workflow
+      },
+      async (progress) => {
+        await submitProgress(
+          request.id,
+          `Step ${progress.step}/${progress.totalSteps}`,
+          "log"
+        );
+        requestEvents.emitProgress({
+          id: request.id,
+          step: progress.step,
+          totalSteps: progress.totalSteps
+        });
+      }
+    );
     await submitResult(request.id, true, {
       imageBase64: result.imageBase64,
       mimeType: result.mimeType,
@@ -1713,13 +1576,14 @@ var TunnelRunner = class {
         fps: config2.fps,
         steps: config2.steps,
         cfgScale: config2.cfgScale,
-        seed: config2.seed
+        seed: config2.seed,
+        workflow: config2.workflow
       },
       async (progress) => {
-        await submitGenerationProgress(
+        await submitProgress(
           request.id,
-          progress.step,
-          progress.totalSteps
+          `Step ${progress.step}/${progress.totalSteps}`,
+          "log"
         );
         requestEvents.emitProgress({
           id: request.id,
@@ -1754,8 +1618,9 @@ export {
   setApiKey,
   getConfigPath,
   verifyApiKey,
-  registerLocalModel,
-  getRegisteredModels,
+  syncLocalModel,
+  updateLocalModel,
+  getSyncedModels,
   requestDeviceAuth,
   pollDeviceAuth,
   discoverAllModels,
@@ -1765,4 +1630,4 @@ export {
   requestEvents,
   TunnelRunner
 };
-//# sourceMappingURL=chunk-PTK4SJQK.js.map
+//# sourceMappingURL=chunk-V3RKCMCQ.js.map