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

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

server/api/edit.post.ts

@@ -1,8 +1,8 @@
 import { addStudioJob, kickStudioQueue, listStudioJobs } from '~/server/utils/studioQueue'
 import { parsePostedLoraStack, listStudioLoras, persistLoraFields } from '~/server/utils/loras'
-import { comfyConfigured } from '~/server/utils/comfy'
+import { assertImageScaleToTotalPixelsNode, comfyConfigured } from '~/server/utils/comfy'
 import { imageDimensions } from '~/server/utils/resolution'
-import { clampImageCfg, clampImageSteps } from '~/utils/generationPresets'
+import { clampImageCfg, clampImageScaleMegapixels, clampImageSteps, parseImageScaleToTotalPixels } from '~/utils/generationPresets'
 
 function parsePasses(raw: string | undefined) {
   if (!raw) return [] as { prompt: string }[]
@@ -66,6 +66,16 @@ export default defineEventHandler(async (event) => {
   const hideThumbnail = fields.hideThumbnail === 'true'
   const steps = clampImageSteps(fields.steps)
   const cfg = clampImageCfg(fields.cfg)
+  const scaleToTotalPixels = parseImageScaleToTotalPixels(fields.scaleToTotalPixels)
+  const scaleMegapixels = clampImageScaleMegapixels(fields.scaleMegapixels)
+  if (scaleToTotalPixels) {
+    try {
+      await assertImageScaleToTotalPixelsNode()
+    } catch (error) {
+      const status = Number((error as { statusCode?: number })?.statusCode || 0)
+      if (status === 400) throw error
+    }
+  }
   const seed = fields.seed && fields.seed !== 'random'
     ? Number(fields.seed)
     : Math.floor(Math.random() * 2_147_483_647)
@@ -130,6 +140,8 @@ export default defineEventHandler(async (event) => {
       passes,
       referenceStillId: savedRef?.id,
       referenceStillFilename: savedRef?.filename,
+      scaleToTotalPixels,
+      scaleMegapixels,
       ...persistLoraFields(loraStack)
     }
   })