@goodandready/dsh-image-gen 0.8.5 → 0.8.7

package/lib/client.js

@@ -99,7 +99,7 @@ window.__ModuleLoader__.load({
     function PluginSettingsCard(props) {
       // Карточка живёт в собственном разделе настроек и на странице одна, так
       // что открыта сразу: свёрнутая выглядела бы как пустая страница.
-      const [open, setOpen] = react.useState(true)
+      const [open, setOpen] = react.useState(false)
       const { state } = props
       // Раньше здесь было `return null`, и раздел оказывался пустым, если хост
       // ещё не объявил настройки. Пустая страница ничего не объясняет — лучше
@@ -407,17 +407,27 @@ window.__ModuleLoader__.load({
 
     // ------------------------------------------------------------- fields
     const IMAGE_SIZES = ['square_hd', 'square', 'portrait_4_3', 'portrait_16_9', 'landscape_4_3', 'landscape_16_9']
+    // Перевод именованного размера в пиксели для подсказки в карточке
+    // (значения совпадают с SIZE_PIXELS в хостовой половине).
+    const SIZE_PIXELS = {
+      square_hd: '1024x1024',
+      square: '512x512',
+      portrait_4_3: '768x1024',
+      portrait_16_9: '576x1024',
+      landscape_4_3: '1024x768',
+      landscape_16_9: '1024x576',
+    }
     const OUTPUT_FORMATS = ['png', 'jpeg', 'webp']
     const DELIVERY_MODES = ['link', 'image']
-    const PROVIDERS = ['fal', 'custom', 'codex', 'grok']
+    const PROVIDERS = ['fal', 'custom', 'codex', 'grok', 'local']
 
     /** Field specs in display order. */
     const FIELDS = [
-      { field: 'provider', spec: selectField('provider', PROVIDERS), kind: 'select', options: PROVIDERS, labelKey: 'f.provider', hintKey: 'f.providerHint' },
+      { field: 'provider', spec: selectField('provider', PROVIDERS), kind: 'select', options: PROVIDERS, labelKey: 'f.provider', hintKey: 'f.providerHint', providerNeeds: { fal: 'ключ FAL', custom: 'адрес + ключ', codex: 'подписка ChatGPT', grok: 'подписка Grok', local: 'адрес сервера' } },
       { field: 'model', spec: textField('model'), kind: 'text', labelKey: 'f.model', hintKey: 'f.modelHint', placeholderKey: 'p.model' },
       { field: 'apiKeyEnv', spec: textField('apiKeyEnv'), kind: 'text', labelKey: 'f.apiKeyEnv', hintKey: 'f.apiKeyEnvHint', placeholderKey: 'p.apiKeyEnv' },
       { field: 'baseURL', spec: textField('baseURL'), kind: 'text', labelKey: 'f.baseURL', hintKey: 'f.baseURLHint', placeholderKey: 'p.baseURL' },
-      { field: 'defaultSize', spec: selectField('defaultSize', IMAGE_SIZES), kind: 'select', options: IMAGE_SIZES, labelKey: 'f.defaultSize', hintKey: 'f.defaultSizeHint' },
+      { field: 'defaultSize', spec: selectField('defaultSize', IMAGE_SIZES), kind: 'select', options: IMAGE_SIZES, labelKey: 'f.defaultSize', hintKey: 'f.defaultSizeHint', sizePixels: SIZE_PIXELS },
       { field: 'defaultFormat', spec: selectField('defaultFormat', OUTPUT_FORMATS), kind: 'select', options: OUTPUT_FORMATS, labelKey: 'f.defaultFormat', hintKey: 'f.defaultFormatHint' },
       { field: 'pollIntervalMs', spec: numberField('pollIntervalMs'), kind: 'number', labelKey: 'f.pollIntervalMs', hintKey: 'f.pollIntervalMsHint', placeholderKey: 'p.pollIntervalMs' },
       { field: 'timeoutMs', spec: numberField('timeoutMs'), kind: 'number', labelKey: 'f.timeoutMs', hintKey: 'f.timeoutMsHint', placeholderKey: 'p.timeoutMs' },
@@ -446,7 +456,10 @@ window.__ModuleLoader__.load({
         onReset,
       }
       if (entry.kind === 'select') {
-        return react_jsx_runtime.jsx(SelectField, { ...base, options: entry.options, inheritLabel: t('f.inherit') })
+        const px = entry.sizePixels && entry.sizePixels[state.text] ? ` — ${entry.sizePixels[state.text]}` : ''
+        const need = entry.providerNeeds && entry.providerNeeds[state.text] ? ` (нужно: ${entry.providerNeeds[state.text]})` : ''
+        const hint = base.hint + px + need
+        return react_jsx_runtime.jsx(SelectField, { ...base, hint, options: entry.options, inheritLabel: t('f.inherit') })
       }
       return react_jsx_runtime.jsx(ValueField, { ...base, numeric: entry.kind === 'number', placeholder: entry.placeholderKey ? t(entry.placeholderKey) : '' })
     }
@@ -511,7 +524,7 @@ window.__ModuleLoader__.load({
     /** Settings namespace the card edits (the Host plugin registers it). */
     const SETTINGS_NS = 'dsh-image-gen'
     /** Required client services. */
-    const inject = ['slots', 'settingsScope', 'locale']
+    const inject = ['slots', 'settingsScope', 'locale', 'sessions']
 
     const en = {
       'settings.navLabel': 'Image generation',
@@ -636,7 +649,7 @@ window.__ModuleLoader__.load({
     // Карточки инструментов не рисуют image-блоки (это умеют только сообщения
     // ассистента), поэтому сгенерированная картинка показывается здесь —
     // keyed-запись слота tool.call.toolview по имени инструмента.
-    const CARD_CSS = '.fal-card{display:flex;flex-direction:column;gap:8px;padding:10px 12px;border:1px solid var(--dsw-alias-border-l1);border-radius:10px;background:var(--dsw-alias-bg-layer-1);max-width:520px}'
+    const CARD_CSS = '.fal-card{display:flex;flex-direction:column;gap:8px;padding:10px 12px;border:1px solid var(--dsw-alias-border-l1);border-radius:10px;background:var(--dsw-alias-bg-layer-1);max-width:520px}' + '.fal-repeat{appearance:none;font:inherit;cursor:pointer;border:1px solid var(--dsw-alias-border-l2);color:var(--dsw-alias-label-primary);background:var(--dsw-alias-bg-layer-3);border-radius:8px;padding:5px 14px;font-size:13px}' + '.fal-repeat:hover:not(:disabled){border-color:var(--dsw-alias-label-dimmed)}' + '.fal-repeat:disabled{opacity:.5;cursor:default}'
       + '.fal-card .fal-head{display:flex;align-items:center;gap:8px;font-size:12px;color:var(--dsw-alias-label-secondary)}'
       + '.fal-card img{max-width:100%;border-radius:8px;display:block}'
       + '.fal-card .fal-prompt{font-size:12px;color:var(--dsw-alias-label-secondary);overflow-wrap:anywhere}'
@@ -697,8 +710,42 @@ window.__ModuleLoader__.load({
       const head = react.createElement('div', { className: 'fal-head' },
         running ? 'Генерирую изображение…' : failed ? 'Генерация не удалась' : 'Изображение')
 
+      // «Повторить генерацию»: отправляет в чат новое сообщение с теми же
+      // промптом и параметрами, извлечёнными из исходного вызова инструмента.
+      // Дальше модель сама решает, как сгенерировать, — это просто повторный
+      // запрос, а не вызов инструмента из карточки.
+      let args = {}
+      try {
+        const raw = (block && (block.call ? block.call.argsRaw : block.argsRaw)) || ''
+        if (raw) args = JSON.parse(raw)
+      } catch (e) { /* ignore */ }
+      const repeatText = (() => {
+        const parts = [String(args.prompt || '').trim()]
+        const size = args.image_size || args.size
+        const seed = args.seed
+        if (size) parts.push(`Размер: ${size}`)
+        if (seed !== undefined) parts.push(`Сид: ${seed}`)
+        return parts.filter(Boolean).join(', ')
+      })()
+      const repeat = async () => {
+        try {
+          const sessions = props.sessions
+          const current = sessions && sessions.list && sessions.list.current
+          const binding = current && sessions.binding(current)
+          const session = binding && binding.session
+          if (!session || !session.prompt) {
+            return
+          }
+          const text = repeatText || String(args.prompt || '')
+          await session.prompt([{ type: 'text', text }], 'queue')
+        } catch (e) { /* контур сам покажет ошибку отправки */ }
+      }
+
       const body = []
       if (prompt) body.push(react.createElement('div', { className: 'fal-prompt', key: 'p' }, prompt))
+      if (args.originalPrompt && args.originalPrompt !== prompt) {
+        body.push(react.createElement('div', { className: 'fal-meta', key: 'orig' }, `Исходный запрос: ${args.originalPrompt}`))
+      }
       if (failed) {
         body.push(react.createElement('div', { className: 'fal-err', key: 'e' }, parsed.text || 'неизвестная ошибка'))
       } else if (!parsed.attachment && parsed.url) {
@@ -729,6 +776,44 @@ window.__ModuleLoader__.load({
         body.push(react.createElement('div', { className: 'fal-meta', key: 'm' }, parsed.text))
       }
 
+      // Ссылка для скачивания: та же, что у картинки (наш роут или провайдерский url).
+      const downloadUrl = parsed.attachment
+        ? '/dsh-image-gen/image?id=' + encodeURIComponent(parsed.attachment.attachmentId ?? parsed.attachment.id ?? '')
+          + '&mt=' + encodeURIComponent(parsed.attachment.mediaType || 'image/png')
+          + '&b=' + encodeURIComponent(String(parsed.attachment.bytes ?? 0))
+          + '&w=' + encodeURIComponent(String(parsed.attachment.width ?? 0))
+          + '&h=' + encodeURIComponent(String(parsed.attachment.height ?? 0))
+        : parsed.url
+
+      // Диагностика: метаданные генерации (prompt/seed/size/format), чтобы было
+      // понятно, по каким параметрам получилась именно эта картинка.
+      const metaParts = []
+      if (args.image_size || args.size) metaParts.push(`size ${args.image_size || args.size}`)
+      if (args.output_format || args.format) metaParts.push(`format ${args.output_format || args.format}`)
+      if (args.seed !== undefined) metaParts.push(`seed ${args.seed}`)
+      const meta = metaParts.join(' · ')
+
+      // Кнопка повтора — только когда генерация завершена (не в процессе и не ошибка).
+      if (!running && !failed && meta) {
+        body.push(react.createElement('div', { key: 'meta', className: 'fal-meta' }, meta))
+      }
+      if (!running && !failed && repeatText) {
+        body.push(react.createElement('button', {
+          key: 'repeat',
+          type: 'button',
+          className: 'fal-repeat',
+          onClick: repeat,
+        }, 'Повторить генерацию'))
+      }
+      if (!running && !failed && downloadUrl) {
+        body.push(react.createElement('a', {
+          key: 'download',
+          className: 'fal-repeat',
+          href: downloadUrl,
+          download: 'generated-image.png',
+        }, 'Скачать'))
+      }
+
       return react.createElement('div', { className: 'fal-card' }, head, body)
     }
 
@@ -810,7 +895,7 @@ window.__ModuleLoader__.load({
       }
       ctx.slots.inject('tool.call.toolview', () =>
         ctx.slots.register(
-          { name: 'tool.call.toolview', key: 'generate_image', locale: NS },
+          { name: 'tool.call.toolview', key: 'generate_image', locale: NS, inject: () => ({ sessions: ctx.sessions }) },
           FalImageCard,
         ),
       )

package/lib/index.js

@@ -15,7 +15,7 @@
 import z from '@deepseek-ai/schemastery'
 import { defineTool } from '@deepseek-ai/dsh-tools'
 import { credentialRef } from '@deepseek-ai/dsh-credentials'
-import { mkdir, writeFile } from 'node:fs/promises'
+import { mkdir, readFile, writeFile } from 'node:fs/promises'
 import path from 'node:path'
 import {
   IMAGE_SIZES,
@@ -24,6 +24,8 @@ import {
   buildSidecar,
   makeProviders,
   normalizeCount,
+  tryGenerate,
+  fallbackOrder,
   normalizeMediaType,
 } from './providers.js'
 
@@ -38,7 +40,7 @@ const NS = 'dsh-image-gen'
 // настройки лежат под старым именем. Оно читается как запасное и переносится
 // под новое имя один раз — молча, при первом запуске после обновления.
 const LEGACY_NS = 'dsh-fal-image-gen'
-export const inject = ['tools', 'attachments', 'credentials', 'webServer', 'settings']
+export const inject = ['tools', 'attachments', 'credentials', 'webServer', 'settings', 'llm']
 
 export const Config = z.object({
   provider: z
@@ -106,13 +108,62 @@ export const Config = z.object({
     .description('provider=custom: fixed size sent to the API, e.g. 1024x1024. '
       + 'Empty means the named size is translated automatically — set this only for an API picky about sizes.')
     .default(''),
+  localKind: z
+    .string()
+    .description('provider=local: which local API to use — "comfyui" or "a1111".')
+    .default('comfyui'),
+  localBaseURL: z
+    .string()
+    .description('provider=local: server address, e.g. http://127.0.0.1:8188 (ComfyUI) or http://127.0.0.1:7860 (A1111).')
+    .default(''),
+  localModel: z
+    .string()
+    .description('provider=local: model id (A1111) or workflow/schema name (ComfyUI). Empty means the server default.')
+    .default(''),
+  localSteps: z
+    .number()
+    .description('provider=local: sampling steps.')
+    .default(20),
+  localCfg: z
+    .number()
+    .description('provider=local: CFG scale.')
+    .default(7),
   outputDir: z
     .string()
     .description('Where generated images are saved. A relative path resolves against the session working directory; an absolute path is used as given.')
     .default('generated/images'),
+  historyLimit: z
+    .number()
+    .description('How many recent generations to keep in the in-memory history list.')
+    .default(50),
+  enhancePrompt: z
+    .boolean()
+    .description('Expand a short prompt into a detailed one through the chat model before generating. Off by default.')
+    .default(false),
+  enhanceModel: z
+    .string()
+    .description('Model used to enhance the prompt. Empty means the same model that leads the conversation.')
+    .default(''),
+  enhanceBelowChars: z
+    .number()
+    .description('Only enhance prompts shorter than this many characters.')
+    .default(200),
 })
 
 /** Keep a file stem safe for the filesystem. */
+/** Прочитать исходное изображение для правки: путь к файлу или attachment id. */
+export async function resolveSource(ctx, exec, ref) {
+  if (!ref) return undefined
+  const sessionCwd = exec.agent?.session?.header?.cwd
+  if (ref.startsWith('sha256:')) {
+    const stored = await ctx.attachments.readImage({ attachmentId: ref, mediaType: 'image/png', bytes: 0, width: 0, height: 0 })
+    return { bytes: Buffer.from(stored.data), mediaType: stored.ref?.mediaType || 'image/png' }
+  }
+  const p = path.resolve(sessionCwd || process.cwd(), ref)
+  const bytes = await readFile(p)
+  return { bytes, mediaType: 'image/png' }
+}
+
 export function slugify(input) {
   const stem = String(input ?? '')
     .toLowerCase()
@@ -163,6 +214,53 @@ function migrateLegacySettings(sctx, scope) {
   }
 }
 
+/** In-memory история генераций (пополняется в execute, где доступен cwd). */
+const history = []
+
+/** Отфильтровать записи, чьи файлы ещё существуют; новые первыми. */
+export function filterHistory(entries, exists) {
+  return entries.filter((e) => exists(e.path)).slice(0, 50)
+}
+
+/** Собрать текст из итератора llm.stream (text-delta / block-end). */
+export async function collectText(iterable) {
+  let out = ''
+  let sawDelta = false
+  for await (const chunk of iterable) {
+    if (chunk && chunk.type === 'text-delta' && typeof chunk.text === 'string') {
+      out += chunk.text
+      sawDelta = true
+    } else if (
+      !sawDelta && chunk && chunk.type === 'block-end'
+      && chunk.block && chunk.block.type === 'text' && typeof chunk.block.text === 'string'
+    ) {
+      out += chunk.block.text
+    }
+  }
+  return out.trim()
+}
+
+/** Развернуть короткий промпт через чат-модель; при ошибке вернуть исходный. */
+export async function enhancePrompt(ctx, cfg, prompt, signal) {
+  if (!cfg.enhancePrompt) return { prompt, enhanced: false }
+  if (String(prompt).length >= (cfg.enhanceBelowChars || 200)) return { prompt, enhanced: false }
+  try {
+    const chunks = ctx.llm.stream({
+      ...(signal ? { signal } : {}),
+      ...(cfg.enhanceModel ? { model: cfg.enhanceModel } : {}),
+      messages: [
+        { role: 'system', content: 'You expand a short image-generation prompt into a detailed one. Reply with only the expanded prompt, no commentary.' },
+        { role: 'user', content: prompt },
+      ],
+    })
+    const text = await collectText(chunks)
+    if (!text) return { prompt, enhanced: false }
+    return { prompt: text, enhanced: true }
+  } catch (e) {
+    return { prompt, enhanced: false }
+  }
+}
+
 export function apply(ctx, config) {
 
   // The Web card edits this namespace; without registering it the card binds to
@@ -240,6 +338,22 @@ export function apply(ctx, config) {
     }), `dsh-image-gen: image route ${path}`)
   }
 
+  // История генераций: in-memory список, отфильтрованный по существованию файлов.
+  ctx.effect(() => ctx.webServer.register({
+    kind: 'exact',
+    path: '/dsh-image-gen/history',
+    handler: async (req, res) => {
+      if (req.method !== 'GET') {
+        res.writeHead(405, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({ error: 'GET only' }))
+        return
+      }
+      const exists = (p) => { try { return require('node:fs').existsSync(p) } catch { return false } }
+      res.writeHead(200, { 'Content-Type': 'application/json' })
+      res.end(JSON.stringify(filterHistory(history, exists)))
+    },
+  }), 'dsh-image-gen: history route')
+
   ctx.tools.register(
     defineTool({
       name: 'generate_image',
@@ -276,6 +390,26 @@ export function apply(ctx, config) {
           type: 'integer',
           description: 'Number of variations to generate in one call, 1-4 (default 1). Cost scales with count.',
         },
+        negative_prompt: {
+          type: 'string',
+          description: 'Optional text describing what NOT to draw; sent to providers that support it (ignored otherwise).',
+        },
+        guidance_scale: {
+          type: 'number',
+          description: 'Optional prompt adherence (e.g. 1-20); sent to providers that support it (ignored otherwise).',
+        },
+        source_image: {
+          type: 'string',
+          description: 'Optional path to an image file or an attachment id (sha256:...) to edit instead of drawing from scratch. Only providers that support image editing accept it; others refuse with a clear reason.',
+        },
+        mask: {
+          type: 'string',
+          description: 'Optional path or attachment id of a mask image (same size as source) to restrict the edit to a region. Only used when source_image is set and the provider supports masks.',
+        },
+        strength: {
+          type: 'number',
+          description: 'Optional edit strength (0-1) for providers that accept it; higher keeps more of the original.',
+        },
       },
       output: {
         schema: {
@@ -351,21 +485,25 @@ export function apply(ctx, config) {
         let subscriptionImages
         try { subscriptionImages = ctx.get && ctx.get('subscriptionImages') } catch (noService) { subscriptionImages = undefined }
 
+        const source = args.source_image ? await resolveSource(ctx, exec, args.source_image) : undefined
+        const mask = args.mask ? await resolveSource(ctx, exec, args.mask) : undefined
+        const enhanced = await enhancePrompt(ctx, cfg, args.prompt, exec.signal)
+        const effectivePrompt = enhanced.prompt
         const providers = makeProviders(
           { fetchImpl: fetch, resolveKey: (ref) => resolveApiKey(ctx, ref), cfg, subscriptionImages },
-          { prompt: args.prompt, size, format, seed: args.seed, signal: exec.signal },
+          { prompt: effectivePrompt, size, format, seed: args.seed, signal: exec.signal, negativePrompt: args.negative_prompt, guidanceScale: args.guidance_scale, source, mask, strength: args.strength },
         )
         // Подписочные провайдеры (codex/grok) отдают {ok:false, reason} вместо исключения:
         // отказ должен дойти до модели текстом. Без проверки execute шёл дальше с пустыми
         // байтами, и пользователь получал битую карточку вместо внятного отказа.
         const guard = (generated) => { if (generated && generated.ok === false) throw new Error(generated.reason) }
-        const one = async (jobSeed) => {
-          const gen = await providers[provider](jobSeed)
+        const one = async (jobSeed, providerKey) => {
+          const gen = await providers[providerKey](jobSeed)
           guard(gen)
           const bytes = gen.bytes
           const mediaType = gen.mediaType
           const extension = mediaType === 'image/jpeg' ? 'jpg' : mediaType === 'image/webp' ? 'webp' : 'png'
-          const stem = `${slugify(args.output_name || args.prompt)}-${Date.now().toString(36)}-${jobSeed}`
+          const stem = `${slugify(args.output_name || effectivePrompt)}-${Date.now().toString(36)}-${jobSeed}`
           const name = `${stem}.${extension}`
 
           const attachment = await ctx.attachments.saveImage({
@@ -391,7 +529,7 @@ export function apply(ctx, config) {
           await writeFile(
             path.join(outDir, `${stem}.json`),
             JSON.stringify(buildSidecar({
-              prompt: args.prompt,
+              prompt: effectivePrompt,
               size,
               format,
               seed: gen.seed,
@@ -402,16 +540,31 @@ export function apply(ctx, config) {
               mediaType,
               attachmentId: attachment.attachmentId,
               url: deliverAs === 'image' && gen.sourceUrl ? gen.sourceUrl : localUrl,
+              cost: gen.cost,
             }), null, 2),
           )
 
+          history.unshift({
+            path: filePath,
+            prompt: effectivePrompt,
+            provider,
+            size,
+            format,
+            seed: gen.seed,
+            createdAt: new Date().toISOString(),
+            attachmentId: attachment.attachmentId,
+          })
+          if (history.length > (cfg.historyLimit || 50)) history.length = cfg.historyLimit || 50
+
           return {
             path: filePath,
             url: deliverAs === 'image' && gen.sourceUrl ? gen.sourceUrl : localUrl,
             width: gen.width || attachment.width,
             height: gen.height || attachment.height,
             seed: gen.seed,
-            prompt: args.prompt,
+            prompt: effectivePrompt,
+            originalPrompt: enhanced.enhanced ? args.prompt : undefined,
+            cost: gen.cost,
             format: mediaType.replace('image/', ''),
             attachment: {
               attachmentId: attachment.attachmentId,
@@ -426,9 +579,13 @@ export function apply(ctx, config) {
 
         const count = normalizeCount(args.count)
         const seedBase = args.seed ?? Math.floor(Math.random() * 100000)
+        // Fallback-цепочка: пробуем текущий провайдер, при отказе — следующий
+        // по порядку (fal → custom → codex → grok), собирая причины отказов.
+        const order = fallbackOrder(provider)
+        const generators = Object.fromEntries(PROVIDER_KEYS.map((k) => [k, (s) => one(s, k)]))
         const images = []
         for (let i = 0; i < count; i += 1) {
-          images.push(await one(seedBase + i))
+          images.push(await tryGenerate(generators, order, seedBase + i))
         }
         const first = images[0]
         return { ...first, images }

package/lib/providers.js

@@ -7,9 +7,33 @@
 // Сеть приходит параметром (fetchImpl), ключ — через resolveKey, поэтому оба
 // провайдера проверяются юнит-тестами без единого реального запроса.
 
-export const PROVIDER_KEYS = ['fal', 'custom', 'codex', 'grok']
+export const PROVIDER_KEYS = ['fal', 'custom', 'codex', 'grok', 'local']
 
 /** Привести count из аргумента инструмента к диапазону 1..4. */
+/** Порядок провайдеров для fallback: основной первым, остальные по PROVIDER_KEYS. */
+export function fallbackOrder(primary) {
+  return [primary, ...PROVIDER_KEYS.filter((k) => k !== primary)]
+}
+
+/**
+ * Перебирает генераторов по порядку, возвращает первый успешный результат.
+ * При отказе всех бросает с перечислением причин.
+ * @param generators - массив функций (key, seed) => Promise<generated>.
+ * @param order - порядок ключей, длина = generators.length.
+ */
+export async function tryGenerate(generators, order, seed) {
+  const refusals = []
+  for (const key of order) {
+    try {
+      const produced = await generators[key](seed)
+      return produced
+    } catch (e) {
+      refusals.push(`${key}: ${e.message || String(e)}`)
+    }
+  }
+  throw new Error(`Image generation failed on all providers. ${refusals.join('; ')}`)
+}
+
 export function normalizeCount(count) {
   const n = Math.floor(Number(count))
   if (!Number.isFinite(n)) return 1
@@ -75,9 +99,9 @@ export function normalizeMediaType(contentType, fallbackFormat) {
 /** Метаданные одной генерации для sidecar-файла рядом с картинкой. */
 export function buildSidecar({
   prompt, size, format, seed, provider, deliverAs,
-  width, height, mediaType, attachmentId, url, createdAt = new Date().toISOString(),
+  width, height, mediaType, attachmentId, url, cost, createdAt = new Date().toISOString(),
 }) {
-  return { prompt, size, format, seed, provider, deliverAs, width, height, mediaType, attachmentId, url, createdAt }
+  return { prompt, size, format, seed, provider, deliverAs, width, height, mediaType, attachmentId, url, cost, createdAt }
 }
 
 /** Submit a generation job to the FAL queue. */
@@ -133,14 +157,36 @@ export async function pollStatus(fetchImpl, statusUrl, key, signal, pollInterval
  *   {bytes, mediaType, width, height, seed, sourceUrl}. Нулевые width/height
  *   означают «спроси у службы вложений»: она всё равно измеряет байты сама.
  */
+
+/** Именованный размер -> [width, height] для локальных API. */
+export function sizeToPixels(size) {
+  const px = SIZE_PIXELS[size]
+  if (!px) return [1024, 1024]
+  const [w, h] = px.split('x').map(Number)
+  return [w, h]
+}
+
+/** Собрать multipart-тело для /images/edits (OpenAI-совместимый edit). */
+export function buildEditForm({ source, mask, prompt, size, strength }) {
+  const form = new FormData()
+  form.append('image', new Blob([source.bytes], { type: source.mediaType || 'image/png' }), 'source.png')
+  if (mask) form.append('mask', new Blob([mask.bytes], { type: mask.mediaType || 'image/png' }), 'mask.png')
+  form.append('prompt', prompt)
+  if (size) form.append('size', size)
+  if (strength !== undefined) form.append('strength', String(strength))
+  return form
+}
+
 export function makeProviders(deps, job) {
   const { fetchImpl, resolveKey, cfg } = deps
-  const { prompt, size, format, seed, signal } = job
+  const { prompt, size, format, seed, signal, negativePrompt, guidanceScale, source, mask, strength } = job
 
   async function fal(seedArg = seed) {
     const key = await resolveKey(cfg.apiKeyEnv)
     const body = { prompt, image_size: size, num_images: 1 }
     if (seedArg !== undefined) body.seed = seedArg
+    if (negativePrompt !== undefined) body.negative_prompt = negativePrompt
+    if (guidanceScale !== undefined) body.guidance_scale = guidanceScale
     if (format !== 'png') body.output_format = format
 
     const submit = await submitJob(fetchImpl, cfg.baseURL, cfg.model, key, body, signal)
@@ -167,6 +213,7 @@ export function makeProviders(deps, job) {
       height: image.height ?? 0,
       seed: result.seed ?? seedArg ?? 0,
       sourceUrl: image.url,
+      cost: result.cost,
     }
   }
 
@@ -184,15 +231,20 @@ export function makeProviders(deps, job) {
 
     // response_format не отправляем: новые модели OpenAI его отвергают, а ответ
     // всё равно приходит либо base64, либо ссылкой — принимаем оба.
-    const res = await fetchImpl(`${base}/images/generations`, {
+    const endpoint = source ? `${base}/images/edits` : `${base}/images/generations`
+    const res = await fetchImpl(endpoint, {
       method: 'POST',
       headers,
-      body: JSON.stringify({
-        model: cfg.customModel,
-        prompt,
-        n: 1,
-        size: cfg.customSize || SIZE_PIXELS[size] || size,
-      }),
+      body: source
+        ? buildEditForm({ source, mask, prompt, size: cfg.customSize || SIZE_PIXELS[size] || size, strength })
+        : JSON.stringify({
+            model: cfg.customModel,
+            prompt,
+            n: 1,
+            size: cfg.customSize || SIZE_PIXELS[size] || size,
+            ...(negativePrompt !== undefined ? { negative_prompt: negativePrompt } : {}),
+            ...(guidanceScale !== undefined ? { guidance_scale: guidanceScale } : {}),
+          }),
       signal,
     })
     const data = await res.json().catch(() => ({}))
@@ -240,6 +292,9 @@ export function makeProviders(deps, job) {
   // отдаёт готовую картинку, а не ключ доступа.
   function subscription(provider) {
     return async function generate(seedArg = seed) {
+      if (source) {
+        return { ok: false, provider, reason: `${provider}: не умеет править изображения — используйте fal, custom или local` }
+      }
       const images = deps.subscriptionImages
       if (!images || typeof images.generate !== 'function') {
         return {
@@ -277,5 +332,102 @@ export function makeProviders(deps, job) {
     }
   }
 
-  return { fal, custom, codex: subscription('codex'), grok: subscription('grok') }
+  // Локальная генерация: ComfyUI (очередь + опрос) или Automatic1111 (txt2img).
+  async function local(seedArg = seed) {
+    const base = String(cfg.localBaseURL || '').replace(/\/+$/, '')
+    if (!base) throw new Error('Local image provider: server address is not configured (Settings → Image generation)')
+    const [width, height] = sizeToPixels(size)
+    const kind = cfg.localKind === 'a1111' ? 'a1111' : 'comfyui'
+
+    if (kind === 'a1111') {
+      const body = {
+        prompt,
+        negative_prompt: negativePrompt,
+        width,
+        height,
+        steps: cfg.localSteps ?? 20,
+        cfg_scale: cfg.localCfg ?? 7,
+        seed: seedArg ?? -1,
+      }
+      if (cfg.localModel) body.override_settings = { sd_model_checkpoint: cfg.localModel }
+      const res = await fetchImpl(`${base}/sdapi/v1/txt2img`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify(body),
+        signal,
+      })
+      if (!res.ok) {
+        throw new Error(`Local A1111 failed (HTTP ${res.status}): ${String(await res.text().catch(() => '')).slice(0, 300)}`)
+      }
+      const data = await res.json().catch(() => ({}))
+      const b64 = data.images && data.images[0]
+      if (!b64) throw new Error('Local A1111 returned no images')
+      return {
+        bytes: Buffer.from(b64, 'base64'),
+        mediaType: normalizeMediaType('image/png', format),
+        width,
+        height,
+        seed: seedArg ?? 0,
+        sourceUrl: '',
+      }
+    }
+
+    // ComfyUI: submit через /prompt, опрос /history/{prompt_id} до готовности.
+    const promptId = `dsh-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`
+    const workflow = {
+      prompt: {
+        '3': { class_type: 'KSampler', inputs: { seed: seedArg ?? 0, steps: cfg.localSteps ?? 20, cfg: cfg.localCfg ?? 7, sampler_name: 'euler', scheduler: 'normal', denoise: 1, model: ['4', 0], positive: ['6', 0], negative: ['7', 0], latent_image: ['5', 0] } },
+        '4': { class_type: 'CheckpointLoaderSimple', inputs: { ckpt_name: cfg.localModel || 'v1-5-pruned-emaonly.safetensors' } },
+        '5': { class_type: 'EmptyLatentImage', inputs: { width, height, batch_size: 1 } },
+        '6': { class_type: 'CLIPTextEncode', inputs: { text: prompt, clip: ['4', 1] } },
+        '7': { class_type: 'CLIPTextEncode', inputs: { text: negativePrompt || '', clip: ['4', 1] } },
+        '8': { class_type: 'VAEDecode', inputs: { samples: ['3', 0], vae: ['4', 2] } },
+        '9': { class_type: 'SaveImage', inputs: { filename_prefix: 'dsh', images: ['8', 0] } },
+      },
+    }
+    const submit = await fetchImpl(`${base}/prompt`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ prompt: workflow, client_id: promptId }),
+      signal,
+    })
+    if (!submit.ok) {
+      throw new Error(`Local ComfyUI submit failed (HTTP ${submit.status}): ${String(await submit.text().catch(() => '')).slice(0, 300)}`)
+    }
+    const submitData = await submit.json().catch(() => ({}))
+    const pid = submitData.prompt_id
+    if (!pid) throw new Error('Local ComfyUI did not return a prompt_id')
+
+    const deadline = Date.now() + cfg.timeoutMs
+    for (;;) {
+      if (signal.aborted) throw new Error('Local ComfyUI generation cancelled')
+      if (Date.now() > deadline) throw new Error(`Local ComfyUI timed out after ${cfg.timeoutMs} ms`)
+      await new Promise((resolve) => {
+        const timer = setTimeout(resolve, cfg.pollIntervalMs || 2000)
+        signal.addEventListener('abort', () => { clearTimeout(timer); resolve() }, { once: true })
+      })
+      const hist = await fetchImpl(`${base}/history/${pid}`, { signal })
+      if (!hist.ok) continue
+      const histData = await hist.json().catch(() => ({}))
+      const entry = histData[pid]
+      if (entry && entry.outputs) {
+        const outputs = entry.outputs
+        const img = Object.values(outputs).flatMap((o) => o.images || []).find((i) => i && i.filename)
+        if (img) {
+          const dl = await fetchImpl(`${base}/view?filename=${encodeURIComponent(img.filename)}&subfolder=${encodeURIComponent(img.subfolder || '')}&type=${encodeURIComponent(img.type || 'output')}`, { signal })
+          if (!dl.ok) throw new Error(`Local ComfyUI download failed (HTTP ${dl.status})`)
+          return {
+            bytes: Buffer.from(await dl.arrayBuffer()),
+            mediaType: normalizeMediaType('image/png', format),
+            width,
+            height,
+            seed: seedArg ?? 0,
+            sourceUrl: '',
+          }
+        }
+      }
+    }
+  }
+
+  return { fal, custom, codex: subscription('codex'), grok: subscription('grok'), local }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-image-gen",
-  "version": "0.8.5",
+  "version": "0.8.7",
   "description": "Image generation for DeepSeek Harness: a generate_image tool with pluggable providers — the FAL queue, any OpenAI-compatible images API, or a ChatGPT/Grok subscription with no API key at all. The picture is shown inline in the conversation; the model receives either a link (works with any chat model) or the image itself (needs dsh-vision-bridge or a vision-capable model).",
   "keywords": [
     "deepseek-harness",