@genfeedai/workflows 0.1.0

package/dist/index.mjs

@@ -0,0 +1,865 @@
+// src/comfyui/client.ts
+var DEFAULT_POLL_MS = 2e3;
+var DEFAULT_TIMEOUT_MS = 3e5;
+var ComfyUIClient = class {
+  constructor(baseUrl) {
+    this.baseUrl = baseUrl;
+  }
+  /**
+   * Queue a prompt for execution on ComfyUI.
+   */
+  async queuePrompt(prompt) {
+    const response = await fetch(`${this.baseUrl}/prompt`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ prompt })
+    });
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`ComfyUI /prompt failed (${response.status}): ${text}`);
+    }
+    return response.json();
+  }
+  /**
+   * Get history for a specific prompt execution.
+   */
+  async getHistory(promptId) {
+    const response = await fetch(`${this.baseUrl}/history/${promptId}`);
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`ComfyUI /history failed (${response.status}): ${text}`);
+    }
+    const data = await response.json();
+    return data[promptId];
+  }
+  /**
+   * Download an output file (image/video) from ComfyUI.
+   */
+  async getOutput(filename, subfolder) {
+    const params = new URLSearchParams({ filename, subfolder, type: "output" });
+    const response = await fetch(`${this.baseUrl}/view?${params.toString()}`);
+    if (!response.ok) {
+      throw new Error(`ComfyUI /view failed (${response.status})`);
+    }
+    const arrayBuffer = await response.arrayBuffer();
+    return Buffer.from(arrayBuffer);
+  }
+  /**
+   * Poll ComfyUI until the prompt completes or times out.
+   * Returns the history entry with outputs.
+   */
+  async waitForCompletion(promptId, opts) {
+    const pollMs = opts?.pollMs ?? DEFAULT_POLL_MS;
+    const timeoutMs = opts?.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+      const history = await this.getHistory(promptId);
+      if (history?.status?.completed) {
+        return history;
+      }
+      if (history?.status?.status_str === "error") {
+        throw new Error(
+          `ComfyUI prompt ${promptId} failed: ${JSON.stringify(history.status.messages)}`
+        );
+      }
+      await this.sleep(pollMs);
+    }
+    throw new Error(`ComfyUI prompt ${promptId} timed out after ${timeoutMs}ms`);
+  }
+  /**
+   * Health check — pings the ComfyUI instance.
+   */
+  async ping() {
+    try {
+      const response = await fetch(`${this.baseUrl}/system_stats`);
+      return response.ok;
+    } catch {
+      return false;
+    }
+  }
+  sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+  }
+};
+
+// src/comfyui/template-runner.ts
+var ComfyUITemplateRunner = class {
+  constructor(client) {
+    this.client = client;
+  }
+  /**
+   * Resolve template inputs into a concrete prompt.
+   * Injects user values into the template's node graph.
+   */
+  resolvePrompt(template, values) {
+    const prompt = JSON.parse(JSON.stringify(template.prompt));
+    for (const input of template.inputs) {
+      const value = values[input.key] ?? input.default;
+      if (input.required && value === void 0) {
+        throw new Error(`Missing required input: ${input.key}`);
+      }
+      if (value !== void 0) {
+        const node = prompt[input.nodeId];
+        if (!node) {
+          throw new Error(`Template references unknown node: ${input.nodeId}`);
+        }
+        node.inputs[input.field] = value;
+      }
+    }
+    return prompt;
+  }
+  /**
+   * Run a template with the given inputs and wait for completion.
+   */
+  async run(template, values) {
+    const prompt = this.resolvePrompt(template, values);
+    const { prompt_id } = await this.client.queuePrompt(prompt);
+    return this.client.waitForCompletion(prompt_id);
+  }
+};
+
+// src/comfyui/prompt-builder.ts
+function buildFluxDevPrompt(params) {
+  const {
+    prompt,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 20,
+    cfg = 1,
+    negativePrompt = ""
+  } = params;
+  return {
+    "1": {
+      class_type: "CheckpointLoaderSimple",
+      inputs: {
+        ckpt_name: "flux1-dev.safetensors"
+      }
+    },
+    "2": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["1", 1]
+      }
+    },
+    "3": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: negativePrompt,
+        clip: ["1", 1]
+      }
+    },
+    "4": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "5": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["1", 0],
+        positive: ["2", 0],
+        negative: ["3", 0],
+        latent_image: ["4", 0],
+        seed,
+        steps,
+        cfg,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "6": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["5", 0],
+        vae: ["1", 2]
+      }
+    },
+    "7": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["6", 0],
+        filename_prefix: "genfeed-flux-dev"
+      }
+    }
+  };
+}
+function buildPulidFluxPrompt(params) {
+  const {
+    prompt,
+    faceImage,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 20,
+    cfg = 1,
+    pulidStrength = 0.8
+  } = params;
+  return {
+    "1": {
+      class_type: "CheckpointLoaderSimple",
+      inputs: {
+        ckpt_name: "flux1-dev.safetensors"
+      }
+    },
+    "2": {
+      class_type: "PulidModelLoader",
+      inputs: {
+        pulid_file: "ip-adapter_pulid_sdxl_fp16.safetensors"
+      }
+    },
+    "3": {
+      class_type: "LoadImage",
+      inputs: {
+        image: faceImage
+      }
+    },
+    "4": {
+      class_type: "PulidInsightFaceLoader",
+      inputs: {
+        provider: "CPU"
+      }
+    },
+    "5": {
+      class_type: "ApplyPulid",
+      inputs: {
+        model: ["1", 0],
+        pulid: ["2", 0],
+        image: ["3", 0],
+        insightface: ["4", 0],
+        weight: pulidStrength
+      }
+    },
+    "6": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["1", 1]
+      }
+    },
+    "7": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: "",
+        clip: ["1", 1]
+      }
+    },
+    "8": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "9": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["5", 0],
+        positive: ["6", 0],
+        negative: ["7", 0],
+        latent_image: ["8", 0],
+        seed,
+        steps,
+        cfg,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "10": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["9", 0],
+        vae: ["1", 2]
+      }
+    },
+    "11": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["10", 0],
+        filename_prefix: "genfeed-pulid-flux"
+      }
+    }
+  };
+}
+function buildZImageTurboPrompt(params) {
+  const {
+    prompt,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 4
+  } = params;
+  return {
+    "1": {
+      class_type: "CheckpointLoaderSimple",
+      inputs: {
+        ckpt_name: "z-image-turbo.safetensors"
+      }
+    },
+    "2": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["1", 1]
+      }
+    },
+    "3": {
+      class_type: "CLIPTextEncode",
+      inputs: {
+        text: "",
+        clip: ["1", 1]
+      }
+    },
+    "4": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "5": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["1", 0],
+        positive: ["2", 0],
+        negative: ["3", 0],
+        latent_image: ["4", 0],
+        seed,
+        steps,
+        cfg: 1,
+        sampler_name: "euler_ancestral",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "6": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["5", 0],
+        vae: ["1", 2]
+      }
+    },
+    "7": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["6", 0],
+        filename_prefix: "genfeed-z-turbo"
+      }
+    }
+  };
+}
+function buildFlux2DevPrompt(params) {
+  const {
+    prompt,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 20,
+    guidance = 3.5
+  } = params;
+  return {
+    "1": {
+      class_type: "UNETLoader",
+      inputs: {
+        unet_name: "flux2_dev_fp8mixed.safetensors",
+        weight_dtype: "fp8_e4m3fn"
+      }
+    },
+    "2": {
+      class_type: "CLIPLoader",
+      inputs: {
+        clip_name: "mistral_3_small_flux2_fp4_mixed.safetensors",
+        type: "flux"
+      }
+    },
+    "3": {
+      class_type: "Flux2TextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["2", 0]
+      }
+    },
+    "4": {
+      class_type: "VAELoader",
+      inputs: {
+        vae_name: "flux2-vae.safetensors"
+      }
+    },
+    "5": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "6": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["1", 0],
+        positive: ["3", 0],
+        negative: ["3", 1],
+        latent_image: ["5", 0],
+        seed,
+        steps,
+        cfg: guidance,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "7": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["6", 0],
+        vae: ["4", 0]
+      }
+    },
+    "8": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["7", 0],
+        filename_prefix: "genfeed-flux2-dev"
+      }
+    }
+  };
+}
+function buildFlux2DevPulidPrompt(params) {
+  const {
+    prompt,
+    faceImage,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 20,
+    guidance = 3.5,
+    pulidStrength = 0.8
+  } = params;
+  return {
+    "1": {
+      class_type: "UNETLoader",
+      inputs: {
+        unet_name: "flux2_dev_fp8mixed.safetensors",
+        weight_dtype: "fp8_e4m3fn"
+      }
+    },
+    "2": {
+      class_type: "CLIPLoader",
+      inputs: {
+        clip_name: "mistral_3_small_flux2_fp4_mixed.safetensors",
+        type: "flux"
+      }
+    },
+    "3": {
+      class_type: "PulidModelLoader",
+      inputs: {
+        pulid_file: "ip-adapter_pulid_sdxl_fp16.safetensors"
+      }
+    },
+    "4": {
+      class_type: "LoadImage",
+      inputs: {
+        image: faceImage
+      }
+    },
+    "5": {
+      class_type: "PulidInsightFaceLoader",
+      inputs: {
+        provider: "CPU"
+      }
+    },
+    "6": {
+      class_type: "ApplyPulid",
+      inputs: {
+        model: ["1", 0],
+        pulid: ["3", 0],
+        image: ["4", 0],
+        insightface: ["5", 0],
+        weight: pulidStrength
+      }
+    },
+    "7": {
+      class_type: "Flux2TextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["2", 0]
+      }
+    },
+    "8": {
+      class_type: "VAELoader",
+      inputs: {
+        vae_name: "flux2-vae.safetensors"
+      }
+    },
+    "9": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "10": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["6", 0],
+        positive: ["7", 0],
+        negative: ["7", 1],
+        latent_image: ["9", 0],
+        seed,
+        steps,
+        cfg: guidance,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "11": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["10", 0],
+        vae: ["8", 0]
+      }
+    },
+    "12": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["11", 0],
+        filename_prefix: "genfeed-flux2-pulid"
+      }
+    }
+  };
+}
+function buildFlux2DevPulidLoraPrompt(params) {
+  const {
+    prompt,
+    faceImage,
+    loraPath,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 20,
+    guidance = 3.5,
+    pulidStrength = 0.8,
+    loraStrength = 0.8
+  } = params;
+  return {
+    "1": {
+      class_type: "UNETLoader",
+      inputs: {
+        unet_name: "flux2_dev_fp8mixed.safetensors",
+        weight_dtype: "fp8_e4m3fn"
+      }
+    },
+    "2": {
+      class_type: "CLIPLoader",
+      inputs: {
+        clip_name: "mistral_3_small_flux2_fp4_mixed.safetensors",
+        type: "flux"
+      }
+    },
+    "3": {
+      class_type: "LoraLoader",
+      inputs: {
+        model: ["1", 0],
+        clip: ["2", 0],
+        lora_name: loraPath,
+        strength_model: loraStrength,
+        strength_clip: loraStrength
+      }
+    },
+    "4": {
+      class_type: "PulidModelLoader",
+      inputs: {
+        pulid_file: "ip-adapter_pulid_sdxl_fp16.safetensors"
+      }
+    },
+    "5": {
+      class_type: "LoadImage",
+      inputs: {
+        image: faceImage
+      }
+    },
+    "6": {
+      class_type: "PulidInsightFaceLoader",
+      inputs: {
+        provider: "CPU"
+      }
+    },
+    "7": {
+      class_type: "ApplyPulid",
+      inputs: {
+        model: ["3", 0],
+        pulid: ["4", 0],
+        image: ["5", 0],
+        insightface: ["6", 0],
+        weight: pulidStrength
+      }
+    },
+    "8": {
+      class_type: "Flux2TextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["3", 1]
+      }
+    },
+    "9": {
+      class_type: "VAELoader",
+      inputs: {
+        vae_name: "flux2-vae.safetensors"
+      }
+    },
+    "10": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "11": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["7", 0],
+        positive: ["8", 0],
+        negative: ["8", 1],
+        latent_image: ["10", 0],
+        seed,
+        steps,
+        cfg: guidance,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "12": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["11", 0],
+        vae: ["9", 0]
+      }
+    },
+    "13": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["12", 0],
+        filename_prefix: "genfeed-flux2-pulid-lora"
+      }
+    }
+  };
+}
+function buildFlux2KleinPrompt(params) {
+  const {
+    prompt,
+    seed = Math.floor(Math.random() * 2 ** 32),
+    width = 1024,
+    height = 1024,
+    steps = 6
+  } = params;
+  return {
+    "1": {
+      class_type: "UNETLoader",
+      inputs: {
+        unet_name: "flux-2-klein-9b-fp8.safetensors",
+        weight_dtype: "fp8_e4m3fn"
+      }
+    },
+    "2": {
+      class_type: "CLIPLoader",
+      inputs: {
+        clip_name: "mistral_3_small_flux2_fp4_mixed.safetensors",
+        type: "flux"
+      }
+    },
+    "3": {
+      class_type: "Flux2TextEncode",
+      inputs: {
+        text: prompt,
+        clip: ["2", 0]
+      }
+    },
+    "4": {
+      class_type: "VAELoader",
+      inputs: {
+        vae_name: "flux2-vae.safetensors"
+      }
+    },
+    "5": {
+      class_type: "EmptyLatentImage",
+      inputs: {
+        width,
+        height,
+        batch_size: 1
+      }
+    },
+    "6": {
+      class_type: "KSampler",
+      inputs: {
+        model: ["1", 0],
+        positive: ["3", 0],
+        negative: ["3", 1],
+        latent_image: ["5", 0],
+        seed,
+        steps,
+        cfg: 3.5,
+        sampler_name: "euler",
+        scheduler: "normal",
+        denoise: 1
+      }
+    },
+    "7": {
+      class_type: "VAEDecode",
+      inputs: {
+        samples: ["6", 0],
+        vae: ["4", 0]
+      }
+    },
+    "8": {
+      class_type: "SaveImage",
+      inputs: {
+        images: ["7", 0],
+        filename_prefix: "genfeed-flux2-klein"
+      }
+    }
+  };
+}
+
+// src/index.ts
+import * as fs from "node:fs";
+import * as path from "node:path";
+var WORKFLOW_REGISTRY = {
+  "single-image": {
+    slug: "single-image",
+    title: "Single Image Generation",
+    description: "Generate an AI image from a source image (img2img) with enhanced styling",
+    category: "image-generation",
+    tags: ["image", "simple", "img2img", "enhancement"],
+    tier: "free",
+    icon: "\u{1F5BC}\uFE0F",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["image", "text"],
+    outputTypes: ["image"],
+    version: 1
+  },
+  "single-video": {
+    slug: "single-video",
+    title: "Single Video Generation",
+    description: "Generate an AI video from a source image (img2video) with motion effects",
+    category: "video-generation",
+    tags: ["video", "simple", "img2video", "animation"],
+    tier: "free",
+    icon: "\u{1F3AC}",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["image", "text"],
+    outputTypes: ["video"],
+    version: 1
+  },
+  "image-series": {
+    slug: "image-series",
+    title: "Image Series Generation",
+    description: "Generate a series of related images from a concept using LLM-powered prompt expansion",
+    category: "image-generation",
+    tags: ["image", "series", "llm", "batch", "automation"],
+    tier: "free",
+    icon: "\u{1F4F8}",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["text"],
+    outputTypes: ["image"],
+    version: 1
+  },
+  "image-to-video": {
+    slug: "image-to-video",
+    title: "Image to Video Transition",
+    description: "Create smooth interpolated video transitions between two images with easing effects",
+    category: "video-generation",
+    tags: ["video", "interpolation", "animation", "transition"],
+    tier: "free",
+    icon: "\u{1F39E}\uFE0F",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["image"],
+    outputTypes: ["video"],
+    version: 1
+  },
+  "full-pipeline": {
+    slug: "full-pipeline",
+    title: "Full Content Pipeline",
+    description: "Complete end-to-end workflow: concept development to final video production",
+    category: "full-pipeline",
+    tags: ["pipeline", "advanced", "complete", "automation", "production"],
+    tier: "free",
+    icon: "\u26A1",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["text"],
+    outputTypes: ["video"],
+    version: 1
+  },
+  "ugc-factory": {
+    slug: "ugc-factory",
+    title: "UGC Content Factory",
+    description: "UGC pipeline: script \u2192 voice \u2192 motion \u2192 lip sync \u2192 download",
+    category: "full-pipeline",
+    tags: ["ugc", "social", "automation", "marketing", "content"],
+    tier: "free",
+    icon: "\u{1F3ED}",
+    defaultModel: "nano-banana-pro",
+    inputTypes: ["text"],
+    outputTypes: ["video"],
+    version: 1
+  }
+};
+function getAllWorkflows() {
+  return Object.values(WORKFLOW_REGISTRY);
+}
+function getWorkflow(slug) {
+  return WORKFLOW_REGISTRY[slug];
+}
+function getWorkflowJson(slug) {
+  try {
+    const workflowPath = path.join(__dirname, "..", "workflows", `${slug}.json`);
+    const jsonContent = fs.readFileSync(workflowPath, "utf-8");
+    return JSON.parse(jsonContent);
+  } catch (_error) {
+    return void 0;
+  }
+}
+function getWorkflowIds() {
+  return Object.keys(WORKFLOW_REGISTRY);
+}
+function getWorkflowMetadata(id) {
+  return WORKFLOW_REGISTRY[id];
+}
+function getWorkflowsByCategory(category) {
+  return Object.values(WORKFLOW_REGISTRY).filter((w) => w.category === category);
+}
+function searchWorkflowsByTag(tag) {
+  return Object.values(WORKFLOW_REGISTRY).filter(
+    (w) => w.tags.some((t) => t.toLowerCase().includes(tag.toLowerCase()))
+  );
+}
+export {
+  ComfyUIClient,
+  ComfyUITemplateRunner,
+  WORKFLOW_REGISTRY,
+  buildFlux2DevPrompt,
+  buildFlux2DevPulidLoraPrompt,
+  buildFlux2DevPulidPrompt,
+  buildFlux2KleinPrompt,
+  buildFluxDevPrompt,
+  buildPulidFluxPrompt,
+  buildZImageTurboPrompt,
+  getAllWorkflows,
+  getWorkflow,
+  getWorkflowIds,
+  getWorkflowJson,
+  getWorkflowMetadata,
+  getWorkflowsByCategory,
+  searchWorkflowsByTag
+};