${escapeHtml(
seen,
)} — pick a checkpoint with architecture krea-2.`
: 'Swarm Assistent is for Krea 2 models only. Select a Krea 2 checkpoint on Generate to enable the chat.';
}
}
}
if (layout) {
layout.classList.toggle('sa-disabled', !ok);
}
return ok;
}
function escapeHtml(s) {
return String(s)
.replace(/&/g, '&')
.replace(//g, '>')
.replace(/"/g, '"');
}
/** Known ### headings → RU display labels. JSON Patch is hidden (real UI is the patch strip). */
const PROSE_SECTION_TITLES = {
critique: 'Критика',
критика: 'Критика',
analysis: 'Разбор',
разбор: 'Разбор',
notes: 'Заметки',
заметки: 'Заметки',
summary: 'Кратко',
кратко: 'Кратко',
prompt: 'Промпт',
промпт: 'Промпт',
'improved prompt': 'Промпт',
'next prompt': 'Промпт',
deliverable: 'Итог',
итог: 'Итог',
verdict: 'Вердикт',
вердикт: 'Вердикт',
issues: 'Проблемы',
проблемы: 'Проблемы',
fixes: 'Правки',
правки: 'Правки',
suggestion: 'Предложение',
suggestions: 'Предложения',
предложения: 'Предложения',
};
function localizeProseHeading(raw) {
const cleaned = String(raw || '').replace(/[*_`#]/g, '').trim();
if (!cleaned) {
return null;
}
const key = cleaned.toLowerCase().replace(/\s+/g, ' ');
if (/^json\s*patch$/.test(key) || /^патч$/.test(key) || /^json\s*патч$/.test(key)) {
return null;
}
if (PROSE_SECTION_TITLES[key]) {
return PROSE_SECTION_TITLES[key];
}
// "Critique — blur" → take first token bucket
const head = key.split(/[—:\-|]/)[0].trim();
if (PROSE_SECTION_TITLES[head]) {
return PROSE_SECTION_TITLES[head];
}
return cleaned;
}
function formatProseInline(escapedLine) {
let t = escapedLine;
t = t.replace(/`([^`]+)`/g, '$1');
t = t.replace(/\*\*([^*]+)\*\*/g, '$1');
t = t.replace(/(^|[^*])\*([^*\n]+)\*(?!\*)/g, '$1$2');
return t;
}
/** Lightweight chat prose: ### Critique → «Критика», lists, bold — not a full markdown engine. */
function formatAssistantProseHtml(raw) {
let text = String(raw || '').replace(/\r\n/g, '\n');
text = text.replace(/(?:^|\n)#{1,6}\s*JSON\s*Patch\s*(?=\n|$)/gi, '\n');
text = text.replace(/(?:^|\n)\s*JSON\s*Patch\s*:?\s*(?=\n|$)/gi, '\n');
text = text.replace(/\n{3,}/g, '\n\n').trim();
if (!text) {
return '';
}
const lines = text.split('\n');
const parts = [];
let listItems = [];
const flushList = () => {
if (!listItems.length) {
return;
}
parts.push(
`- ${listItems.map((li) => `
- ${formatProseInline(escapeHtml(li))} `).join('')}
${escapeHtml(title)}
`,
);
continue;
}
const bullet = line.match(/^\s*[-*•]\s+(.+)$/);
if (bullet) {
listItems.push(bullet[1]);
continue;
}
flushList();
if (!line.trim()) {
parts.push('');
continue;
}
parts.push(`${formatProseInline(escapeHtml(line))}
`); } flushList(); return parts.join(''); } function setAssistantBody(div, text, { live = false } = {}) { if (!div) { return; } let body = div.querySelector('.sa-msg-body'); if (!body) { body = document.createElement('div'); body.className = 'sa-msg-body'; div.appendChild(body); } const raw = text || ''; // While streaming, plain text — full HTML reformat every token reflows and shakes scroll. if (live) { body.classList.add('sa-prose', 'sa-prose-live'); body.classList.remove('sa-prose-rich'); body.textContent = raw; return; } body.classList.add('sa-prose', 'sa-prose-rich'); body.classList.remove('sa-prose-live'); const html = formatAssistantProseHtml(raw); if (html) { body.innerHTML = html; } else { body.textContent = ''; } } function val(id) { const el = document.getElementById(id); return el ? el.value : ''; } function setVal(id, value) { const el = document.getElementById(id); if (!el) { return; } el.value = value; el.dispatchEvent(new Event('input', { bubbles: true })); el.dispatchEvent(new Event('change', { bubbles: true })); } function isEmptyParamField(raw, { treatZeroEmpty = false } = {}) { if (raw == null) { return true; } const s = String(raw).trim(); if (s === '') { return true; } if (treatZeroEmpty && (s === '0' || Number(s) === 0)) { return true; } return false; } /** JS \\b/\\w are ASCII-only — use this for RU tokens. `alts` = regex alternatives without outer parens. */ function cyrTokenRe(alts) { const boundary = '(^|[^0-9A-Za-z_А-Яа-яЁё])'; const end = '(?=$|[^0-9A-Za-z_А-Яа-яЁё])'; return new RegExp(`${boundary}(?:${alts})${end}`, 'i'); } /** Parse aspect from chat: 9:16, 9x16, 9×16, «9 на 16», «9 к 16». */ function parseAspectFromUserText(text) { const t = String(text || ''); if (!t.trim()) { return null; } const ratio = t.match(/(?:^|[^0-9])(\d+(?:\.\d+)?)\s*[:x×хX]\s*(\d+(?:\.\d+)?)(?=$|[^0-9])/); if (ratio) { const key = normalizeAspect(`${ratio[1]}:${ratio[2]}`); if (key) { return key; } } const na = t.match(/(?:^|[^0-9])(\d+(?:\.\d+)?)\s*(?:на|к|to)\s*(\d+(?:\.\d+)?)(?=$|[^0-9])/i); if (na) { const key = normalizeAspect(`${na[1]}:${na[2]}`); if (key) { return key; } } const named = t.match(/\b(16:9|9:16|1:1|4:5|2:3|3:2|4:3|2\.35:1)\b/i); if (named) { return normalizeAspect(named[1]); } if (cyrTokenRe('портрет|вертикал[а-яё]*').test(t) || /\b(portrait|vertical)\b/i.test(t)) { return normalizeAspect('9:16') || normalizeAspect('2:3'); } if (cyrTokenRe('альбом|горизонтал[а-яё]*').test(t) || /\b(landscape|horizontal|widescreen)\b/i.test(t)) { return normalizeAspect('16:9'); } return null; } /** «такую же, только 9 на 16» — keep prompt, change aspect, generate (no LLM needed). */ function isSameButAspectRequest(text) { const t = String(text || ''); if (!parseAspectFromUserText(t)) { return false; } return /такую\s+же|тот\s+же\s+промпт|same\s+(one|prompt|thing|again)|только\s+(поменя|смени|поставь)|поменяй\s+на|смени\s+на|only\s+change|just\s+change/i.test(t) || /поменяй\s+(размер|aspect|соотношен)/i.test(t) || /смени\s+(размер|aspect|соотношен)/i.test(t); } function userTextMentionsControls(text) { const t = String(text || ''); if (/\b(horny|controls?|preference[_\s-]?bias)\b/i.test(t)) { return true; } // Do not match bare «вкус» — too common in RU chat and was disabling the echo filter. return cyrTokenRe('хорни|остынь|слайдер').test(t) || /\/\s*(остынь|ostyn|horny-game)/i.test(t) || /слайдер\s*вкус|вкус\s*(на|в)\s*\d|поставь\s*вкус|крутани\s*вкус/i.test(t); } function patchLooksLikeGeneration(patch) { if (!patch || typeof patch !== 'object') { return false; } if (patch.prompt != null || patch.loras || patch.aspect != null || patch.width != null || patch.height != null || patch.steps != null || patch.cfg != null || patch.seed != null) { return true; } return Array.isArray(patch.actions) && patch.actions.map(String).includes('generate'); } /** Drop Generate-patch control noise; drop default-echo that would wipe a user-tuned slider. */ function filterControlPatch(incoming, patch) { const schema = state.config?.controls || {}; const out = {}; if (!incoming || typeof incoming !== 'object') { return out; } // Image patches must not move Хорни / Вкус unless the user asked this turn. if (patchLooksLikeGeneration(patch) && !state.lastUserControlIntent) { return out; } for (const [id, raw] of Object.entries(incoming)) { if (!schema[id]) { continue; } const n = Number(raw); if (!Number.isFinite(n)) { continue; } const def = Number(schema[id]?.default); const cur = getControlValue(id, Number.isFinite(def) ? def : n); if (Math.abs(n - cur) < 0.0005) { continue; } if (!state.lastUserControlIntent && Number.isFinite(def) && Math.abs(n - def) < 0.0005 && Math.abs(cur - def) > 0.0005) { continue; } out[id] = n; } return out; } function userTextMentionsParams(text) { const t = String(text || ''); if (parseAspectFromUserText(t)) { return true; } if (/\b(steps?|cfg|seed|sigma|aspect|resolution|batch|turbo|raw)\b/i.test(t)) { return true; } return cyrTokenRe('размер|ширин[а-яё]*|высот[а-яё]*|соотношен[а-яё]*|турбо|портрет|вертикал[а-яё]*').test(t); } function replyMissingJsonPatch(reply) { const t = String(reply || ''); if (!t.trim()) { return false; } if (/```(?:json)?\s*\{[\s\S]*?\}```/i.test(t)) { return false; } return /###\s*JSON\s*Patch\b/i.test(t) || /JSON\s*Patch\s*:?\s*$/im.test(t); } function userAsksGenerate(text) { const t = String(text || '').trim(); if (!t) { return false; } // Short imperatives only — do NOT treat bare «давай» as Generate (false positive on chat). if (/^(gen|generate|go|рисуй|нарисуй)([!.…\s]|$)/i.test(t)) { return true; } if (/^(ещё|еще)(\s+раз)?([!.…\s]|$)/i.test(t)) { return true; } // Do NOT use \b or \w — ASCII-only in JS; breaks «сделай картинку». const letter = '[0-9A-Za-z_А-Яа-яЁё]'; const stem = `${letter}*`; return cyrTokenRe( 'сгенерируй|сгенерировать|генерируй|generate|нарисуй|перегенерируй|перерисуй|' + `сделай\\s+(картинк${stem}|изображен${stem}|фото${stem})|` + `хочу\\s+(картинк${stem}|изображен${stem}|фото${stem})|` + 'run\\s+generat|/gen', ).test(t); } /** «давай дальше / продолжай / следующий кадр» — continue the series with Generate. */ function userAsksContinue(text) { const t = String(text || '').trim(); if (!t) { return false; } if (/^(давай\s+дальше|продолжай|продолжим|go\s+on|continue|keep\s+going|next(\s+one)?|next\s+frame)([!.…\s]|$)/i.test(t)) { return true; } return cyrTokenRe( 'давай\\s+дальше|следующ(ий|ая|ее|ую)\\s+кадр|ещё\\s+кадр|еще\\s+кадр|' + 'кадр\\s*№?\\s*\\d+|сделай\\s+следующ', ).test(t); } /** Pull a usable prompt out of assistant prose when the JSON fence is missing. */ function extractPromptFromProse(reply) { const t = String(reply || '').replace(/\r\n/g, '\n'); if (!t.trim()) { return null; } const bq = []; for (const line of t.split('\n')) { const m = line.match(/^\s{0,3}>\s?(.*)$/); if (m) { bq.push(m[1]); continue; } if (bq.length) { break; } } const fromBq = bq.join('\n').trim(); if (fromBq.length >= 48) { return fromBq.slice(0, 4000); } const section = t.match( /(?:^|\n)#{1,6}\s*(?:📷\s*)?(?:prompt|промпт|improved\s+prompt|next\s+prompt|кадр[^\n]*)\s*\n+([\s\S]+?)(?=\n#{1,6}\s|\n```|$)/i, ); if (section) { const body = section[1].replace(/^\s{0,3}>\s?/gm, '').trim(); if (body.length >= 48) { return body.slice(0, 4000); } } return null; } /** When the model wrote ### JSON Patch with no fence — build Apply+Generate from prose / last patch. */ function synthesizePatchAfterEmptyFence(reply, userText, opts = {}) { const wants = !!(opts.userWantsGenerate || state.pendingSilentGen || userAsksGenerate(userText) || userAsksContinue(userText)); const missing = replyMissingJsonPatch(reply); if (!wants && !missing) { return null; } const fromProse = extractPromptFromProse(reply); const prompt = fromProse || state.lastPatch?.prompt || null; if (!prompt) { return null; } const patch = { prompt, actions: ['generate'] }; if (Array.isArray(state.lastPatch?.loras) && state.lastPatch.loras.length) { patch.loras = state.lastPatch.loras; } if (state.lastPatch?.aspect) { patch.aspect = state.lastPatch.aspect; } return patch; } /** * Chat model prepares Generate prompts for Krea. Skip the prep hop only when the * prompt already looks like solid English Krea prose for Qwen3-VL. */ function promptLooksKreaReady(prompt) { const t = String(prompt || '').trim(); if (t.length < 80) { return false; } const cyr = (t.match(/[\u0400-\u04FF]/g) || []).length; const lat = (t.match(/[A-Za-z]/g) || []).length; if (cyr >= 12) { return false; } if (lat < 55) { return false; } // Prefer a real paragraph (clauses), not a one-liner dump. if (t.length < 120 && (t.match(/[,.;:]/g) || []).length < 2) { return false; } return true; } function promptNeedsKreaPrep(prompt) { return !promptLooksKreaReady(prompt); } function buildKreaPromptPrepRequest(patch) { const keep = { actions: Array.isArray(patch.actions) && patch.actions.length ? patch.actions : ['generate'], }; if (patch.aspect) { keep.aspect = patch.aspect; } if (Array.isArray(patch.loras)) { keep.loras = patch.loras; } if (patch.width != null) { keep.width = patch.width; } if (patch.height != null) { keep.height = patch.height; } return ( 'You are the Krea 2 prompt prep step (chat model). Rewrite SOURCE into the final Swarm Generate box text.\n' + 'HARD RULES:\n' + '- JSON "prompt": English only (no Cyrillic) — translate if needed.\n' + '- Natural photographer/director prose for Qwen3-VL — not Danbooru tags, not (word:1.5), not masterpiece/best quality/8k.\n' + '- Structure & front-load: subject → pose/action → body/wardrobe → setting → materials/textures → camera/framing → lighting → medium/mood.\n' + '- Expand thin ideas; fix anti-patterns; one coherent scene.\n' + '- Keep LoRA trigger phrases in English near the subject they affect.\n' + '- Prefer positives over negatives; preserve meaning and NSFW level from SOURCE.\n' + '- One short ack in the user language max, then ONE fenced JSON merging these keys: ' + `${JSON.stringify(keep)} plus the new English "prompt".\n` + '- Include actions:["generate"] when an image was requested.\n\n' + `SOURCE:\n${patch.prompt}` ); } function mergePromptEnRewrite(effective) { const base = state.pendingPromptEnMerge; state.pendingPromptEnMerge = null; if (!base || !effective) { return effective; } return { ...base, ...effective, prompt: effective.prompt || base.prompt, actions: (Array.isArray(effective.actions) && effective.actions.length) ? effective.actions : (base.actions || ['generate']), loras: effective.loras || base.loras, aspect: effective.aspect || base.aspect, }; } /** «запомни как базовый промпт» — apply/save only, never Generate / auto look_at. */ function userAsksNoGenerate(text) { const t = String(text || '').trim(); if (!t || userAsksGenerate(t)) { return false; } if (/\b(remember|save\s+(this\s+)?(as\s+)?(the\s+)?(base\s+)?(prompt|template)|don'?t\s+generat|do\s+not\s+generat|no\s+generat|without\s+generat)\b/i.test(t)) { return true; } return cyrTokenRe( 'запомн|запомни|запомним|сохрани|сохраним|шаблон|' + 'базов(ый|ого|ому|ым|ая|ую|ое)?\\s+промпт|' + 'не\\s+генерир|без\\s+генерац|не\\s+надо\\s+генер|только\\s+запомн|пока\\s+запомн|' + 'не\\s+рисуй|не\\s+запускай\\s+генер', ).test(t); } function stripGenerateAction(patch) { if (!patch || typeof patch !== 'object') { return patch; } if (!Array.isArray(patch.actions)) { return patch; } const next = patch.actions.map(String).filter((a) => a !== 'generate'); if (next.length === patch.actions.length) { return patch; } const out = { ...patch }; if (next.length) { out.actions = next; } else { delete out.actions; } return out; } function stripLookAt(patch) { if (!patch || typeof patch !== 'object') { return patch; } if (patch.look_at == null && patch.vision_from == null && patch.vision_slots == null) { return patch; } const out = { ...patch }; delete out.look_at; delete out.vision_from; delete out.vision_slots; return out; } function rememberLastPatch(patch) { if (patch && typeof patch === 'object' && !isCardObject(patch)) { state.lastPatch = patch; syncBuildGenButton(); } } function syncBuildGenButton() { const btn = $('sa_btn_build_gen'); if (!btn) { return; } if (state.lastPatch) { const keys = Object.keys(state.lastPatch).filter((k) => state.lastPatch[k] != null).slice(0, 6); btn.title = `Есть патч Assistent (${keys.join(', ') || '…'}) → Apply + Generate`; btn.classList.add('sa-has-patch'); } else { btn.title = 'Нет патча — Generate с текущим промптом SwarmUI'; btn.classList.remove('sa-has-patch'); } } function defaultPackId() { return state.config?.assistant?.default_pack || $('sa_pack')?.querySelector('option')?.value || 'ordinary'; } function syncModeBadge() { const badge = $('sa_mode_badge'); const pack = $('sa_pack')?.value || defaultPackId(); if (!badge) { return; } const shortMap = { ordinary: 'обычный', write_prompt: 'write', critique_image: 'critique', compose_scene: 'compose', fix_params: 'params', inpaint_edit: 'inpaint', describe_ref: 'describe', catalog_card: 'card', author_persona: 'persona', }; const short = shortMap[pack] || pack.replace(/_/g, ' ').slice(0, 12); badge.textContent = short; badge.dataset.pack = pack; badge.title = `Режим: ${pack}`; badge.classList.toggle('sa-mode-hot', pack === 'critique_image' || pack === 'inpaint_edit'); } function syncLiveParamsBar() { const el = $('sa_live_params'); if (!el) { return; } const w = parseInt(val('input_width') || '0', 10) || null; const h = parseInt(val('input_height') || '0', 10) || null; const aspect = guessAspectFromSize(w, h) || '—'; const steps = val('input_steps') || '—'; const cfg = val('input_cfgscale') || val('input_cfg') || '—'; const seed = val('input_seed') || '—'; const profile = detectKreaProfileName(); el.textContent = `${aspect} · ${w || '?'}×${h || '?'} · steps ${steps} · cfg ${cfg} · ${profile} · seed ${seed}`; } function applyAspectTableFrom(obj) { if (!obj || typeof obj !== 'object') { return false; } const next = {}; for (const [k, v] of Object.entries(obj)) { if (Array.isArray(v) && v.length >= 2) { next[k] = [Number(v[0]), Number(v[1])]; } } if (!Object.keys(next).length) { return false; } ASPECT_TABLE = next; return true; } function resolveExactBundle() { const exact = state.exact || state.config?.exact || {}; const profiles = exact.profiles || state.kreaProfiles || {}; return { exact, profiles }; } function detectKreaProfileName() { try { const model = resolveCurrentCheckpoint(); const blob = `${model?.name || ''} ${model?.title || ''}`.toLowerCase(); const hasRaw = /\braw\b/.test(blob); const hasTurbo = /\bturbo\b/.test(blob); return hasRaw && !hasTurbo ? 'raw' : 'turbo'; } catch (e) { return (state.exact?.generation?.profile) || 'turbo'; } } function mergedGenerationDefaults(profileName) { const { exact, profiles } = resolveExactBundle(); const gen = exact.generation && typeof exact.generation === 'object' ? { ...exact.generation } : {}; const profile = profileName || gen.profile || detectKreaProfileName(); const fromProfile = profiles[profile] && typeof profiles[profile] === 'object' ? { ...profiles[profile] } : {}; const session = state.sessionExact && typeof state.sessionExact === 'object' ? { ...state.sessionExact } : {}; // Profile (turbo/raw) overrides generation defaults; session overrides both. return { ...gen, ...fromProfile, profile, ...session }; } function exactDefaultFor(key, profileName) { const { exact, profiles } = resolveExactBundle(); const profile = profileName || exact.generation?.profile || detectKreaProfileName(); const fromProfile = profiles[profile]?.[key]; if (fromProfile != null) { return fromProfile; } return exact.generation?.[key]; } function rememberSessionExact(partial) { if (state.restoringChat || !partial || typeof partial !== 'object') { return; } const keys = ['steps', 'cfg', 'sigma_shift', 'aspect', 'width', 'height', 'images', 'batch', 'seed', 'sampler', 'scheduler']; for (const k of keys) { if (partial[k] != null) { state.sessionExact[k] = partial[k]; } } if (partial.images == null && partial.batch != null) { state.sessionExact.images = partial.batch; } } function fillEmptyParamsFromExact() { const defaults = mergedGenerationDefaults(); if (isEmptyParamField(val('input_steps'), { treatZeroEmpty: true }) && defaults.steps != null) { setVal('input_steps', String(defaults.steps)); } const cfgRaw = val('input_cfgscale') || val('input_cfg'); if (isEmptyParamField(cfgRaw, { treatZeroEmpty: true }) && defaults.cfg != null) { if (document.getElementById('input_cfgscale')) { setVal('input_cfgscale', String(defaults.cfg)); } else if (document.getElementById('input_cfg')) { setVal('input_cfg', String(defaults.cfg)); } } if (isEmptyParamField(val('input_sigmashift')) && defaults.sigma_shift != null) { setVal('input_sigmashift', String(defaults.sigma_shift)); } const wEmpty = isEmptyParamField(val('input_width'), { treatZeroEmpty: true }); const hEmpty = isEmptyParamField(val('input_height'), { treatZeroEmpty: true }); if ((wEmpty || hEmpty) && defaults.aspect) { const size = sizeFromAspect(defaults.aspect); if (size) { if (wEmpty) { setVal('input_width', String(size[0])); } if (hEmpty) { setVal('input_height', String(size[1])); } } } else { if (wEmpty && defaults.width != null) { setVal('input_width', String(defaults.width)); } if (hEmpty && defaults.height != null) { setVal('input_height', String(defaults.height)); } } const batchId = document.getElementById('input_images') ? 'input_images' : (document.getElementById('input_batchsize') ? 'input_batchsize' : null); if (batchId && isEmptyParamField(val(batchId), { treatZeroEmpty: true })) { const batch = defaults.images != null ? defaults.images : defaults.batch; if (batch != null) { setVal(batchId, String(batch)); } } } function shouldSkipSessionRollback(key, patchValue) { if (state.restoringChat) { return false; } if (state.lastUserParamIntent) { return false; } if (state.sessionExact[key] == null) { return false; } const sessionVal = state.sessionExact[key]; if (String(sessionVal) === String(patchValue)) { return false; } const exactVal = exactDefaultFor(key); if (exactVal == null) { return false; } // Model trying to restore file exact while session override differs — keep session. return String(patchValue) === String(exactVal); } function openAssistentTab() { const tab = document.getElementById(TAB_BUTTON_ID); if (tab) { tab.click(); setTimeout(() => $('sa_input')?.focus(), 50); return true; } const pane = document.getElementById('assistent'); if (pane && typeof bootstrap !== 'undefined' && bootstrap.Tab) { try { bootstrap.Tab.getOrCreateInstance(tab || pane).show(); } catch (e) { /* ignore */ } } setTimeout(() => $('sa_input')?.focus(), 50); return !!tab; } function historyMessageLimit() { const turns = Math.max(1, Number(HISTORY_KEEP_TURNS) || 4); return turns * 2; } function flashImagePane(slotId) { const el = document.querySelector(`.sa-slot[data-id="${slotId || state.selectedSlotId}"]`); if (!el) { return; } el.classList.remove('sa-flash'); void el.offsetWidth; el.classList.add('sa-flash'); } function ensureBoard() { if (state.slots.length) { return; } state.slots = [ { id: GEN_ID, type: 'generate', label: 'Generate', src: null, attach: false }, { id: 'ref1', type: 'ref', label: 'Ref 1', src: null, attach: true }, ]; state.refSeq = 1; state.selectedSlotId = 'ref1'; } function slotById(id) { ensureBoard(); const key = normalizeSlotId(id); return state.slots.find((s) => s.id === key) || null; } function generateSlot() { return slotById(GEN_ID); } function refSlots() { ensureBoard(); return state.slots.filter((s) => s.type === 'ref'); } function normalizeSlotId(id) { const raw = String(id || '').trim().toLowerCase(); if (!raw) { return ''; } if (raw === 'gen' || raw === 'current' || raw === 'live' || raw === 'generation') { return GEN_ID; } if (raw === 'selected' || raw === 'sel') { return state.selectedSlotId; } const m = raw.match(/^ref\s*[_-]?\s*(\d+)$/); if (m) { return `ref${m[1]}`; } return raw; } function selectedSlot() { return slotById(state.selectedSlotId) || generateSlot(); } function selectedSrc() { return selectedSlot()?.src || null; } function syncLastImageAlias() { const attached = attachableSlots(); state.lastImageDataUrl = (attached[0] || selectedSlot() || generateSlot())?.src || null; } function attachableSlots() { ensureBoard(); return state.slots.filter((s) => s.attach && s.src); } /** Real board frames (not model previews). Used for has_vision_image even when JPEG is not sent. */ function visionReadySlots() { ensureBoard(); return state.slots.filter((s) => s && s.src && !looksLikeModelPreview(s.src)); } function setSlotSrc(id, src, { select = true, attach = null, note = null, switchTab = false, allowPreview = false } = {}) { const slot = slotById(id); if (!slot) { return false; } const cleaned = src ? String(src).trim().split(/\s+/)[0] : null; if (cleaned && cleaned.startsWith('#')) { return false; } if (cleaned && !allowPreview && looksLikeModelPreview(cleaned)) { setStatus('Пропуск превью модели (нужна реальная генерация)'); return false; } slot.src = cleaned || null; if (attach != null) { slot.attach = !!attach; } else if (slot.type === 'ref' && slot.src) { slot.attach = true; } if (select) { state.selectedSlotId = slot.id; } syncLastImageAlias(); renderBoard(); flashImagePane(slot.id); if (switchTab) { openAssistentTab(); } if (note) { setStatus(note); } return true; } function addRefSlot({ src = null, select = true } = {}) { ensureBoard(); if (refSlots().length >= MAX_REF_SLOTS) { setStatus(`Max ${MAX_REF_SLOTS} reference windows`); const empty = refSlots().find((s) => !s.src); if (empty && src) { return setSlotSrc(empty.id, src, { select, note: `Loaded into ${empty.label}` }); } return empty || null; } state.refSeq += 1; const id = `ref${state.refSeq}`; const slot = { id, type: 'ref', label: `Ref ${state.refSeq}`, src: src || null, attach: !!src, }; state.slots.push(slot); if (select) { state.selectedSlotId = id; } renderBoard(); return slot; } function clearSlot(id, { silent = false } = {}) { const slot = slotById(id); if (!slot) { return; } if (slot.type === 'generate') { if (!silent) { setStatus('Generate window is live — use Snapshot gen to copy it'); } return; } slot.src = null; slot.attach = true; syncLastImageAlias(); renderBoard(); if (!silent) { setStatus(`${slot.label} cleared`); } } function snapshotGenerateToRef() { // Never promote checkpoint/LoRA card previews into Refs. const src = generateSlot()?.src && !looksLikeModelPreview(generateSlot().src) ? generateSlot().src : findCurrentGenerateSrc({ allowPreview: false }); if (!src) { setStatus('Нет текущего кадра Generate (превью модели не считается)'); return false; } const empty = refSlots().find((s) => !s.src); let ok = false; if (empty) { ok = setSlotSrc(empty.id, src, { note: `Снимок → ${empty.label}` }); } else { const created = addRefSlot({ src, select: true }); if (created?.src) { setStatus(`Снимок → ${created.label}`); flashImagePane(created.id); ok = true; } else { const last = refSlots()[refSlots().length - 1]; if (last) { ok = setSlotSrc(last.id, src, { note: `Снимок → ${last.label} (замена)` }); } } } if (ok) { setBoardTab('refs'); } return ok; } function putImageOnBoard(src, { note = null, switchTab = false, preferSelected = true } = {}) { if (!src) { return false; } ensureBoard(); const sel = selectedSlot(); if (preferSelected && sel && sel.type === 'ref') { return setSlotSrc(sel.id, src, { note: note || `Loaded into ${sel.label}`, switchTab }); } const empty = refSlots().find((s) => !s.src); if (empty) { return setSlotSrc(empty.id, src, { note: note || `Loaded into ${empty.label}`, switchTab }); } const created = addRefSlot({ src, select: true }); if (created) { if (switchTab) { openAssistentTab(); } if (note) { setStatus(note); } return true; } return false; } function setImageFromSrc(src, opts = {}) { return putImageOnBoard(src, opts); } function clearVisionImage(opts) { clearSlot(state.selectedSlotId, opts); } function slotCatalog() { ensureBoard(); return state.slots.map((s) => ({ id: s.id, type: s.type, label: s.label, has_image: !!s.src, attach: !!s.attach, selected: s.id === state.selectedSlotId, })); } function lookAtIdsFromPatch(patch) { if (!patch) { return []; } const raw = patch.look_at || patch.vision_from || patch.vision_slots; const list = Array.isArray(raw) ? raw : (raw ? [raw] : []); if (Array.isArray(patch.actions)) { for (const a of patch.actions.map(String)) { const m = a.match(/^look_at[_:]?(generate|ref\d+|selected)$/i); if (m) { list.push(m[1]); } } } return [...new Set(list.map(normalizeSlotId).filter(Boolean))]; } function resolveSlotSrc(id) { if (!id) { return selectedSrc() || generateSlot()?.src || findCurrentGenerateSrc(); } const slot = slotById(id); if (slot?.src) { return slot.src; } if (normalizeSlotId(id) === GEN_ID) { return findCurrentGenerateSrc(); } return null; } function isSwarmGenerateRunning() { try { if (typeof num_live_gens === 'number' && num_live_gens > 0) { return true; } if (typeof num_waiting_gens === 'number' && num_waiting_gens > 0) { return true; } } catch (e) { /* ignore */ } try { if (typeof mainGenHandler !== 'undefined' && mainGenHandler) { if (mainGenHandler.isGenerating === true || mainGenHandler.running === true) { return true; } } } catch (e) { /* ignore */ } const interrupt = document.getElementById('interrupt_button') || document.getElementById('alt_interrupt_button'); if (interrupt && !interrupt.hidden && interrupt.offsetParent !== null) { return true; } const genBtn = document.getElementById('generate_button') || document.getElementById('alt_generate_button'); if (genBtn && (genBtn.disabled || /interrupt/i.test(genBtn.textContent || ''))) { return true; } return false; } function isGenerateUnavailable() { if (state.generating || state.busy) { return true; } return isSwarmGenerateRunning(); } function syncGenerateBusy() { const overlay = document.querySelector('.sa-slot-gen .sa-slot-busy'); if (overlay) { // If Swarm already finished but our waiter is stuck on same URL, drop the overlay // as soon as the live frame is on the board. const stuck = state.generating && !isSwarmGenerateRunning(); overlay.hidden = (!state.generating && state.busyPhase !== 'generating') || stuck; } const running = state.generating || state.busyPhase === 'generating'; document.querySelectorAll('.sa-slot-gen-result').forEach((el) => { const busy = el.querySelector('.sa-slot-busy'); if (!busy) { return; } const hasImg = el.classList.contains('sa-has-image'); busy.hidden = !running || hasImg; }); } function syncPatchActionAvailability() { const bar = document.querySelector('.sa-patch-actions.sa-patch-current'); if (!bar) { return; } const locked = isGenerateUnavailable(); bar.querySelectorAll('.sa-btn-gen').forEach((btn) => { btn.disabled = locked; let spin = btn.querySelector('.sa-spinner'); if (locked) { if (!spin) { spin = document.createElement('span'); spin.className = 'sa-spinner sa-spinner-btn'; spin.setAttribute('aria-hidden', 'true'); btn.prepend(spin); } } else if (spin) { spin.remove(); } }); } function retireStalePatchActions() { document.querySelectorAll('.sa-patch-actions').forEach((el) => { const note = document.createElement('div'); note.className = 'sa-patch-stale'; note.textContent = 'Superseded — use the latest proposal'; el.replaceWith(note); }); } function mountPatchBlock(host, patch, { silent = false } = {}) { if (!host || !patch) { return; } rememberLastPatch(patch); const wrap = document.createElement('div'); wrap.className = 'sa-patch' + (silent ? ' sa-patch-auto' : ''); const details = document.createElement('details'); details.className = 'sa-patch-details'; const summary = document.createElement('summary'); const keys = Object.keys(patch).filter((k) => patch[k] != null && k !== 'notes' && k !== 'actions'); summary.textContent = silent ? `Патч применён · ${keys.slice(0, 6).join(', ') || 'generate'}` : `JSON патч · ${keys.slice(0, 8).join(', ') || '…'}`; const pre = document.createElement('pre'); pre.textContent = JSON.stringify(patch, null, 2); details.appendChild(summary); details.appendChild(pre); wrap.appendChild(details); mountPatchActions(wrap, patch, { silent }); host.appendChild(wrap); } function mountPatchActions(parent, patch, { silent = false } = {}) { if (!parent || !patch) { return; } rememberLastPatch(patch); retireStalePatchActions(); const wrap = parent.classList.contains('sa-patch') ? parent : null; const host = wrap || parent; if (silent) { const note = document.createElement('div'); note.className = 'sa-patch-actions sa-patch-silent sa-patch-current'; const willGen = Array.isArray(patch.actions) && patch.actions.map(String).includes('generate') || !!state.pendingSilentGen; note.textContent = willGen ? 'Применено автоматически · Generate…' : 'Применено автоматически'; host.appendChild(note); return; } const actions = document.createElement('div'); actions.className = 'sa-patch-actions sa-patch-current'; for (const [label, which] of [ ['Применить всё', 'all'], ['Промпт', 'prompt'], ['LoRAs', 'loras'], ['Параметры', 'params'], ]) { const btn = document.createElement('button'); btn.type = 'button'; btn.className = 'basic-button'; btn.textContent = label; btn.addEventListener('click', () => applyPatch(patch, which)); actions.appendChild(btn); } const genBtn = document.createElement('button'); genBtn.type = 'button'; genBtn.className = 'basic-button sa-btn-gen'; genBtn.textContent = 'Применить + Generate'; genBtn.addEventListener('click', async () => { if (isGenerateUnavailable()) { return; } startBusyUi('silent_gen'); await applyPatch(patch, 'all'); await runGenerateFromPatch({ ...patch, actions: ['generate'] }, { force: true }); }); actions.appendChild(genBtn); host.appendChild(actions); syncPatchActionAvailability(); } async function buildCurrentAndGenerate() { if (state.busy || state.generating) { setStatus('Занято — подожди или нажми Стоп'); return; } if (isGenerateUnavailable()) { setStatus('Generate недоступен — дождись SwarmUI'); return; } const patch = state.lastPatch; if (patch) { startBusyUi('silent_gen'); setStatus('Собираю патч → Generate…'); await applyPatch(patch, 'all'); syncLiveParamsBar(); await runGenerateFromPatch({ ...patch, actions: ['generate'] }, { force: true }); return; } startBusyUi('generating'); setStatus('Generate с текущим промптом…'); await runGenerateFromPatch({ actions: ['generate'] }, { force: true }); } function renderBoard() { const board = $('sa_board'); if (!board) { return; } ensureBoard(); const tab = state.boardTab === 'refs' ? 'refs' : 'generate'; const refs = refSlots(); const showVariantGrid = tab === 'generate' && (state.genResults || []).length > 1; board.classList.toggle('sa-board-many', (tab === 'refs' && (refs.some((s) => s.src) || refs.length > 1)) || showVariantGrid); board.classList.toggle('sa-board-gen-only', tab === 'generate' && !showVariantGrid); board.classList.toggle('sa-board-variants', showVariantGrid); board.innerHTML = ''; if (tab === 'generate' && showVariantGrid) { for (const row of state.genResults) { board.appendChild(buildGenResultEl(row)); } } else { const toShow = tab === 'generate' ? state.slots.filter((s) => s.type === 'generate') : state.slots.filter((s) => s.type !== 'generate'); for (const slot of toShow) { board.appendChild(buildSlotEl(slot)); } } if (tab === 'refs' && refs.length < MAX_REF_SLOTS) { const add = document.createElement('div'); add.className = 'sa-add-cell'; add.textContent = '+ Ref'; add.title = 'Добавить окно референса'; add.addEventListener('click', (e) => { e.stopPropagation(); addRefSlot({ select: true }); }); add.addEventListener('dragover', (e) => { e.preventDefault(); e.stopPropagation(); }); add.addEventListener('drop', async (e) => { e.preventDefault(); e.stopPropagation(); const created = addRefSlot({ select: true }); if (created) { state.selectedSlotId = created.id; await handleDropDataTransfer(e.dataTransfer, created.id); } }); board.appendChild(add); } syncBoardChrome(); syncGenerateBusy(); syncLastImageAlias(); } function buildGenResultEl(row) { const el = document.createElement('div'); el.className = 'sa-slot sa-slot-gen-result'; el.dataset.genResultId = row.id; if (row.src) { el.classList.add('sa-has-image'); } if (row.id === state.selectedGenResultId) { el.classList.add('sa-selected'); } const bar = document.createElement('div'); bar.className = 'sa-slot-bar'; const chip = document.createElement('span'); chip.className = 'sa-slot-chip sa-live'; chip.textContent = row.label || row.id; bar.appendChild(chip); const openBtn = document.createElement('button'); openBtn.type = 'button'; openBtn.className = 'sa-slot-open'; openBtn.textContent = 'Открыть'; openBtn.title = 'Просмотр'; openBtn.hidden = !row.src; openBtn.addEventListener('click', (e) => { e.stopPropagation(); selectGenResult(row.id, { restore: true, openViewer: true }); }); bar.appendChild(openBtn); el.appendChild(bar); if (row.src) { const img = document.createElement('img'); img.alt = row.label || row.id; img.src = row.src; el.appendChild(img); } else { const empty = document.createElement('div'); empty.className = 'sa-image-empty'; empty.innerHTML = '…
Жду кадр
';
el.appendChild(empty);
}
const busy = document.createElement('div');
busy.className = 'sa-slot-busy';
const pending = !row.src && (state.generating || state.busyPhase === 'generating');
busy.hidden = !pending;
busy.innerHTML = '';
el.appendChild(busy);
el.addEventListener('click', () => {
const already = row.id === state.selectedGenResultId;
selectGenResult(row.id, { restore: true, openViewer: already && !!row.src });
});
el.addEventListener('dblclick', (e) => {
e.preventDefault();
if (row.src) {
selectGenResult(row.id, { restore: true, openViewer: true });
}
});
return el;
}
function ensureGenLightbox() {
let root = $('sa_gen_lightbox');
if (root) {
return root;
}
const host = $('swarm_assistent_root') || document.body;
root = document.createElement('div');
root.id = 'sa_gen_lightbox';
root.className = 'sa-lightbox';
root.hidden = true;
root.innerHTML = `
Generate
Живой просмотр текущей генерации
'
: 'Reference
Drop · paste · Снимок gen
';
el.appendChild(empty);
}
const busy = document.createElement('div');
busy.className = 'sa-slot-busy';
busy.hidden = !(slot.type === 'generate' && (state.generating || state.busyPhase === 'generating'));
busy.innerHTML = '';
el.appendChild(busy);
el.addEventListener('click', () => {
state.selectedSlotId = slot.id;
renderBoard();
});
el.addEventListener('dragover', (e) => {
e.preventDefault();
e.stopPropagation();
el.classList.add('sa-dragover');
if (e.dataTransfer) {
e.dataTransfer.dropEffect = 'copy';
}
});
el.addEventListener('dragleave', () => el.classList.remove('sa-dragover'));
el.addEventListener('drop', async (e) => {
e.preventDefault();
e.stopPropagation();
el.classList.remove('sa-dragover');
const targetId = slot.type === 'generate' ? null : slot.id;
if (slot.type === 'generate') {
const created = addRefSlot({ select: true });
await handleDropDataTransfer(e.dataTransfer, created?.id);
setBoardTab('refs');
} else {
await handleDropDataTransfer(e.dataTransfer, targetId);
}
});
return el;
}
function syncGenerateSlot() {
const slot = generateSlot();
if (!slot) {
return;
}
// Drop checkpoint/LoRA card previews that slipped into the Generate pane.
if (scrubPreviewFromGenerateSlot()) {
renderBoard();
}
const src = findCurrentGenerateSrc();
if (src && src !== slot.src) {
slot.src = src;
const img = document.querySelector('.sa-slot-gen img');
const empty = document.querySelector('.sa-slot-gen .sa-image-empty');
const frame = document.querySelector('.sa-slot-gen');
if (img) {
img.src = src;
} else if (frame) {
renderBoard();
return;
}
if (empty) {
empty.hidden = true;
}
frame?.classList.add('sa-has-image');
} else if (src && slot.src === src) {
// Same URL, possibly new bytes after overwrite — nudge Новый чат
Параметры Generate остаются как сейчас.
+ — ещё один чат · История — вернуться к прошлому (с его параметрами).
';
box.appendChild(empty);
}
function renderHistoryIntoUi(messages) {
const box = $('sa_messages');
if (!box) {
return;
}
box.innerHTML = '';
const list = slimHistoryMessages(messages);
if (!list.length) {
resetMessagesUi();
return;
}
for (const m of list) {
if (m.role === 'user') {
appendMessage('user', m.content, null, null, { historical: true });
} else {
appendMessage('assistant', m.content, null, null, {
persona: m.persona ? { id: m.persona, title: m.persona } : null,
pack: m.pack,
historical: true,
});
}
}
}
function updateSessionLabel() {
const el = $('sa_session_label');
if (!el) {
return;
}
const chat = findChat(state.activeChatId);
el.textContent = chat?.title || 'Новый чат';
el.title = (chat?.title || 'Новый чат') + ' — клик: История';
}
function savedChatsCount() {
return (state.chats || []).filter((c) => (c.messages || []).length > 0).length;
}
function syncHistoryBadge() {
const btn = $('sa_btn_chats');
if (!btn) {
return;
}
const n = savedChatsCount();
btn.textContent = n > 0 ? `История (${n})` : 'История';
btn.title = n > 0
? `Сохранённых чатов: ${n}. Переключение восстанавливает параметры.`
: 'История чатов (пока пусто)';
}
function formatChatWhen(ts) {
if (!ts) {
return '';
}
try {
return new Date(ts).toLocaleString(undefined, { month: 'short', day: 'numeric', hour: '2-digit', minute: '2-digit' });
} catch (e) {
return '';
}
}
function chatMatchesQuery(chat, q) {
if (!q) {
return true;
}
const title = String(chat?.title || '').toLowerCase();
if (title.includes(q)) {
return true;
}
const msgs = chat?.messages || [];
for (const m of msgs) {
if (String(m?.content || '').toLowerCase().includes(q)) {
return true;
}
}
return false;
}
function renderChatsList() {
const root = $('sa_chats_list');
if (!root) {
return;
}
root.innerHTML = '';
syncHistoryBadge();
const q = (state.chatsQuery || '').trim().toLowerCase();
let chats = (state.chats || [])
.slice()
.sort((a, b) => (b.updatedAt || 0) - (a.updatedAt || 0))
.filter((c) => (c.messages || []).length > 0 || (c.messages_count || 0) > 0);
if (q) {
const local = chats.filter((c) => chatMatchesQuery(c, q));
const seen = new Set(local.map((c) => c.id));
const extra = (state.chatsSearchHits || []).filter((h) => h && h.id && !seen.has(h.id));
chats = local.concat(extra);
}
if (!chats.length) {
root.innerHTML = q
? '+ — ещё один чат · История — вернуться к прошлому (с его параметрами).
Ничего не нашлось.
'
: 'Пока пусто. Напиши что-нибудь в чат — он появится здесь. Переключение вернёт и сообщения, и параметры Generate.
';
return;
}
for (const c of chats) {
const row = document.createElement('div');
row.className = 'sa-chat-row' + (c.id === state.activeChatId ? ' sa-chat-row-active' : '');
row.dataset.id = c.id;
const n = (c.messages || []).length || Number(c.messages_count) || 0;
const bits = [];
if (c.params?.width && c.params?.height) {
bits.push(`${c.params.width}×${c.params.height}`);
}
if (c.params?.steps != null) {
bits.push(`steps ${c.params.steps}`);
}
if (c.params?.cfg != null) {
bits.push(`cfg ${c.params.cfg}`);
}
if (Array.isArray(c.params?.loras) && c.params.loras.length) {
bits.push(`LoRA ${c.params.loras.length}`);
}
const noParams = !c.params ? ' · без снимка params' : '';
row.innerHTML = ``;
root.appendChild(row);
}
}
function setChatsPanelOpen(open) {
state.chatsPanelOpen = !!open;
const panel = $('sa_chats_panel');
const btn = $('sa_btn_chats');
if (panel) {
panel.hidden = !state.chatsPanelOpen;
}
btn?.setAttribute('aria-expanded', state.chatsPanelOpen ? 'true' : 'false');
if (state.chatsPanelOpen) {
saveActiveChatToStore();
const search = $('sa_chats_search');
if (search) {
search.value = state.chatsQuery || '';
search.focus();
}
renderChatsList();
}
}
async function startNewChat({ saveCurrent = true, force = false } = {}) {
if (!force && (state.busy || state.generating)) {
setStatus('Занято — дождись конца ответа или Стоп');
return;
}
if (force) {
// Drop in-flight reply so it cannot land in the new chat.
abortInFlightWork({ status: '' });
}
setChatsPanelOpen(false);
if (saveCurrent) {
saveActiveChatToStore({ dropEmpty: true });
}
state.sessionExact = {};
state.lastUserParamIntent = false;
state.pendingSilentGen = false;
state.lastPatch = null;
clearGenResults();
const chat = {
id: chatUid(),
title: 'Новый чат',
createdAt: Date.now(),
updatedAt: Date.now(),
messages: [],
params: snapshotChatParams(),
};
state.chats.unshift(chat);
state.activeChatId = chat.id;
state.history = [];
state.critiqueHopUsed = false;
state.visionHopUsed = false;
state.packUserTouched = false;
state.pendingPersonaNote = null;
if (state.streamEl) {
try { state.streamEl.remove(); } catch (e) { /* ignore */ }
state.streamEl = null;
}
syncBuildGenButton();
resetMessagesUi();
persistChatsStore();
updateSessionLabel();
renderBoard();
syncHistoryBadge();
renderChatsList();
setStatus('Новый чат — параметры Generate как сейчас');
maybeWelcome();
}
async function switchToChat(id) {
if (!id || id === state.activeChatId) {
setChatsPanelOpen(false);
return;
}
if (state.busy || state.generating) {
setStatus('Занято — нельзя сменить чат сейчас');
return;
}
saveActiveChatToStore({ dropEmpty: true });
let chat = findChat(id);
if (!chat || !(chat.messages || []).length) {
try {
const full = await diskPersist()?.getChat?.(id);
if (full) {
const idx = (state.chats || []).findIndex((c) => c.id === id);
if (idx >= 0) {
state.chats[idx] = full;
} else {
state.chats.unshift(full);
}
chat = full;
}
} catch (e) {
console.warn('Assistent: getChat failed', id, e);
}
}
if (!chat) {
setStatus('Чат не найден');
return;
}
state.activeChatId = chat.id;
state.history = slimHistoryMessages(chat.messages);
state.critiqueHopUsed = false;
state.visionHopUsed = false;
state.packUserTouched = false;
state.pendingPersonaNote = null;
state.pendingSilentGen = false;
state.lastUserParamIntent = false;
if (state.streamEl) {
try { state.streamEl.remove(); } catch (e) { /* ignore */ }
state.streamEl = null;
}
renderHistoryIntoUi(state.history);
const result = await restoreChatParams(chat.params);
updateSessionLabel();
syncHistoryBadge();
renderChatsList();
setChatsPanelOpen(false);
setView('chat');
if (result?.restored) {
setStatus(`Чат «${chat.title}» · параметры восстановлены`);
} else {
setStatus(`Чат «${chat.title}» · снимок параметров отсутствует — Generate не менялся`);
}
}
function deleteChat(id) {
if (!id) {
return;
}
const wasActive = id === state.activeChatId;
state.chats = state.chats.filter((c) => c.id !== id);
if (wasActive) {
state.activeChatId = null;
}
diskPersist()?.deleteChat(id)?.catch?.((e) => console.warn('Assistent: disk delete failed', e));
persistChatsStore();
syncHistoryBadge();
if (wasActive) {
startNewChat({ saveCurrent: false, force: true });
} else {
renderChatsList();
}
}
async function initChatSessions() {
loadChatsStore();
await loadChatsFromDisk();
// Always open a fresh chat on startup; past chats stay in History.
startNewChat({ saveCurrent: false, force: true });
syncHistoryBadge();
renderChatsList();
}
function persistHistory() {
if (state.restoringChat) {
return;
}
if (!state.activeChatId) {
const chat = {
id: chatUid(),
title: 'Новый чат',
createdAt: Date.now(),
updatedAt: Date.now(),
messages: [],
params: snapshotChatParams(),
};
state.chats.unshift(chat);
state.activeChatId = chat.id;
}
const chat = findChat(state.activeChatId);
if (!chat) {
return;
}
chat.messages = slimHistoryMessages(state.history);
chat.params = snapshotChatParams();
chat.updatedAt = Date.now();
chat.title = titleFromMessages(chat.messages);
persistChatsStore();
updateSessionLabel();
syncHistoryBadge();
}
function restoreHistory() {
// Replaced by initChatSessions — kept as no-op for safety.
}
function clearPersistedHistory() {
if (state.activeChatId) {
const chat = findChat(state.activeChatId);
if (chat) {
chat.messages = [];
chat.title = 'Новый чат';
chat.params = snapshotChatParams();
chat.updatedAt = Date.now();
}
persistChatsStore();
}
updateSessionLabel();
renderChatsList();
}
function clearChatHistory() {
abortInFlightWork({ status: '' });
state.history = [];
state.critiqueHopUsed = false;
state.visionHopUsed = false;
state.packUserTouched = false;
state.pendingPersonaNote = null;
state.sessionExact = {};
state.lastUserParamIntent = false;
state.pendingSilentGen = false;
state.lastPatch = null;
clearGenResults();
syncBuildGenButton();
clearPersistedHistory();
resetMessagesUi('Чат очищен
Сообщения сброшены. Параметры Generate на месте. + — новый чат в Историю, История — прошлые диалоги.
');
setStatus('Чат очищен');
updateSessionLabel();
syncHistoryBadge();
renderBoard();
}
function hideSlashMenu() {
const menu = $('sa_slash_menu');
if (menu) {
menu.hidden = true;
menu.innerHTML = '';
}
state.slashIndex = 0;
}
function slashMatches(text) {
const t = String(text || '');
if (!t.startsWith('/')) {
return [];
}
const q = t.toLowerCase();
return SLASH_COMMANDS.filter((c) => c.cmd.toLowerCase().startsWith(q) || q === '/' || c.cmd.toLowerCase().includes(q.slice(1)));
}
function renderSlashMenu(items) {
const menu = $('sa_slash_menu');
if (!menu) {
return;
}
if (!items.length) {
hideSlashMenu();
return;
}
menu.hidden = false;
menu.innerHTML = '';
state.slashIndex = Math.max(0, Math.min(state.slashIndex, items.length - 1));
items.forEach((item, i) => {
const btn = document.createElement('button');
btn.type = 'button';
btn.className = 'sa-slash-item' + (i === state.slashIndex ? ' sa-slash-active' : '');
btn.setAttribute('role', 'option');
btn.innerHTML = `${escapeHtml(item.cmd.trim())} — ${escapeHtml(item.hint)}`;
btn.addEventListener('mousedown', (e) => {
e.preventDefault();
applySlashPick(item);
});
menu.appendChild(btn);
});
}
function applySlashPick(item) {
const input = $('sa_input');
if (!input || !item) {
return;
}
input.value = item.cmd;
hideSlashMenu();
input.focus();
const pos = input.value.length;
input.setSelectionRange(pos, pos);
}
function updateSlashMenuFromInput() {
const text = $('sa_input')?.value || '';
if (!text.startsWith('/') || text.includes('\n') || /\s/.test(text.trim().slice(1)) && !text.endsWith(' ')) {
// show while typing command token only
const token = text.split(/\s/)[0] || '';
if (!token.startsWith('/') || (text.includes(' ') && !SLASH_COMMANDS.some((c) => c.cmd.startsWith(token)))) {
if (!(token.startsWith('/') && !text.includes(' '))) {
hideSlashMenu();
return;
}
}
}
const token = (text.split(/\s/)[0] || '');
if (!token.startsWith('/') || text.indexOf(' ') > 0) {
hideSlashMenu();
return;
}
renderSlashMenu(slashMatches(token));
}
function onPersonaChanged() {
const id = $('sa_persona')?.value || 'neutral';
state.sessionExact = {};
state.lastUserParamIntent = false;
saveSettings();
loadConfig(id, (data) => {
const title = data?.personas?.find((p) => p.id === id)?.title
|| (state.personas || []).find((p) => p.id === id)?.title
|| id;
if (data?.personas) {
state.personas = data.personas;
}
appendSystemNote(`Тон → ${title}`);
state.pendingPersonaNote = `Persona is now ${id} (${title}). Adopt this voice from now on.`;
if (data?.assistant?.default_pack && $('sa_pack') && !state.packUserTouched) {
const packId = data.assistant.default_pack;
if ([...($('sa_pack').options || [])].some((o) => o.value === packId)) {
$('sa_pack').value = packId;
}
}
fillEmptyParamsFromExact();
renderPersonaControls(data?.controls || {}, data?.control_values || data?.exact?.controls || {});
syncPersonaDeleteButton(data?.persona_source || data?.personas?.find((p) => p.id === id)?.source);
if (state.view === 'settings') {
if (state.settingsTab === 'user') {
refreshUserPrefs();
}
if (state.settingsTab === 'craft') {
renderMemoryList();
}
if (state.settingsTab === 'more') {
fillKnobsFromConfig(data);
}
}
});
}
function countPromptImages() {
try {
const box = document.getElementById('alt_prompt_textbox') || document.getElementById('input_prompt');
if (!box) {
return 0;
}
const text = box.value || '';
const matches = text.match(/Крафт-память пуста — карточки, seed и патчи
';
return;
}
for (const row of rows) {
const el = document.createElement('div');
el.className = 'sa-mem-row';
const bundled = row.source === 'bundled';
const when = row.updated ? formatChatWhen(row.updated * 1000) : '';
const scope = row.scope === 'personal' ? `персона ${row.persona || '—'}` : 'общая';
el.innerHTML = `memory_upsert.${escapeHtml(row.kind || 'note')}${escapeHtml(row.key || '')}
${escapeHtml(clipDebug(row.text, 220))}
Читаю память…
';
}
genericRequest(
'AssistentListMemory',
{ limit: 200 },
(data) => {
state.memoryRows = Array.isArray(data?.memories) ? data.memories : [];
renderMemoryKinds(data?.kinds || []);
renderMemoryList();
const foot = $('sa_mem_total');
if (foot) {
foot.textContent = `Всего: ${data?.total ?? state.memoryRows.length} · ${data?.embed_model || '—'}`;
}
},
0,
(err) => {
if (list) {
list.innerHTML = `Память недоступна: ${escapeHtml(String(err || 'ошибка'))}
`;
}
},
);
}
function forgetMemory(row) {
if (!row?.kind || !row?.key || typeof genericRequest !== 'function') {
return;
}
genericRequest(
'AssistentForgetMemory',
{
kind: row.kind,
key: row.key,
source: row.source || '',
scope: row.scope || '',
persona: row.scope === 'personal' ? (row.persona || '') : '',
},
() => {
state.memoryRows = (state.memoryRows || []).filter((m) => !(m.kind === row.kind && m.key === row.key && m.source === row.source && m.persona === row.persona));
renderMemoryList();
setStatus(`Забыто: ${row.kind}/${row.key}`);
},
0,
(err) => setStatus(String(err || 'Не удалось забыть')),
);
}
function clearCraftMemory(opts = {}) {
if (typeof genericRequest !== 'function') {
return;
}
const label = opts.label || 'крафт-память';
if (!window.confirm(`Очистить ${label}? Bundled seed останется.`)) {
return;
}
const body = {
scope: opts.scope || '',
kind: opts.kind || '',
persona: opts.persona || '',
};
genericRequest(
'AssistentClearMemory',
body,
(data) => {
setStatus(`Удалено крафт-записей: ${data?.deleted ?? 0}`);
refreshMemoryList();
},
0,
(err) => setStatus(String(err || 'Очистка не удалась')),
);
}
function setSettingsTab(id) {
state.settingsTab = id || 'behavior';
document.querySelectorAll('#sa_settings .sa-stab').forEach((btn) => {
const on = btn.getAttribute('data-stab') === state.settingsTab;
btn.classList.toggle('sa-stab-active', on);
btn.setAttribute('aria-selected', on ? 'true' : 'false');
});
document.querySelectorAll('#sa_settings .sa-spane').forEach((pane) => {
pane.hidden = pane.getAttribute('data-spane') !== state.settingsTab;
});
if (state.settingsTab === 'craft') {
refreshMemoryList();
refreshWantedQueue();
}
if (state.settingsTab === 'user') {
refreshUserPrefs();
}
if (state.settingsTab === 'personas') {
renderPersonaSettingsList();
}
if (state.settingsTab === 'models') {
syncSettingsHealthLine();
const m = $('sa_model')?.value;
if (m && $('sa_settings_chat_model')) {
$('sa_settings_chat_model').value = m;
}
}
if (state.settingsTab === 'more') {
fillKnobsFromConfig(state.config);
}
}
function fillKnobsFromConfig(data) {
const asst = data?.assistant || state.config?.assistant || {};
const exact = data?.exact || state.config?.exact || state.exact || {};
const setNum = (id, v) => {
const el = $(id);
if (el && v != null && Number.isFinite(Number(v))) {
el.value = String(v);
}
};
setNum('sa_num_ctx', asst.num_ctx);
setNum('sa_history_keep', asst.history_keep_turns);
setNum('sa_memory_top_k', asst.memory_top_k);
const w = asst.user_prefs_weight != null ? Number(asst.user_prefs_weight) : 1;
const weightEl = $('sa_user_prefs_weight');
if (weightEl) {
weightEl.value = String(Math.max(0, Math.min(1.5, w)));
const lab = $('sa_user_prefs_weight_val');
if (lab) {
lab.textContent = Number(weightEl.value).toFixed(1);
}
}
const turbo = exact.profiles?.turbo || {};
const raw = exact.profiles?.raw || {};
setNum('sa_exact_turbo_steps', turbo.steps);
setNum('sa_exact_turbo_cfg', turbo.cfg);
setNum('sa_exact_turbo_sigma', turbo.sigma_shift);
setNum('sa_exact_raw_steps', raw.steps);
setNum('sa_exact_raw_cfg', raw.cfg);
setNum('sa_exact_raw_sigma', raw.sigma_shift);
}
function saveKnobs() {
if (typeof genericRequest !== 'function') {
return;
}
const num = (id) => {
const v = parseFloat($(id)?.value);
return Number.isFinite(v) ? v : null;
};
const assistant = {
num_ctx: num('sa_num_ctx'),
history_keep_turns: num('sa_history_keep'),
memory_top_k: num('sa_memory_top_k'),
user_prefs_weight: num('sa_user_prefs_weight'),
};
Object.keys(assistant).forEach((k) => {
if (assistant[k] == null) {
delete assistant[k];
}
});
const exact = {
profiles: {
turbo: {
steps: num('sa_exact_turbo_steps'),
cfg: num('sa_exact_turbo_cfg'),
sigma_shift: num('sa_exact_turbo_sigma'),
},
raw: {
steps: num('sa_exact_raw_steps'),
cfg: num('sa_exact_raw_cfg'),
sigma_shift: num('sa_exact_raw_sigma'),
},
},
};
genericRequest(
'AssistentSaveKnobs',
{ assistant, exact },
(data) => {
if (data?.assistant || data?.exact) {
applyConfigPayload({
...state.config,
assistant: data.assistant || state.config?.assistant,
exact: data.exact || state.config?.exact,
});
}
setStatus('Knobs сохранены в overlay');
},
0,
(err) => setStatus(String(err || 'Не удалось сохранить knobs')),
);
}
function syncSettingsHealthLine() {
const line = $('sa_settings_health_line');
const badge = $('sa_ollama_health');
if (line && badge) {
line.textContent = badge.textContent || 'Ollama · …';
line.className = 'sa-settings-health ' + (badge.className || '').replace('sa-health', '').trim();
}
}
function personaSourceLabel(source) {
if (source === 'overlay') {
return 'моя';
}
if (source === 'overlay+bundled') {
return 'встроено+правка';
}
return 'встроено';
}
function renderPersonaSettingsList() {
const root = $('sa_persona_list');
if (!root) {
return;
}
const list = state.personas || state.config?.personas || [];
const cur = state.settingsPersonaId || $('sa_persona')?.value || list[0]?.id;
state.settingsPersonaId = cur;
root.innerHTML = '';
for (const p of list) {
const btn = document.createElement('button');
btn.type = 'button';
btn.className = 'sa-persona-item' + (p.id === cur ? ' sa-persona-item-active' : '');
const accent = p.accent || 'currentColor';
btn.innerHTML = `${escapeHtml(p.title || p.id)}
${escapeHtml(personaSourceLabel(p.source))}
`;
btn.addEventListener('click', () => {
state.settingsPersonaId = p.id;
renderPersonaSettingsList();
loadPersonaPreview(p.id);
});
root.appendChild(btn);
}
syncPersonaPanelActions();
if (cur) {
loadPersonaPreview(cur);
}
}
function syncPersonaPanelActions() {
const id = state.settingsPersonaId;
const p = (state.personas || []).find((x) => x.id === id);
const canDelete = p && (p.source === 'overlay' || p.source === 'overlay+bundled');
const del = $('sa_btn_persona_delete_panel');
if (del) {
del.disabled = !canDelete;
}
}
function loadPersonaPreview(id) {
const box = $('sa_persona_preview');
if (!box || typeof genericRequest !== 'function') {
return;
}
box.innerHTML = 'Загрузка…
';
genericRequest(
'AssistentGetPersonaShelves',
{ persona: id },
(data) => {
const summary = data?.identity_summary || '';
const src = data?.source || '';
box.textContent = `${id} · ${personaSourceLabel(src)}\n\n${summary || '(пусто)'}`;
syncPersonaPanelActions();
},
0,
(err) => {
box.innerHTML = `${escapeHtml(String(err || 'ошибка'))}
`;
},
);
}
function exportSelectedPersona() {
const id = state.settingsPersonaId || $('sa_persona')?.value;
if (!id || typeof genericRequest !== 'function') {
return;
}
genericRequest(
'AssistentExportPersona',
{ persona: id },
(data) => {
const pack = data?.pack;
if (!pack) {
setStatus('Пустой экспорт');
return;
}
const blob = new Blob([JSON.stringify(pack, null, 2)], { type: 'application/json' });
const a = document.createElement('a');
a.href = URL.createObjectURL(blob);
a.download = `${pack.id || id}.assistent-persona.json`;
a.click();
URL.revokeObjectURL(a.href);
setStatus(`Экспорт: ${a.download}`);
},
0,
(err) => setStatus(String(err || 'Экспорт не удался')),
);
}
function importPersonaFile(file) {
if (!file || typeof genericRequest !== 'function') {
return;
}
const reader = new FileReader();
reader.onload = () => {
let pack;
try {
pack = JSON.parse(String(reader.result || ''));
} catch (e) {
setStatus('Невалидный JSON');
return;
}
let newId = pack?.id || '';
if ((state.personas || []).some((p) => p.id === newId && (p.source === 'bundled' || p.source === 'overlay+bundled'))) {
newId = window.prompt('Id занят bundled — новый id:', `${newId}_import`) || '';
}
genericRequest(
'AssistentImportPersona',
{ pack, new_id: newId || null, overwrite: false },
(data) => {
if (Array.isArray(data?.personas)) {
state.personas = data.personas;
renderPersonaOptions(data.personas, data.persona?.id);
}
state.settingsPersonaId = data?.persona?.id || newId;
renderPersonaSettingsList();
setStatus(`Импорт: ${data?.persona?.id || newId}`);
},
0,
(err) => setStatus(String(err || 'Импорт не удался')),
);
};
reader.readAsText(file);
}
function cloneSelectedPersona() {
const from = state.settingsPersonaId || $('sa_persona')?.value;
if (!from) {
return;
}
const to = window.prompt('Новый id личности:', `${from}_copy`);
if (!to) {
return;
}
genericRequest(
'AssistentClonePersona',
{ from, to, title: to },
(data) => {
if (Array.isArray(data?.personas)) {
state.personas = data.personas;
renderPersonaOptions(data.personas, to);
}
state.settingsPersonaId = to;
renderPersonaSettingsList();
setStatus(`Клон: ${to}`);
},
0,
(err) => setStatus(String(err || 'Клон не удался')),
);
}
function deleteSelectedOverlayPersona() {
const id = state.settingsPersonaId;
const p = (state.personas || []).find((x) => x.id === id);
if (!p || (p.source !== 'overlay' && p.source !== 'overlay+bundled')) {
setStatus('Можно удалить только overlay');
return;
}
if (!window.confirm(`Удалить overlay-личность «${id}»?`)) {
return;
}
genericRequest(
'AssistentDeletePersona',
{ persona: id },
(data) => {
if (Array.isArray(data?.personas)) {
state.personas = data.personas;
renderPersonaOptions(data.personas, data.default_persona);
}
state.settingsPersonaId = data?.default_persona || null;
renderPersonaSettingsList();
setStatus(`Удалено: ${id}`);
},
0,
(err) => setStatus(String(err || 'Удаление не удалось')),
);
}
function refreshUserPrefs() {
if (typeof genericRequest !== 'function') {
return;
}
const persona = $('sa_persona')?.value || 'neutral';
genericRequest(
'AssistentListUserPrefs',
{ persona, limit: 200 },
(data) => {
state.userPrefs = Array.isArray(data?.prefs) ? data.prefs : [];
renderUserPrefsLists();
},
0,
(err) => setStatus(String(err || 'User prefs недоступны')),
);
}
function renderUserPrefsLists() {
const persona = $('sa_persona')?.value || 'neutral';
const global = (state.userPrefs || []).filter((p) => p.scope === 'global');
const personal = (state.userPrefs || []).filter((p) => p.scope === 'persona' && (p.persona_id === persona || p.persona === persona));
const fill = (rootId, rows) => {
const root = $(rootId);
if (!root) {
return;
}
root.innerHTML = '';
if (!rows.length) {
root.innerHTML = 'Пусто
';
return;
}
for (const row of rows) {
const el = document.createElement('div');
el.className = 'sa-mem-row';
const pin = row.pinned ? ' ★' : '';
el.innerHTML = `${escapeHtml(row.key || '')}${pin}
${escapeHtml(clipDebug(row.text, 200))}
Inventory пуст — Обновить.
';
return;
}
for (const row of rows) {
const btn = document.createElement('button');
btn.type = 'button';
btn.className = 'sa-card-row';
if (state.cardsSelection && state.cardsSelection.kind === row.kind && state.cardsSelection.name === row.name) {
btn.classList.add('sa-selected');
}
const thumb = row.preview_url
? `${escapeHtml(row.kind)}
${escapeHtml(row.title || row.name)}
Нет LoRA
';
}
};
filter.addEventListener('input', draw);
draw();
const composer = $('sa_composer') || document.body;
composer.style.position = composer.style.position || 'relative';
composer.appendChild(picker);
const onDoc = (ev) => {
if (!picker.contains(ev.target) && ev.target !== anchor) {
picker.remove();
document.removeEventListener('mousedown', onDoc);
}
};
setTimeout(() => document.addEventListener('mousedown', onDoc), 0);
filter.focus();
}
function loadTasteFromServer() {
if (typeof genericRequest !== 'function') {
return;
}
genericRequest(
'AssistentGetTaste',
{},
(data) => {
const remote = data?.taste;
if (!remote || typeof remote !== 'object') {
return;
}
const remoteUpdated = remote.updated || 0;
const localUpdated = state.taste?.updated || 0;
// sqlite taste is the source of truth; localStorage only wins when it is strictly newer.
const localEmpty = !localUpdated
&& !(state.taste?.styles?.length || state.taste?.likes?.length || state.taste?.avoid?.length);
if (localEmpty || remoteUpdated >= localUpdated) {
state.taste = {
styles: Array.isArray(remote.styles) ? remote.styles.slice(0, 12) : [],
likes: Array.isArray(remote.likes) ? remote.likes.slice(0, 16) : [],
avoid: Array.isArray(remote.avoid) ? remote.avoid.slice(0, 12) : [],
notes: String(remote.notes || '').slice(0, 400),
updated: remoteUpdated || Date.now(),
};
saveTasteLocalOnly();
}
},
0,
() => {},
);
}
function saveTasteLocalOnly() {
try {
localStorage.setItem(LS_TASTE, JSON.stringify(state.taste || {}));
} catch (e) { /* ignore */ }
}
function saveTasteToServerDebounced() {
if (state.tasteSaveTimer) {
clearTimeout(state.tasteSaveTimer);
}
state.tasteSaveTimer = setTimeout(() => {
if (typeof genericRequest !== 'function') {
return;
}
genericRequest(
'AssistentSaveTaste',
{ taste: state.taste || {} },
() => {},
0,
() => {},
);
}, 800);
}
async function generateCardWithAssistent() {
const sel = state.cardsSelection;
if (!sel) {
setCardStatus('Выбери модель');
return;
}
if (state.busy) {
setCardStatus('Чат занят');
return;
}
setPackValue('catalog_card', { flash: true });
setView('chat');
const meta = await new Promise((resolve) => {
genericRequest(
'AssistentGetCardMeta',
{ kind: sel.kind, name: sel.name },
(data) => resolve(data),
0,
() => resolve(null),
);
});
const forced = `Write a recommendation card for this ${sel.kind}: ${sel.name}. Use metadata/triggers only; output one JSON card.`;
await sendChat({
forcedUserText: forced,
skipSlash: true,
skipAutoPack: true,
fromCards: true,
cardTarget: {
kind: sel.kind,
name: sel.name,
meta,
},
});
}
function inventoryIsStale(maxAgeMs = 20000) {
if (!state.inventoryFetchedAt) {
return true;
}
return (Date.now() - state.inventoryFetchedAt) > maxAgeMs;
}
async function ensureFreshInventory({ forceRescan } = {}) {
const rescan = forceRescan || inventoryIsStale(20000);
await refreshInventoryAsync({ rescan });
}
function triggerSwarmModelRefresh(done) {
if (typeof genericRequest !== 'function') {
if (done) {
done();
}
return;
}
genericRequest(
'TriggerRefresh',
{ strong: true },
() => {
if (done) {
done();
}
},
0,
() => {
if (done) {
done();
}
},
);
}
async function handleReplySideEffects(reply, civitaiResults, opts = {}) {
const { fromAutoCritique, fromVisionHop, fromCards, fromDebug } = opts;
if (fromCards) {
const card = extractCardJson(reply);
if (card) {
if ($('sa_card_json')) {
$('sa_card_json').value = JSON.stringify(card, null, 2);
}
if (opts.fromDownload || opts.cardTarget) {
const kind = card.kind || opts.cardTarget?.kind || 'lora';
const name = card.name || opts.cardTarget?.name;
if (name && typeof genericRequest === 'function') {
genericRequest(
'AssistentSaveCard',
{ kind, name, card, enqueue_wanted: false },
(data) => {
if (data?.path) {
state.modelCards[`${kind}:${name}`] = card;
setCardStatus(data.installed ? `Card saved → ${data.path}` : `Card draft → ${data.path}`);
setStatus(`Card saved for ${name}`);
}
},
0,
() => setCardStatus('Card draft ready — Save manually'),
);
}
} else {
setView('cards');
setCardStatus('Draft from Assistent — review & Save');
}
}
return;
}
const { patch } = extractPatch(reply);
let effective = patch;
if (!effective && !fromVisionHop && !fromAutoCritique && !fromDebug) {
const aspect = parseAspectFromUserText(opts.userText || '');
if (aspect && (replyMissingJsonPatch(reply) || isSameButAspectRequest(opts.userText || ''))) {
effective = { aspect, actions: ['generate'] };
if (state.lastPatch?.prompt) {
effective.prompt = state.lastPatch.prompt;
}
appendSystemNote(`Патч пустой — применил aspect ${aspect} сам.`);
}
}
if (!effective && !fromVisionHop && !fromAutoCritique && !fromDebug) {
const synthesized = synthesizePatchAfterEmptyFence(reply, opts.userText || '', opts);
if (synthesized) {
effective = synthesized;
appendSystemNote('Патч пустой — собрал prompt из ответа и запустил Generate.');
}
}
if (!effective && !fromVisionHop && !fromAutoCritique && !fromDebug
&& !opts.fromEmptyPatchRetry
&& replyMissingJsonPatch(reply)
&& (opts.userWantsGenerate || state.pendingSilentGen || userAsksContinue(opts.userText || ''))) {
appendSystemNote('Патч пустой — прошу модель дописать JSON.');
await sendChat({
skipSlash: true,
skipAutoPack: true,
fromEmptyPatchRetry: true,
userWantsGenerate: true,
forcedUserText:
'Ты написал «JSON Patch» без fenced ```json```. '
+ 'Сейчас ответь ТОЛЬКО одним fenced JSON объектом: '
+ '{"prompt":"