Add a Scale to 1 MP toggle on image edit before Klein.

Co-authored-by: Cursor <cursoragent@cursor.com>

server/utils/imageChain.ts

@@ -1,6 +1,6 @@
 import { createJob, emitJob, type Job } from '~/server/utils/jobs'
 import { ensureComfyReady } from '~/server/utils/comfyLifecycle'
-import { getComfyHost, uploadImage, queuePrompt, purgeComfyArtifacts } from '~/server/utils/comfy'
+import { assertImageScaleToTotalPixelsNode, getComfyHost, uploadImage, queuePrompt, purgeComfyArtifacts } from '~/server/utils/comfy'
 import { withImageComfyHost, waitForImageEdit, downloadEditedImage } from '~/server/utils/imageComfy'
 import { buildEditWorkflow } from '~/server/utils/imageWorkflow'
 import { ensureComfyLoraNames } from '~/server/utils/loras'
@@ -21,6 +21,8 @@ export type EditRunParams = {
   cfg: number
   loraName?: string
   loraStack?: import('~/utils/loras').LoraStackItem[]
+  scaleToTotalPixels?: boolean
+  scaleMegapixels?: number
 }
 
 export async function runEdit(job: Job, params: EditRunParams) {
@@ -93,6 +95,7 @@ export async function runEdit(job: Job, params: EditRunParams) {
           progress: 12
         })
         await ensureComfyLoraNames('image')
+        if (params.scaleToTotalPixels) await assertImageScaleToTotalPixelsNode()
         const graph = buildEditWorkflow({
           prompt,
           negative: params.negative,
@@ -102,7 +105,9 @@ export async function runEdit(job: Job, params: EditRunParams) {
           seed,
           cfg: params.cfg,
           filenamePrefix: `aigen_edit_${job.id.slice(0, 8)}_p${index + 1}`,
-          loraStack: params.loraStack || params.loraName
+          loraStack: params.loraStack || params.loraName,
+          scaleToTotalPixels: params.scaleToTotalPixels,
+          scaleMegapixels: params.scaleMegapixels
         })
         const queued = await queuePrompt(graph, job.clientId)
         job.promptId = queued.prompt_id