1969 lines
88 KiB
Plaintext
1969 lines
88 KiB
Plaintext
import { app } from "../../scripts/app.js";
|
||
|
||
const i18n = {
|
||
zh: {
|
||
title: '📦 Anomalous 模型浏览器',
|
||
scan: '扫描',
|
||
scanTitle: '扫描目录',
|
||
scanConfirm: '注意:\n扫描过程将会自动比对 Civitai 并重命名本地模型，同时清理损坏文件。\n确定要开始扫描吗？',
|
||
scanning: '扫描中...',
|
||
scanBg: '🚀 扫描后台启动！',
|
||
scanDone: '完成',
|
||
scanCompleteMsg: '✅ 扫描完成！\n\n⚠️ 提示：由于部分模型已被重命名或去重，请点击 ComfyUI 的 [Refresh] 按钮以同步最新的模型列表。',
|
||
eco: '节能',
|
||
autoPlay: '自动播放',
|
||
togglePlayTitle: '切换播放模式',
|
||
clean: '清理',
|
||
cleanTitle: '清理重复 Info',
|
||
cleanConfirm: '将扫描并删除所有无用的 .civitai.info 残留文件。是否继续？',
|
||
cleaning: '清理中',
|
||
cleanDone: '✅ 清理完毕！删除了',
|
||
files: '个文件。',
|
||
cleanFail: '❌ 清理失败: ',
|
||
cleanErr: '❌ 清理出错: ',
|
||
folders: '📂 文件夹',
|
||
collapseAll: '➖ 收起全部',
|
||
expandAll: '➕ 展开全部',
|
||
noModels: '此文件夹中没有找到模型。',
|
||
noPreview: '暂无预览图',
|
||
clickScan: '点击扫描',
|
||
back: '⬅️ 返回网格',
|
||
delModel: '🗑️ 删除',
|
||
delConfirm: '确定彻底删除',
|
||
delConfirm2: '及其所有关联配置、预览图和缓存文件吗？此操作不可逆！',
|
||
delSuccess: '✅ 删除成功！\n',
|
||
delNote: '\n\n⚠️ 提示：请务必【重启 ComfyUI 服务端】。如果不重启，继续使用可能会报错！',
|
||
delFail: '❌ 删除失败: ',
|
||
copyAll: '📋 复制全部',
|
||
copied: '✅ 已复制!',
|
||
clickToCopy: '点击复制: ',
|
||
compatibleModels: '🔗 兼容模型',
|
||
loadingCompatible: '加载中...',
|
||
backToPrev: '🔙 返回上一层',
|
||
applyToCanvas: '➕ 插入节点',
|
||
applySuccess: '✅ 已添加至工作流',
|
||
help: '帮助',
|
||
helpTitle: '📖 使用说明',
|
||
helpContent: `
|
||
<div style="line-height: 1.6; font-size: 0.95em; color: #eee; padding: 10px;">
|
||
<p><strong>1. 🔑 API Key (Civitai)</strong>: 在顶部输入你的 API Key，可用于扫描时获取限制级 (NSFW) 模型的封面和信息。</p>
|
||
<p><strong>2. 🔄 扫描 (Scan)</strong>: 自动扫描当前选中的文件夹，从 C 站匹配并下载封面图，同时规范化重命名你的本地模型文件。</p>
|
||
<p><strong>3. 🧹 清理 (Clean Info)</strong>: 全局深度清理：自动扫描并删除所有因为模型被删而遗留的无效 .civitai.info 文件。</p>
|
||
<p><strong>4. 🔋 节能 / 🎬 自动播放</strong>: 切换视频封面的播放行为。“节能”模式下，鼠标悬浮在卡片上时才会播放预览视频。</p>
|
||
<p><strong>5. 🗂️ 详情与删除</strong>: 点击任何模型卡片，可查看介绍并<strong>一键复制触发词</strong>。也可以在此处彻底删除模型及关联配置（删除后须重启 ComfyUI 引擎生效）。</p>
|
||
<p><strong>6. 📂 目录折叠</strong>: 点击左侧边栏的“收起/展开全部”快速管理网格视图。</p>
|
||
<p><strong>7. 🔗 兼容模型匹配</strong>: 点开大模型或 Lora 详细页，系统将自动基于 Base Model 架构，双向匹配并展示关联模型。</p>
|
||
<p><strong>8. ➕ 一键投放节点</strong>: 在网格卡片右上角或详细页顶部，点击【➕】按钮，可将当前模型节点直接贴在鼠标上，并无缝插入工作流画布！</p>
|
||
</div>`,
|
||
closeHelp: '关闭说明',
|
||
|
||
},
|
||
en: {
|
||
title: '📦 Anomalous Model Browser',
|
||
scan: 'Scan',
|
||
scanTitle: 'Scan Folder',
|
||
scanConfirm: 'Notice:\nThe scan process will automatically compare with Civitai and rename your local files.\nStart scan?',
|
||
scanning: 'Scanning...',
|
||
scanBg: '🚀 Scan started in background!',
|
||
scanDone: 'Done',
|
||
scanCompleteMsg: '✅ Scan Complete!\n\n⚠️ Note: Please click [Refresh] to sync the model list since files were renamed.',
|
||
eco: 'Eco',
|
||
autoPlay: 'AutoPlay',
|
||
togglePlayTitle: 'Toggle Play Mode',
|
||
clean: 'Clean Info',
|
||
cleanTitle: 'Clean Duplicate Info',
|
||
cleanConfirm: 'This will globally scan and delete all redundant .civitai.info files. Continue?',
|
||
cleaning: 'Cleaning',
|
||
cleanDone: '✅ Clean Complete! Deleted',
|
||
files: 'files.',
|
||
cleanFail: '❌ Failed: ',
|
||
cleanErr: '❌ Error: ',
|
||
folders: '📂 Folders',
|
||
collapseAll: '➖ Collapse All',
|
||
expandAll: '➕ Expand All',
|
||
noModels: 'No models found in this folder.',
|
||
noPreview: 'No preview available',
|
||
clickScan: 'Click Scan',
|
||
back: '⬅️ Back to grid',
|
||
delModel: '🗑️ Delete',
|
||
delConfirm: 'Are you sure you want to permanently delete',
|
||
delConfirm2: ' and all its associated configs, previews, and cache files? This action is irreversible!',
|
||
delSuccess: '✅ Delete Complete!\n',
|
||
delNote: '\n\n⚠️ Note: Please restart the ComfyUI backend server, otherwise errors may occur!',
|
||
delFail: '❌ Delete Failed: ',
|
||
copyAll: '📋 Copy All',
|
||
copied: '✅ Copied!',
|
||
clickToCopy: 'Click to copy: ',
|
||
compatibleModels: '🔗 Compatible Models',
|
||
loadingCompatible: 'Loading...',
|
||
backToPrev: '🔙 Back to prev',
|
||
applyToCanvas: '➕ Add Node',
|
||
applySuccess: '✅ Added to workflow',
|
||
help: 'Help',
|
||
helpTitle: '📖 User Manual',
|
||
helpContent: `
|
||
<div style="line-height: 1.6; font-size: 0.95em; color: #eee; padding: 10px;">
|
||
<p><strong>1. 🔑 API Key (Civitai)</strong>: Enter your key at the top to fetch metadata and previews for NSFW models during scanning.</p>
|
||
<p><strong>2. 🔄 Scan</strong>: Automatically scans the current folder, fetches Civitai metadata, downloads previews, and standardizes model filenames.</p>
|
||
<p><strong>3. 🧹 Clean Info</strong>: Globally scans and deletes orphaned .civitai.info files to free up disk space.</p>
|
||
<p><strong>4. 🔋 Eco / 🎬 AutoPlay</strong>: Toggle video playback modes. Eco mode plays videos only when hovering over a model card.</p>
|
||
<p><strong>5. 🗂️ Details & Delete</strong>: Click a model card to view details and <strong>1-click copy trained words</strong>. You can also permanently delete the model here (requires ComfyUI restart).</p>
|
||
<p><strong>6. 📂 Folder Toggle</strong>: Use the Collapse/Expand All button in the sidebar to manage your view.</p>
|
||
</div>`,
|
||
closeHelp: 'Close Manual',
|
||
|
||
}
|
||
};
|
||
|
||
let currentLang = localStorage.getItem('anomalous_lang') || 'zh';
|
||
const t = (key) => i18n[currentLang][key] || key;
|
||
|
||
class AnomalousBrowser {
|
||
constructor() {
|
||
this.modal = null;
|
||
this.sidebar = null;
|
||
this.grid = null;
|
||
this.detailPanel = null;
|
||
this.currentType = 'loras';
|
||
this.currentPathIdx = 0;
|
||
this.currentSubfolder = '/';
|
||
this.foldersData = null;
|
||
this.expandedFolders = new Set(['/', 'checkpoints', 'loras', 'unet', 'diffusion_models']);
|
||
this.energySaving = localStorage.getItem('anomalous_energy_saving') === 'true';
|
||
this.createDOM();
|
||
}
|
||
|
||
createDOM() {
|
||
this.modal = document.createElement('div');
|
||
this.modal.id = 'anomalous-modal';
|
||
this.modal.onclick = (e) => { if (e.target === this.modal) this.close(); };
|
||
|
||
const container = document.createElement('div');
|
||
container.id = 'anomalous-container';
|
||
|
||
const updateLangClass = () => {
|
||
if (currentLang === 'en') container.classList.add('anomalous-lang-en');
|
||
else container.classList.remove('anomalous-lang-en');
|
||
};
|
||
updateLangClass();
|
||
|
||
// Sidebar
|
||
this.sidebarWrapper = document.createElement('div');
|
||
this.sidebarWrapper.id = 'anomalous-sidebar-wrapper';
|
||
|
||
this.sidebar = document.createElement('div');
|
||
this.sidebar.id = 'anomalous-sidebar';
|
||
|
||
this.settingsArea = document.createElement('div');
|
||
this.settingsArea.id = 'anomalous-settings';
|
||
|
||
// API key moved to header
|
||
|
||
const langBtn = document.createElement('button');
|
||
langBtn.style.padding = '6px';
|
||
langBtn.style.width = '100%';
|
||
langBtn.style.background = '#444';
|
||
langBtn.style.color = '#fff';
|
||
langBtn.style.border = '1px solid #555';
|
||
langBtn.style.borderRadius = '4px';
|
||
langBtn.style.cursor = 'pointer';
|
||
langBtn.innerHTML = currentLang === 'zh' ? '🌐 Language: EN' : '🌐 语言: 中文';
|
||
langBtn.onclick = () => {
|
||
currentLang = currentLang === 'zh' ? 'en' : 'zh';
|
||
localStorage.setItem('anomalous_lang', currentLang);
|
||
langBtn.innerHTML = currentLang === 'zh' ? '🌐 Language: EN' : '🌐 语言: 中文';
|
||
// re-render UI
|
||
updateLangClass();
|
||
title.innerHTML = t('title');
|
||
scanBtn.title = t('scanTitle');
|
||
scanBtn.innerHTML = `🔄 <span class="anomalous-btn-text">${t('scan')}</span>`;
|
||
cleanBtn.title = t('cleanTitle');
|
||
cleanBtn.innerHTML = `🧹 <span class="anomalous-btn-text">${t('clean')}</span>`;
|
||
helpBtn.title = t('helpTitle');
|
||
helpBtn.innerHTML = `❓ <span class="anomalous-btn-text">${t('help')}</span>`;
|
||
renderEnergyBtn();
|
||
this.renderSidebar();
|
||
this.loadModels();
|
||
if (this.detailPanel.style.display !== 'none' && this.currentDetailModel) {
|
||
this.showDetail(this.currentDetailModel);
|
||
}
|
||
};
|
||
this.settingsArea.appendChild(langBtn);
|
||
|
||
const transSettingsTitle = document.createElement('div');
|
||
transSettingsTitle.innerHTML = '<b style="color:#aaa; font-size:0.85em;">🌐 Translation API</b>';
|
||
transSettingsTitle.style.marginTop = '15px';
|
||
transSettingsTitle.style.marginBottom = '5px';
|
||
this.settingsArea.appendChild(transSettingsTitle);
|
||
|
||
const engineSel = document.createElement('select');
|
||
engineSel.innerHTML = '<option value="google">Google Translate (Free)</option><option value="deepl">DeepL API</option>';
|
||
engineSel.style.width = '100%';
|
||
engineSel.style.padding = '4px';
|
||
engineSel.style.marginBottom = '5px';
|
||
engineSel.style.background = '#333';
|
||
engineSel.style.color = '#fff';
|
||
this.settingsArea.appendChild(engineSel);
|
||
|
||
const deeplKeyInput = document.createElement('input');
|
||
deeplKeyInput.type = 'password';
|
||
deeplKeyInput.placeholder = 'DeepL API Key (if DeepL selected)';
|
||
deeplKeyInput.style.width = '100%';
|
||
deeplKeyInput.style.padding = '4px';
|
||
deeplKeyInput.style.marginBottom = '5px';
|
||
deeplKeyInput.style.background = '#333';
|
||
deeplKeyInput.style.color = '#fff';
|
||
this.settingsArea.appendChild(deeplKeyInput);
|
||
|
||
const targetSel = document.createElement('select');
|
||
targetSel.innerHTML = '<option value="zh-CN">Chinese (Simplified)</option><option value="zh-TW">Chinese (Traditional)</option><option value="ja">Japanese</option><option value="ko">Korean</option><option value="es">Spanish</option>';
|
||
targetSel.style.width = '100%';
|
||
targetSel.style.padding = '4px';
|
||
targetSel.style.marginBottom = '10px';
|
||
targetSel.style.background = '#333';
|
||
targetSel.style.color = '#fff';
|
||
this.settingsArea.appendChild(targetSel);
|
||
|
||
const saveConfigBtn = document.createElement('button');
|
||
saveConfigBtn.innerHTML = '💾 Save Settings';
|
||
saveConfigBtn.style.width = '100%';
|
||
saveConfigBtn.style.padding = '6px';
|
||
saveConfigBtn.style.background = '#28a745';
|
||
saveConfigBtn.style.color = '#fff';
|
||
saveConfigBtn.style.border = 'none';
|
||
saveConfigBtn.style.borderRadius = '4px';
|
||
saveConfigBtn.style.cursor = 'pointer';
|
||
saveConfigBtn.onclick = async () => {
|
||
const cfg = {
|
||
trans_engine: engineSel.value,
|
||
deepl_key: deeplKeyInput.value,
|
||
trans_target: targetSel.value
|
||
};
|
||
await fetch('/anomalous/save_config', { method:'POST', body: JSON.stringify(cfg) });
|
||
alert('Settings saved!');
|
||
};
|
||
this.settingsArea.appendChild(saveConfigBtn);
|
||
|
||
fetch('/anomalous/config').then(r => r.json()).then(cfg => {
|
||
if(cfg.trans_engine) engineSel.value = cfg.trans_engine;
|
||
if(cfg.deepl_key) deeplKeyInput.value = cfg.deepl_key;
|
||
if(cfg.trans_target) targetSel.value = cfg.trans_target;
|
||
}).catch(e=>{});
|
||
|
||
this.sidebarWrapper.appendChild(this.sidebar);
|
||
this.sidebarWrapper.appendChild(this.settingsArea);
|
||
|
||
// Content Area
|
||
const content = document.createElement('div');
|
||
content.id = 'anomalous-content';
|
||
|
||
const header = document.createElement('div');
|
||
header.id = 'anomalous-header';
|
||
|
||
const title = document.createElement('div');
|
||
title.id = 'anomalous-title';
|
||
title.innerHTML = t('title');
|
||
|
||
|
||
const helpBtn = document.createElement('button');
|
||
helpBtn.id = 'anomalous-help-btn';
|
||
helpBtn.title = t('helpTitle');
|
||
helpBtn.innerHTML = `❓ <span class="anomalous-btn-text">${t('help')}</span>`;
|
||
helpBtn.onclick = () => this.showHelp();
|
||
|
||
const apiKeyInput = document.createElement('input');
|
||
apiKeyInput.id = 'anomalous-api-key';
|
||
apiKeyInput.placeholder = '🔑 API Key';
|
||
apiKeyInput.title = '输入 Civitai API Key / Enter Civitai API Key';
|
||
apiKeyInput.type = 'password';
|
||
apiKeyInput.value = localStorage.getItem('anomalous_api_key') || '';
|
||
apiKeyInput.onchange = async () => {
|
||
localStorage.setItem('anomalous_api_key', apiKeyInput.value);
|
||
try {
|
||
await fetch('/anomalous/save_config', {
|
||
method: 'POST',
|
||
headers: { 'Content-Type': 'application/json' },
|
||
body: JSON.stringify({ api_key: apiKeyInput.value })
|
||
});
|
||
} catch(e) {}
|
||
};
|
||
fetch('/anomalous/config').then(r=>r.json()).then(d=>{
|
||
if (d.has_api_key) apiKeyInput.placeholder = 'API Key Saved';
|
||
}).catch(()=>{});
|
||
|
||
const scanBtn = document.createElement('button');
|
||
scanBtn.id = 'anomalous-scan-btn';
|
||
scanBtn.title = t('scanTitle');
|
||
scanBtn.innerHTML = `🔄 <span class="anomalous-btn-text">${t('scan')}</span>`;
|
||
scanBtn.onclick = async () => {
|
||
if (!confirm(t('scanConfirm'))) return;
|
||
scanBtn.innerHTML = `⏳ <span class="anomalous-btn-text">${t('scanning')}</span>`;
|
||
scanBtn.disabled = true;
|
||
try {
|
||
const params = new URLSearchParams({ type: this.currentType, path_idx: this.currentPathIdx, subfolder: this.currentSubfolder });
|
||
const res = await fetch('/anomalous/scan?' + params.toString(), {method: 'POST'});
|
||
const data = await res.json();
|
||
if (data.status === 'ok') {
|
||
alert(t('scanBg'));
|
||
const poll = setInterval(async () => {
|
||
try {
|
||
const sr = await fetch('/anomalous/scan_status?' + params.toString());
|
||
const sd = await sr.json();
|
||
if (!sd.scanning) {
|
||
clearInterval(poll);
|
||
scanBtn.innerHTML = `✅ <span class="anomalous-btn-text">${t('scanDone')}</span>`;
|
||
|
||
let msg = t('scanCompleteMsg');
|
||
if (sd.result) {
|
||
msg = `✅ 扫描结束！\n成功处理：${sd.result.success} 个\n处理失败：${sd.result.fail} 个\n\n⚠️ 提示：请点击 ComfyUI 的 [Refresh] 按钮以同步，若有模型未生效，请【重启 ComfyUI 服务端】。`;
|
||
if (document.documentElement.lang !== 'zh') {
|
||
msg = `✅ Scan Complete!\nSuccess: ${sd.result.success}\nFailed: ${sd.result.fail}\n\n⚠️ Tip: Click [Refresh] in ComfyUI, or RESTART the ComfyUI backend if models don't appear.`;
|
||
}
|
||
} else {
|
||
msg = msg.replace('[Refresh] 按钮以同步最新的模型列表。', '[Refresh] 按钮以同步，若有模型未生效，请【重启 ComfyUI 服务端】。');
|
||
}
|
||
alert(msg);
|
||
|
||
this.loadModels();
|
||
setTimeout(() => { scanBtn.innerHTML = `🔄 <span class="anomalous-btn-text">${t('scan')}</span>`; scanBtn.disabled = false; }, 2000);
|
||
}
|
||
} catch(e) { clearInterval(poll); scanBtn.disabled = false; }
|
||
}, 3000);
|
||
} else {
|
||
scanBtn.disabled = false;
|
||
}
|
||
} catch (e) { scanBtn.disabled = false; }
|
||
};
|
||
|
||
const energyBtn = document.createElement('button');
|
||
energyBtn.id = 'anomalous-energy-btn';
|
||
energyBtn.title = t('togglePlayTitle');
|
||
const renderEnergyBtn = () => {
|
||
energyBtn.innerHTML = this.energySaving
|
||
? `🔋 <span class="anomalous-btn-text">${t('eco')}</span>`
|
||
: `🎬 <span class="anomalous-btn-text">${t('autoPlay')}</span>`;
|
||
};
|
||
renderEnergyBtn();
|
||
energyBtn.onclick = () => {
|
||
this.energySaving = !this.energySaving;
|
||
localStorage.setItem('anomalous_energy_saving', this.energySaving);
|
||
renderEnergyBtn();
|
||
this.loadModels();
|
||
};
|
||
|
||
const closeBtn = document.createElement('div');
|
||
closeBtn.id = 'anomalous-close';
|
||
closeBtn.innerHTML = '×';
|
||
closeBtn.onclick = () => this.close();
|
||
|
||
const cleanBtn = document.createElement('button');
|
||
cleanBtn.id = 'anomalous-clean-btn';
|
||
cleanBtn.title = t('cleanTitle');
|
||
cleanBtn.innerHTML = `🧹 <span class="anomalous-btn-text">${t('clean')}</span>`;
|
||
cleanBtn.onclick = async () => {
|
||
if (!confirm(t('cleanConfirm'))) return;
|
||
cleanBtn.innerHTML = `⏳ <span class="anomalous-btn-text">${t('cleaning')}</span>`;
|
||
cleanBtn.disabled = true;
|
||
try {
|
||
const res = await fetch('/anomalous/clean_civitai_info', {method: 'POST'});
|
||
const data = await res.json();
|
||
if (data.status === 'success') {
|
||
alert(`${t('cleanDone')} ${data.count} ${t('files')}`);
|
||
} else {
|
||
alert(t('cleanFail') + data.message);
|
||
}
|
||
} catch (e) { alert(t('cleanErr') + e.message); }
|
||
cleanBtn.innerHTML = `🧹 <span class="anomalous-btn-text">${t('clean')}</span>`;
|
||
cleanBtn.disabled = false;
|
||
};
|
||
header.appendChild(helpBtn);
|
||
|
||
const nbBtn = document.createElement('button');
|
||
nbBtn.title = t('notebooksTitle') || 'Notebooks';
|
||
nbBtn.innerHTML = `📓 <span class="anomalous-btn-text">${t('notebooks') || 'Notebooks'}</span>`;
|
||
nbBtn.onclick = () => {
|
||
if (!this.notebookManager) this.notebookManager = new NotebookManager(this);
|
||
this.notebookManager.show();
|
||
};
|
||
nbBtn.style.padding = '8px 15px';
|
||
nbBtn.style.background = '#444';
|
||
nbBtn.style.color = '#fff';
|
||
nbBtn.style.border = 'none';
|
||
nbBtn.style.borderRadius = '4px';
|
||
nbBtn.style.cursor = 'pointer';
|
||
nbBtn.style.display = 'flex';
|
||
nbBtn.style.alignItems = 'center';
|
||
nbBtn.style.gap = '8px';
|
||
header.appendChild(nbBtn);
|
||
header.appendChild(apiKeyInput);
|
||
header.appendChild(cleanBtn);
|
||
header.appendChild(scanBtn);
|
||
header.appendChild(energyBtn);
|
||
header.appendChild(closeBtn);
|
||
|
||
this.grid = document.createElement('div');
|
||
this.grid.id = 'anomalous-grid';
|
||
|
||
this.detailPanel = document.createElement('div');
|
||
this.detailPanel.id = 'anomalous-detail';
|
||
this.detailPanel.style.display = 'none';
|
||
|
||
content.appendChild(header);
|
||
content.appendChild(this.grid);
|
||
content.appendChild(this.detailPanel);
|
||
|
||
container.appendChild(this.sidebarWrapper);
|
||
container.appendChild(content);
|
||
|
||
this.modal.appendChild(container);
|
||
document.body.appendChild(this.modal);
|
||
}
|
||
|
||
|
||
showHelp() {
|
||
if (this.helpModal) {
|
||
this.helpModal.remove();
|
||
}
|
||
this.helpModal = document.createElement('div');
|
||
this.helpModal.style.position = 'absolute';
|
||
this.helpModal.style.top = '0';
|
||
this.helpModal.style.left = '0';
|
||
this.helpModal.style.width = '100%';
|
||
this.helpModal.style.height = '100%';
|
||
this.helpModal.style.background = 'rgba(0,0,0,0.85)';
|
||
this.helpModal.style.zIndex = '9999';
|
||
this.helpModal.style.display = 'flex';
|
||
this.helpModal.style.alignItems = 'center';
|
||
this.helpModal.style.justifyContent = 'center';
|
||
|
||
const box = document.createElement('div');
|
||
box.style.background = 'var(--bg-color, #222)';
|
||
box.style.border = '1px solid var(--border-color, #444)';
|
||
box.style.borderRadius = '8px';
|
||
box.style.width = '550px';
|
||
box.style.maxWidth = '90%';
|
||
box.style.boxShadow = '0 10px 40px rgba(0,0,0,0.8)';
|
||
box.style.display = 'flex';
|
||
box.style.flexDirection = 'column';
|
||
box.style.maxHeight = '90vh';
|
||
|
||
const header = document.createElement('div');
|
||
header.style.padding = '15px 20px';
|
||
header.style.borderBottom = '1px solid #444';
|
||
header.style.background = '#333';
|
||
header.style.display = 'flex';
|
||
header.style.alignItems = 'center';
|
||
header.style.justifyContent = 'space-between';
|
||
header.style.flexWrap = 'wrap';
|
||
header.style.gap = '10px';
|
||
|
||
const title = document.createElement('h2');
|
||
title.innerHTML = t('helpTitle');
|
||
title.style.margin = '0';
|
||
title.style.color = '#fff';
|
||
title.style.fontSize = '1.2em';
|
||
|
||
const closeX = document.createElement('div');
|
||
closeX.innerHTML = '×';
|
||
closeX.style.fontSize = '1.8em';
|
||
closeX.style.cursor = 'pointer';
|
||
closeX.style.color = '#ff4444';
|
||
closeX.onclick = () => this.helpModal.remove();
|
||
|
||
header.appendChild(title);
|
||
header.appendChild(closeX);
|
||
|
||
const body = document.createElement('div');
|
||
body.style.padding = '20px';
|
||
body.innerHTML = t('helpContent');
|
||
body.style.overflowY = 'auto';
|
||
body.style.flex = '1';
|
||
|
||
const footer = document.createElement('div');
|
||
footer.style.padding = '15px';
|
||
footer.style.borderTop = '1px solid #444';
|
||
footer.style.display = 'flex';
|
||
footer.style.justifyContent = 'flex-end';
|
||
|
||
const closeBtn = document.createElement('button');
|
||
closeBtn.innerHTML = t('closeHelp');
|
||
closeBtn.style.padding = '8px 16px';
|
||
closeBtn.style.background = '#444';
|
||
closeBtn.style.color = '#fff';
|
||
closeBtn.style.border = 'none';
|
||
closeBtn.style.borderRadius = '4px';
|
||
closeBtn.style.cursor = 'pointer';
|
||
closeBtn.onclick = () => this.helpModal.remove();
|
||
|
||
footer.appendChild(closeBtn);
|
||
|
||
box.appendChild(header);
|
||
box.appendChild(body);
|
||
box.appendChild(footer);
|
||
this.helpModal.appendChild(box);
|
||
|
||
document.getElementById('anomalous-container').appendChild(this.helpModal);
|
||
}
|
||
|
||
async loadFolders() {
|
||
try {
|
||
const res = await fetch('/anomalous/folders');
|
||
const data = await res.json();
|
||
this.foldersData = data.folders || {};
|
||
|
||
// Auto expand all
|
||
(this.foldersData || []).forEach(typeGroup => {
|
||
this.expandedFolders.add(typeGroup.type);
|
||
Object.keys(typeGroup.folders).forEach(path => {
|
||
this.expandedFolders.add(typeGroup.type + path);
|
||
});
|
||
});
|
||
|
||
this.renderSidebar();
|
||
this.loadModels();
|
||
} catch(e) {}
|
||
}
|
||
|
||
renderSidebar() {
|
||
this.sidebar.innerHTML = '';
|
||
|
||
const topBar = document.createElement('div');
|
||
topBar.style.display = 'flex';
|
||
topBar.style.flexDirection = 'column';
|
||
topBar.style.gap = '10px';
|
||
topBar.style.padding = '15px';
|
||
|
||
const logoRow = document.createElement('div');
|
||
logoRow.innerHTML = '<span style="font-size: 0.85em; font-weight: bold; color: #aaa; letter-spacing: 0.5px; white-space: nowrap;">📦 ANOMALOUS BROWSER</span>';
|
||
|
||
const folderRow = document.createElement('div');
|
||
folderRow.style.display = 'flex';
|
||
folderRow.style.justifyContent = 'space-between';
|
||
folderRow.style.alignItems = 'center';
|
||
|
||
const title = document.createElement('h3');
|
||
title.innerHTML = t('folders');
|
||
title.style.color = '#fff';
|
||
title.style.margin = '0';
|
||
|
||
const isAllCollapsed = this.expandedFolders.size === 0;
|
||
const collapseAllBtn = document.createElement('button');
|
||
collapseAllBtn.innerHTML = isAllCollapsed ? t('expandAll') : t('collapseAll');
|
||
collapseAllBtn.style.padding = '4px 8px';
|
||
collapseAllBtn.style.background = '#444';
|
||
collapseAllBtn.style.color = '#fff';
|
||
collapseAllBtn.style.border = 'none';
|
||
collapseAllBtn.style.borderRadius = '4px';
|
||
collapseAllBtn.style.cursor = 'pointer';
|
||
collapseAllBtn.style.whiteSpace = 'nowrap';
|
||
collapseAllBtn.onclick = () => {
|
||
if (isAllCollapsed) {
|
||
(this.foldersData || []).forEach(typeGroup => {
|
||
this.expandedFolders.add(typeGroup.type);
|
||
Object.keys(typeGroup.folders).forEach(path => {
|
||
this.expandedFolders.add(typeGroup.type + path);
|
||
});
|
||
});
|
||
} else {
|
||
this.expandedFolders.clear();
|
||
}
|
||
this.renderSidebar();
|
||
};
|
||
|
||
folderRow.appendChild(title);
|
||
folderRow.appendChild(collapseAllBtn);
|
||
|
||
topBar.appendChild(logoRow);
|
||
topBar.appendChild(folderRow);
|
||
|
||
this.sidebar.appendChild(topBar);
|
||
|
||
(this.foldersData || []).forEach(typeGroup => {
|
||
const header = document.createElement('div');
|
||
header.className = 'anomalous-type-header';
|
||
header.style.display = 'flex';
|
||
header.style.justifyContent = 'space-between';
|
||
header.style.flexWrap = 'wrap';
|
||
header.style.gap = '10px';
|
||
header.style.cursor = 'pointer';
|
||
|
||
const isTypeExpanded = this.expandedFolders.has(typeGroup.type);
|
||
header.innerHTML = `<span>${typeGroup.label}</span> <span>${isTypeExpanded ? '▼' : '▶'}</span>`;
|
||
|
||
header.onclick = () => {
|
||
if (isTypeExpanded) this.expandedFolders.delete(typeGroup.type);
|
||
else this.expandedFolders.add(typeGroup.type);
|
||
this.renderSidebar();
|
||
};
|
||
this.sidebar.appendChild(header);
|
||
|
||
if (!isTypeExpanded) return;
|
||
|
||
const sortedPaths = Object.keys(typeGroup.folders).sort();
|
||
|
||
sortedPaths.forEach(path => {
|
||
const info = typeGroup.folders[path];
|
||
const parts = path.split('/').filter(p => p);
|
||
const parentPath = parts.length > 1 ? '/' + parts.slice(0, -1).join('/') : '/';
|
||
|
||
if (path !== '/' && parentPath !== '/') {
|
||
let parentId = typeGroup.type + parentPath;
|
||
if (!this.expandedFolders.has(parentId)) return;
|
||
}
|
||
|
||
const hasChildren = sortedPaths.some(p => p !== path && p.startsWith(path === '/' ? '/' : path + '/'));
|
||
|
||
const item = document.createElement('div');
|
||
item.className = 'anomalous-folder-item';
|
||
|
||
const depth = path === '/' ? 0 : parts.length;
|
||
item.style.paddingLeft = (15 + depth * 15) + 'px';
|
||
|
||
const myId = typeGroup.type + path;
|
||
const isExpanded = this.expandedFolders.has(myId);
|
||
|
||
let toggleIcon = '';
|
||
if (hasChildren) {
|
||
toggleIcon = `<span class="anomalous-folder-toggle" style="margin-right: 5px; width: 15px; display: inline-block;">${isExpanded ? '▼' : '▶'}</span>`;
|
||
} else {
|
||
toggleIcon = `<span style="margin-right: 5px; width: 15px; display: inline-block;"></span>`;
|
||
}
|
||
|
||
item.innerHTML = `${toggleIcon}📁 ${info.name} <span style="opacity:0.5; font-size:0.9em;">(${info.model_count})</span>`;
|
||
|
||
if (this.currentType === typeGroup.type && this.currentPathIdx === typeGroup.path_idx && this.currentSubfolder === path) {
|
||
item.classList.add('active');
|
||
}
|
||
|
||
item.onclick = (e) => {
|
||
if (e.target.classList.contains('anomalous-folder-toggle')) {
|
||
if (isExpanded) this.expandedFolders.delete(myId);
|
||
else this.expandedFolders.add(myId);
|
||
this.renderSidebar();
|
||
return;
|
||
}
|
||
this.currentType = typeGroup.type;
|
||
this.currentPathIdx = typeGroup.path_idx;
|
||
this.currentSubfolder = path;
|
||
|
||
this.detailPanel.style.display = 'none';
|
||
this.grid.style.display = 'grid';
|
||
|
||
this.renderSidebar();
|
||
this.loadModels();
|
||
};
|
||
|
||
this.sidebar.appendChild(item);
|
||
});
|
||
});
|
||
}
|
||
|
||
async loadModels() {
|
||
try {
|
||
const params = new URLSearchParams({ type: this.currentType, path_idx: this.currentPathIdx, subfolder: this.currentSubfolder });
|
||
const res = await fetch('/anomalous/models?' + params.toString());
|
||
const data = await res.json();
|
||
this.grid.innerHTML = '';
|
||
|
||
if (!data.models || data.models.length === 0) {
|
||
this.grid.innerHTML = `<div style="color:white; padding:20px;">${t('noModels')}</div>`;
|
||
return;
|
||
}
|
||
|
||
data.models.forEach(model => {
|
||
const card = document.createElement('div');
|
||
card.className = 'anomalous-card';
|
||
if (model.preview_url) {
|
||
const isVideo = model.preview_url.toLowerCase().endsWith('.mp4') || model.preview_url.toLowerCase().endsWith('.webm');
|
||
if (isVideo) {
|
||
const video = document.createElement('video');
|
||
video.src = model.preview_url;
|
||
video.muted = true; video.loop = true; video.playsInline = true;
|
||
if (this.energySaving) {
|
||
video.pause();
|
||
card.addEventListener('mouseenter', () => video.play().catch(e=>{}));
|
||
card.addEventListener('mouseleave', () => { video.pause(); video.currentTime = 0; });
|
||
} else {
|
||
video.autoplay = true;
|
||
}
|
||
card.appendChild(video);
|
||
} else {
|
||
const img = document.createElement('img');
|
||
img.src = model.preview_url;
|
||
card.appendChild(img);
|
||
}
|
||
} else {
|
||
const ph = document.createElement('div');
|
||
ph.className = 'anomalous-card-placeholder';
|
||
ph.innerHTML = `<div style="text-align:center;color:#666;margin-top:80px;">${t('noPreview')}</div><div style="font-size:0.8em;text-align:center;opacity:0.5;margin-top:5px">${t('clickScan')}</div>`;
|
||
card.appendChild(ph);
|
||
}
|
||
const title = document.createElement('div');
|
||
if (model.metadata && model.metadata.baseModel) {
|
||
const badge = document.createElement('div');
|
||
badge.innerText = model.metadata.baseModel;
|
||
badge.style.position = 'absolute';
|
||
badge.style.top = '6px';
|
||
badge.style.left = '6px';
|
||
badge.style.background = 'rgba(0,0,0,0.85)';
|
||
badge.style.color = '#00ffcc';
|
||
badge.style.padding = '3px 6px';
|
||
badge.style.borderRadius = '4px';
|
||
badge.style.fontSize = '0.75em';
|
||
badge.style.fontWeight = 'bold';
|
||
badge.style.border = '1px solid rgba(0,255,204,0.3)';
|
||
badge.style.pointerEvents = 'none';
|
||
badge.style.zIndex = '10';
|
||
card.appendChild(badge);
|
||
}
|
||
title.className = 'anomalous-card-title';
|
||
title.innerText = model.filename;
|
||
card.appendChild(title);
|
||
|
||
card.onclick = () => { this.historyStack = []; this.currentDetailModel = model; this.showDetail(model); };
|
||
|
||
const applyBtn = document.createElement('button');
|
||
applyBtn.innerHTML = '➕';
|
||
applyBtn.title = t('applyToCanvas');
|
||
applyBtn.style.position = 'absolute';
|
||
applyBtn.style.top = '6px';
|
||
applyBtn.style.right = '6px';
|
||
applyBtn.style.background = 'rgba(0,0,0,0.7)';
|
||
applyBtn.style.color = '#fff';
|
||
applyBtn.style.border = '1px solid rgba(255,255,255,0.2)';
|
||
applyBtn.style.borderRadius = '4px';
|
||
applyBtn.style.cursor = 'pointer';
|
||
applyBtn.style.padding = '4px 6px';
|
||
applyBtn.style.zIndex = '20';
|
||
applyBtn.style.fontSize = '1em';
|
||
applyBtn.style.display = 'none';
|
||
|
||
card.addEventListener('mouseenter', () => applyBtn.style.display = 'block');
|
||
card.addEventListener('mouseleave', () => applyBtn.style.display = 'none');
|
||
|
||
applyBtn.onclick = (e) => {
|
||
e.stopPropagation();
|
||
this.applyModelToCanvas(this.currentType, this.currentSubfolder, model);
|
||
};
|
||
card.appendChild(applyBtn);
|
||
|
||
|
||
this.grid.appendChild(card);
|
||
});
|
||
} catch(e) {}
|
||
}
|
||
|
||
|
||
applyModelToCanvas(type, subfolder, model) {
|
||
const nodeTypeMap = {
|
||
'checkpoints': 'CheckpointLoaderSimple',
|
||
'loras': 'LoraLoader',
|
||
'unet': 'UNETLoader',
|
||
'diffusion_models': 'UNETLoader'
|
||
};
|
||
const nodeType = nodeTypeMap[type];
|
||
if (!nodeType) {
|
||
alert('Unsupported model type for auto-apply.');
|
||
return;
|
||
}
|
||
|
||
const node = LiteGraph.createNode(nodeType);
|
||
if (!node) {
|
||
alert('Failed to create node: ' + nodeType);
|
||
return;
|
||
}
|
||
|
||
if (app.canvas && app.canvas.graph_mouse) {
|
||
node.pos = [
|
||
app.canvas.graph_mouse[0] || (window.innerWidth / 2),
|
||
app.canvas.graph_mouse[1] || (window.innerHeight / 2)
|
||
];
|
||
} else {
|
||
node.pos = [ window.innerWidth / 2, window.innerHeight / 2 ];
|
||
}
|
||
|
||
app.graph.add(node);
|
||
|
||
const sub = subfolder.replace(/^\/+/, '').replace(/\/+$/, '');
|
||
const relPath = sub ? `${sub}/${model.filename}` : model.filename;
|
||
|
||
if (node.widgets) {
|
||
// First try to find a combo widget (dropdown)
|
||
const combo = node.widgets.find(w => w.type === 'combo');
|
||
if (combo) {
|
||
combo.value = relPath;
|
||
} else if (node.widgets.length > 0) {
|
||
// fallback to first widget
|
||
node.widgets[0].value = relPath;
|
||
}
|
||
}
|
||
|
||
this.close();
|
||
|
||
// 粘到鼠标上的逻辑
|
||
let isSticking = true;
|
||
const stickHandler = (e) => {
|
||
if (!isSticking || !app.canvas) return;
|
||
const canvas = app.canvas;
|
||
|
||
// LiteGraph内置了坐标转换，它会完美处理缩放和偏移带来的坐标偏移问题
|
||
let canvasX, canvasY;
|
||
if (canvas.convertEventToCanvasOffset) {
|
||
const pos = canvas.convertEventToCanvasOffset(e);
|
||
canvasX = pos[0];
|
||
canvasY = pos[1];
|
||
} else {
|
||
// 如果API不可用，使用标准降级计算
|
||
const rect = canvas.canvas.getBoundingClientRect();
|
||
canvasX = (e.clientX - rect.left - canvas.ds.offset[0]) / canvas.ds.scale;
|
||
canvasY = (e.clientY - rect.top - canvas.ds.offset[1]) / canvas.ds.scale;
|
||
}
|
||
|
||
node.pos = [canvasX - node.size[0] / 2, canvasY - 20];
|
||
canvas.setDirty(true, true);
|
||
};
|
||
const dropHandler = (e) => {
|
||
isSticking = false;
|
||
window.removeEventListener('mousemove', stickHandler, true);
|
||
window.removeEventListener('pointerdown', dropHandler, true);
|
||
window.removeEventListener('mousedown', dropHandler, true);
|
||
window.removeEventListener('click', dropHandler, true);
|
||
};
|
||
window.addEventListener('mousemove', stickHandler, true);
|
||
setTimeout(() => {
|
||
window.addEventListener('pointerdown', dropHandler, true);
|
||
window.addEventListener('mousedown', dropHandler, true);
|
||
window.addEventListener('click', dropHandler, true);
|
||
}, 100);
|
||
}
|
||
|
||
showDetail(model) {
|
||
this.grid.style.display = 'none';
|
||
this.detailPanel.style.display = 'flex';
|
||
this.detailPanel.innerHTML = '';
|
||
|
||
const header = document.createElement('div');
|
||
header.style.width = '100%';
|
||
header.style.padding = '8px 15px';
|
||
header.style.background = 'var(--comfy-menu-bg, #333)';
|
||
header.style.borderBottom = '1px solid var(--border-color, #444)';
|
||
header.style.display = 'flex';
|
||
header.style.alignItems = 'center';
|
||
header.style.boxSizing = 'border-box';
|
||
|
||
const backBtn = document.createElement('button');
|
||
backBtn.innerHTML = this.historyStack.length > 0 ? t('backToPrev') : t('back');
|
||
backBtn.style.padding = '6px 12px';
|
||
backBtn.style.background = '#444';
|
||
backBtn.style.color = '#fff';
|
||
backBtn.style.border = 'none';
|
||
backBtn.style.borderRadius = '4px';
|
||
backBtn.style.cursor = 'pointer';
|
||
backBtn.onclick = () => {
|
||
if (this.historyStack.length > 0) {
|
||
const prev = this.historyStack.pop();
|
||
this.currentType = prev.type;
|
||
this.currentPathIdx = prev.pathIdx;
|
||
this.currentSubfolder = prev.subfolder;
|
||
this.currentDetailModel = prev.model;
|
||
this.renderSidebar();
|
||
this.showDetail(prev.model);
|
||
} else {
|
||
this.detailPanel.style.display = 'none';
|
||
this.detailPanel.innerHTML = '';
|
||
this.grid.style.display = 'grid';
|
||
}
|
||
};
|
||
|
||
const title = document.createElement('h2');
|
||
title.innerHTML = model.filename;
|
||
title.style.margin = '0 20px 0 20px';
|
||
title.style.color = '#fff';
|
||
title.style.fontSize = '1.2em';
|
||
// 强制单行并溢出显示省略号
|
||
title.style.whiteSpace = 'nowrap';
|
||
title.style.overflow = 'hidden';
|
||
title.style.textOverflow = 'ellipsis';
|
||
title.style.flex = '1'; // 撑开剩余空间，把右侧按钮挤到最右边
|
||
|
||
const delBtn = document.createElement('button');
|
||
delBtn.innerHTML = t('delModel');
|
||
delBtn.style.padding = '6px 12px';
|
||
delBtn.style.background = '#ff4444';
|
||
delBtn.style.color = '#fff';
|
||
delBtn.style.border = 'none';
|
||
delBtn.style.borderRadius = '4px';
|
||
delBtn.style.cursor = 'pointer';
|
||
delBtn.style.marginLeft = 'auto'; // push to the right
|
||
delBtn.style.whiteSpace = 'nowrap';
|
||
delBtn.onclick = async () => {
|
||
if (!confirm(`${t('delConfirm')} ${model.filename} ${t('delConfirm2')}`)) return;
|
||
try {
|
||
const res = await fetch('/anomalous/delete_model', {
|
||
method: 'POST',
|
||
headers: { 'Content-Type': 'application/json' },
|
||
body: JSON.stringify({
|
||
type: this.currentType,
|
||
path_idx: this.currentPathIdx,
|
||
subfolder: this.currentSubfolder,
|
||
filename: model.filename
|
||
})
|
||
});
|
||
const data = await res.json();
|
||
if (data.status === 'success') {
|
||
alert(t('delSuccess') + data.deleted.join('\n') + t('delNote'));
|
||
this.detailPanel.style.display = 'none';
|
||
this.detailPanel.innerHTML = '';
|
||
this.grid.style.display = 'grid';
|
||
this.loadModels(); // refresh grid
|
||
} else {
|
||
alert(t('delFail') + data.message);
|
||
}
|
||
} catch (e) {
|
||
alert(t('delFail') + e.message);
|
||
}
|
||
};
|
||
|
||
const jumpBtn = document.createElement('button');
|
||
jumpBtn.innerHTML = '⬇️';
|
||
jumpBtn.title = t('jumpToBottom') || 'Jump to bottom';
|
||
jumpBtn.style.padding = '6px 12px';
|
||
jumpBtn.style.background = '#444';
|
||
jumpBtn.style.color = '#fff';
|
||
jumpBtn.style.border = 'none';
|
||
jumpBtn.style.borderRadius = '4px';
|
||
jumpBtn.style.cursor = 'pointer';
|
||
jumpBtn.style.marginLeft = '10px';
|
||
jumpBtn.onclick = () => {
|
||
// Find rightPanel which is created later, so we bind it dynamically
|
||
const rp = this.detailPanel.querySelector('.anomalous-split-right');
|
||
if (rp) rp.scrollTo({ top: rp.scrollHeight, behavior: 'smooth' });
|
||
};
|
||
|
||
header.appendChild(backBtn);
|
||
header.appendChild(title);
|
||
header.appendChild(delBtn);
|
||
header.appendChild(jumpBtn);
|
||
|
||
const applyDetailBtn = document.createElement('button');
|
||
applyDetailBtn.innerHTML = t('applyToCanvas');
|
||
applyDetailBtn.style.padding = '6px 12px';
|
||
applyDetailBtn.style.background = '#28a745';
|
||
applyDetailBtn.style.color = '#fff';
|
||
applyDetailBtn.style.border = 'none';
|
||
applyDetailBtn.style.borderRadius = '4px';
|
||
applyDetailBtn.style.cursor = 'pointer';
|
||
applyDetailBtn.style.marginLeft = '10px';
|
||
applyDetailBtn.style.fontWeight = 'bold';
|
||
applyDetailBtn.onclick = () => {
|
||
this.applyModelToCanvas(this.currentType, this.currentSubfolder, model);
|
||
};
|
||
header.appendChild(applyDetailBtn);
|
||
|
||
|
||
const splitContainer = document.createElement('div');
|
||
splitContainer.className = 'anomalous-split-container';
|
||
|
||
const leftPanel = document.createElement('div');
|
||
leftPanel.className = 'anomalous-split-left';
|
||
|
||
if (model.preview_url) {
|
||
const isVideo = model.preview_url.toLowerCase().endsWith('.mp4') || model.preview_url.toLowerCase().endsWith('.webm');
|
||
if (isVideo) {
|
||
const video = document.createElement('video');
|
||
video.src = model.preview_url;
|
||
video.controls = true;
|
||
video.autoplay = true;
|
||
video.loop = true;
|
||
video.style.width = '100%';
|
||
video.style.height = '100%';
|
||
video.style.objectFit = 'contain';
|
||
leftPanel.appendChild(video);
|
||
} else {
|
||
const img = document.createElement('img');
|
||
img.src = model.preview_url;
|
||
img.style.width = '100%';
|
||
img.style.height = '100%';
|
||
img.style.objectFit = 'contain';
|
||
leftPanel.appendChild(img);
|
||
}
|
||
} else {
|
||
leftPanel.innerHTML = `<div style="color:#aaa; text-align:center; margin-top:50px;">${t('noPreview')}</div>`;
|
||
}
|
||
|
||
const resizer = document.createElement('div');
|
||
resizer.className = 'anomalous-resizer';
|
||
|
||
let isResizing = false;
|
||
resizer.addEventListener('mousedown', (e) => {
|
||
isResizing = true;
|
||
document.body.style.cursor = 'col-resize';
|
||
e.preventDefault();
|
||
});
|
||
|
||
window.addEventListener('mousemove', (e) => {
|
||
if (!isResizing) return;
|
||
const containerRect = splitContainer.getBoundingClientRect();
|
||
let newWidth = ((e.clientX - containerRect.left) / containerRect.width) * 100;
|
||
if (newWidth < 20) newWidth = 20;
|
||
if (newWidth > 80) newWidth = 80;
|
||
leftPanel.style.width = newWidth + '%';
|
||
rightPanel.style.width = (100 - newWidth) + '%';
|
||
});
|
||
|
||
window.addEventListener('mouseup', () => {
|
||
if (isResizing) {
|
||
isResizing = false;
|
||
document.body.style.cursor = '';
|
||
}
|
||
});
|
||
|
||
const rightPanel = document.createElement('div');
|
||
rightPanel.className = 'anomalous-split-right';
|
||
rightPanel.style.display = 'flex';
|
||
rightPanel.style.flexDirection = 'column';
|
||
rightPanel.style.height = '100%';
|
||
rightPanel.style.boxSizing = 'border-box';
|
||
rightPanel.style.overflow = 'auto';
|
||
rightPanel.style.padding = '15px';
|
||
rightPanel.style.color = '#eee';
|
||
|
||
const m = model.metadata || {};
|
||
|
||
// 1. Top Bar (Title + Size + Model + Button) in a single compact row if possible
|
||
const topRow = document.createElement('div');
|
||
topRow.style.flexShrink = '0';
|
||
topRow.style.display = 'flex';
|
||
topRow.style.flexWrap = 'wrap';
|
||
topRow.style.alignItems = 'center';
|
||
topRow.style.gap = '10px';
|
||
topRow.style.paddingBottom = '10px';
|
||
topRow.style.marginBottom = '10px';
|
||
topRow.style.borderBottom = '1px solid #444';
|
||
|
||
const titleEl = document.createElement('h3');
|
||
titleEl.style.margin = '0';
|
||
titleEl.style.fontSize = '1.3em';
|
||
titleEl.style.marginRight = '10px';
|
||
titleEl.innerText = m.name || model.filename;
|
||
topRow.appendChild(titleEl);
|
||
|
||
const metaSpan = document.createElement('span');
|
||
metaSpan.style.fontSize = '0.9em';
|
||
metaSpan.style.color = '#aaa';
|
||
metaSpan.innerHTML = `<strong>Size:</strong> ${model.size_mb} MB` + (m.baseModel ? ` <strong style="margin-left:10px;">Base:</strong> ${m.baseModel}` : '');
|
||
topRow.appendChild(metaSpan);
|
||
|
||
if (m.civitai_url) {
|
||
const cBtn = document.createElement('a');
|
||
cBtn.href = m.civitai_url;
|
||
cBtn.target = '_blank';
|
||
cBtn.innerHTML = '🌐 Civitai';
|
||
cBtn.style.marginLeft = 'auto';
|
||
cBtn.style.padding = '4px 8px';
|
||
cBtn.style.background = '#1a73e8';
|
||
cBtn.style.color = '#fff';
|
||
cBtn.style.textDecoration = 'none';
|
||
cBtn.style.borderRadius = '4px';
|
||
cBtn.style.fontSize = '0.85em';
|
||
cBtn.style.fontWeight = 'bold';
|
||
topRow.appendChild(cBtn);
|
||
}
|
||
|
||
rightPanel.appendChild(topRow);
|
||
|
||
// 2. Trained Words Section
|
||
if (m.trainedWords && m.trainedWords.length > 0) {
|
||
const twCont = document.createElement('div');
|
||
twCont.style.flexShrink = '0';
|
||
twCont.style.marginBottom = '10px';
|
||
|
||
const twHeader = document.createElement('div');
|
||
twHeader.style.display = 'flex';
|
||
twHeader.style.alignItems = 'center';
|
||
twHeader.style.marginBottom = '5px';
|
||
|
||
const twLabel = document.createElement('strong');
|
||
twLabel.innerText = 'Trained Words:';
|
||
twHeader.appendChild(twLabel);
|
||
|
||
const copyAll = document.createElement('button');
|
||
copyAll.innerText = t('copyAll');
|
||
copyAll.style.marginLeft = '10px';
|
||
copyAll.style.padding = '2px 6px';
|
||
copyAll.style.background = '#444';
|
||
copyAll.style.color = '#fff';
|
||
copyAll.style.border = 'none';
|
||
copyAll.style.borderRadius = '3px';
|
||
copyAll.style.cursor = 'pointer';
|
||
copyAll.style.fontSize = '0.8em';
|
||
copyAll.onclick = () => {
|
||
const allWords = m.trainedWords.join(', ');
|
||
navigator.clipboard.writeText(allWords).then(() => {
|
||
const old = copyAll.innerText;
|
||
copyAll.innerText = t('copied');
|
||
setTimeout(() => { copyAll.innerText = old; }, 1500);
|
||
});
|
||
};
|
||
twHeader.appendChild(copyAll);
|
||
twCont.appendChild(twHeader);
|
||
|
||
const tagsCont = document.createElement('div');
|
||
tagsCont.style.display = 'flex';
|
||
tagsCont.style.flexWrap = 'wrap';
|
||
tagsCont.style.gap = '4px';
|
||
|
||
m.trainedWords.forEach(w => {
|
||
const tag = document.createElement('span');
|
||
tag.innerText = w;
|
||
tag.style.background = '#333';
|
||
tag.style.padding = '2px 6px';
|
||
tag.style.borderRadius = '4px';
|
||
tag.style.fontSize = '0.85em';
|
||
tag.style.cursor = 'pointer';
|
||
tag.style.border = '1px solid #555';
|
||
tag.title = t('clickToCopy') + w;
|
||
tag.onclick = () => {
|
||
navigator.clipboard.writeText(w).then(() => {
|
||
tag.style.background = '#28a745';
|
||
setTimeout(() => { tag.style.background = '#333'; }, 500);
|
||
});
|
||
};
|
||
tagsCont.appendChild(tag);
|
||
});
|
||
twCont.appendChild(tagsCont);
|
||
rightPanel.appendChild(twCont);
|
||
}
|
||
|
||
// 3. Description Section (Expands to fill remaining height)
|
||
if (m.description) {
|
||
const descCont = document.createElement('div');
|
||
descCont.style.flex = 'none';
|
||
descCont.style.display = 'flex';
|
||
descCont.style.flexDirection = 'column';
|
||
// important for flex scroll
|
||
|
||
const descLabel = document.createElement('strong');
|
||
descLabel.innerText = 'Description:';
|
||
descLabel.style.marginBottom = '5px';
|
||
descCont.appendChild(descLabel);
|
||
|
||
const descText = document.createElement('div');
|
||
descText.style.flex = 'none';
|
||
|
||
descText.style.background = '#222';
|
||
descText.style.padding = '10px';
|
||
descText.style.borderRadius = '6px';
|
||
descText.style.border = '1px solid #333';
|
||
descText.style.fontSize = '0.95em';
|
||
descText.style.lineHeight = '1.4';
|
||
descText.innerHTML = m.description;
|
||
descCont.appendChild(descText);
|
||
|
||
rightPanel.appendChild(descCont);
|
||
}
|
||
|
||
// 4. Notes Section
|
||
if (m.notes) {
|
||
const notesCont = document.createElement('div');
|
||
notesCont.style.flexShrink = '0';
|
||
notesCont.style.marginTop = '10px';
|
||
|
||
const notesLabel = document.createElement('strong');
|
||
notesLabel.innerText = 'Notes:';
|
||
notesCont.appendChild(notesLabel);
|
||
|
||
const notesText = document.createElement('div');
|
||
notesText.style.background = '#332b00';
|
||
notesText.style.padding = '8px';
|
||
notesText.style.borderRadius = '6px';
|
||
notesText.style.marginTop = '5px';
|
||
notesText.style.border = '1px solid #554400';
|
||
notesText.style.fontSize = '0.9em';
|
||
notesText.innerHTML = m.notes;
|
||
notesCont.appendChild(notesText);
|
||
|
||
rightPanel.appendChild(notesCont);
|
||
}
|
||
|
||
// --- Compatible Models Section ---
|
||
if (m.baseModel) {
|
||
const compSec = document.createElement('div');
|
||
compSec.className = 'anomalous-compatible-section';
|
||
|
||
const compTitle = document.createElement('div');
|
||
compTitle.className = 'anomalous-compatible-title';
|
||
compTitle.innerHTML = `${t('compatibleModels') || '🔗 Compatible'} <span style="font-size:0.8em; opacity:0.6;">(${m.baseModel})</span>`;
|
||
|
||
const compList = document.createElement('div');
|
||
compList.className = 'anomalous-compatible-list';
|
||
compList.innerHTML = `<span style="color:#888;">${t('loadingCompatible') || 'Loading...'}</span>`;
|
||
|
||
compSec.appendChild(compTitle);
|
||
compSec.appendChild(compList);
|
||
rightPanel.appendChild(compSec);
|
||
|
||
const targetType = this.currentType === 'loras' ? 'checkpoints,unet,diffusion_models' : 'loras';
|
||
fetch(`/anomalous/compatible_models?base_model=${encodeURIComponent(m.baseModel)}&target_type=${encodeURIComponent(targetType)}`)
|
||
.then(r => r.json())
|
||
.then(d => {
|
||
compList.innerHTML = '';
|
||
if (!d.models || d.models.length === 0) {
|
||
compList.innerHTML = `<span style="color:#888;">No compatible models found.</span>`;
|
||
return;
|
||
}
|
||
d.models.forEach(m_comp => {
|
||
const mItem = document.createElement('div');
|
||
mItem.className = 'anomalous-compatible-item';
|
||
mItem.title = m_comp.filename;
|
||
|
||
let thumb = '';
|
||
let isVid = false;
|
||
if (m_comp.preview_url) {
|
||
isVid = m_comp.preview_url.toLowerCase().endsWith('.mp4') || m_comp.preview_url.toLowerCase().endsWith('.webm');
|
||
if (isVid) thumb = `<video class="comp-vid" src="${m_comp.preview_url}" muted loop playsinline style="pointer-events: none;"></video>`;
|
||
else thumb = `<img src="${m_comp.preview_url}" />`;
|
||
} else {
|
||
thumb = `<div style="width:30px; height:30px; background:#222; border-radius:4px; display:flex; align-items:center; justify-content:center; font-size:10px; color:#555;">?</div>`;
|
||
}
|
||
|
||
mItem.innerHTML = `${thumb}<div class="anomalous-compatible-item-name">${m_comp.filename}</div>`;
|
||
if (isVid) {
|
||
mItem.onmouseenter = () => { const v = mItem.querySelector('video'); if(v) v.play(); };
|
||
mItem.onmouseleave = () => { const v = mItem.querySelector('video'); if(v) { v.pause(); v.currentTime = 0; } };
|
||
}
|
||
|
||
mItem.onclick = () => {
|
||
this.historyStack.push({
|
||
type: this.currentType,
|
||
pathIdx: this.currentPathIdx,
|
||
subfolder: this.currentSubfolder,
|
||
model: this.currentDetailModel
|
||
});
|
||
|
||
this.currentType = m_comp.type;
|
||
this.currentPathIdx = m_comp.path_idx;
|
||
this.currentSubfolder = m_comp.subfolder;
|
||
this.currentDetailModel = m_comp;
|
||
|
||
this.renderSidebar();
|
||
this.showDetail(m_comp);
|
||
};
|
||
|
||
compList.appendChild(mItem);
|
||
});
|
||
})
|
||
.catch(e => {
|
||
compList.innerHTML = `<span style="color:#ff4444;">Failed to load.</span>`;
|
||
});
|
||
}
|
||
// ---------------------------------
|
||
|
||
splitContainer.appendChild(leftPanel);
|
||
splitContainer.appendChild(resizer);
|
||
splitContainer.appendChild(rightPanel);
|
||
|
||
this.detailPanel.appendChild(header);
|
||
this.detailPanel.appendChild(splitContainer);
|
||
}
|
||
|
||
show() {
|
||
this.modal.classList.add('visible');
|
||
if (!this.foldersData) {
|
||
this.loadFolders();
|
||
} else {
|
||
this.loadModels();
|
||
}
|
||
}
|
||
close() { this.modal.classList.remove('visible'); }
|
||
}
|
||
|
||
app.registerExtension({
|
||
name: 'Anomalous.ModelBrowser',
|
||
async setup() {
|
||
try {
|
||
const cssUrl = '/extensions/Anomalous_Model_Browser/styles.css?v=' + Date.now();
|
||
if (!document.querySelector(`link[href^="/extensions/Anomalous_Model_Browser/styles.css"]`)) {
|
||
const link = document.createElement("link");
|
||
link.rel = "stylesheet";
|
||
link.type = "text/css";
|
||
link.href = cssUrl;
|
||
document.head.appendChild(link);
|
||
}
|
||
|
||
const browser = new AnomalousBrowser();
|
||
const btn = document.createElement('button');
|
||
btn.id = 'anomalous-trigger-btn';
|
||
btn.innerHTML = '📦';
|
||
btn.title = 'Open Anomalous Model Browser';
|
||
btn.style.position = 'fixed';
|
||
btn.style.bottom = '20px';
|
||
btn.style.right = '20px';
|
||
btn.style.zIndex = '99999';
|
||
btn.style.width = '48px';
|
||
btn.style.height = '48px';
|
||
btn.style.borderRadius = '50%';
|
||
btn.style.border = '2px solid #555';
|
||
btn.style.background = 'linear-gradient(135deg, #1a1a2e, #16213e)';
|
||
btn.style.color = '#fff';
|
||
btn.style.fontSize = '22px';
|
||
btn.style.cursor = 'pointer';
|
||
btn.style.boxShadow = '0 4px 15px rgba(0,0,0,0.5)';
|
||
btn.style.transition = 'transform 0.2s, box-shadow 0.2s';
|
||
btn.onmouseenter = () => { btn.style.transform = 'scale(1.15)'; btn.style.boxShadow = '0 6px 20px rgba(100,100,255,0.4)'; };
|
||
btn.onmouseleave = () => { btn.style.transform = 'scale(1)'; btn.style.boxShadow = '0 4px 15px rgba(0,0,0,0.5)'; };
|
||
btn.onclick = () => browser.show();
|
||
document.body.appendChild(btn);
|
||
} catch(e) {
|
||
console.error('[Anomalous Model Browser] Setup error:', e);
|
||
}
|
||
}
|
||
});
|
||
|
||
|
||
class NotebookManager {
|
||
constructor(browser) {
|
||
this.browser = browser;
|
||
this.modal = null;
|
||
this.currentNotebook = null;
|
||
this.checkpoints = [];
|
||
this.loras = [];
|
||
this.init();
|
||
}
|
||
|
||
async init() {
|
||
try {
|
||
const res = await fetch('/anomalous/all_models');
|
||
const data = await res.json();
|
||
this.checkpoints = data.checkpoints || [];
|
||
this.loras = data.loras || [];
|
||
|
||
const bases = new Set();
|
||
[...this.checkpoints, ...this.loras].forEach(m => {
|
||
if (m.baseModel && m.baseModel.trim() !== '') {
|
||
bases.add(m.baseModel.trim());
|
||
}
|
||
});
|
||
this.dynamicBases = Array.from(bases).sort();
|
||
} catch(e) {}
|
||
}
|
||
|
||
show() {
|
||
if (this.modal) this.modal.remove();
|
||
|
||
this.modal = document.createElement('div');
|
||
this.modal.style.position = 'absolute';
|
||
this.modal.style.top = '0';
|
||
this.modal.style.left = '0';
|
||
this.modal.style.width = '100%';
|
||
this.modal.style.height = '100%';
|
||
this.modal.style.background = 'rgba(0,0,0,0.85)';
|
||
this.modal.style.zIndex = '99999';
|
||
this.modal.style.display = 'flex';
|
||
this.modal.style.alignItems = 'center';
|
||
this.modal.style.justifyContent = 'center';
|
||
|
||
const box = document.createElement('div');
|
||
box.style.background = '#222';
|
||
box.style.border = '1px solid #444';
|
||
box.style.borderRadius = '8px';
|
||
box.style.width = '1000px';
|
||
box.style.maxWidth = '95%';
|
||
box.style.height = '80vh';
|
||
box.style.display = 'flex';
|
||
box.style.flexDirection = 'column';
|
||
|
||
const header = document.createElement('div');
|
||
header.style.padding = '15px 20px';
|
||
header.style.borderBottom = '1px solid #444';
|
||
header.style.background = '#333';
|
||
header.style.display = 'flex';
|
||
header.style.alignItems = 'center';
|
||
header.style.justifyContent = 'space-between';
|
||
header.style.flexWrap = 'wrap';
|
||
header.style.gap = '10px';
|
||
|
||
const title = document.createElement('h2');
|
||
title.innerHTML = '📓 Prompt Notebooks';
|
||
title.style.margin = '0';
|
||
title.style.color = '#fff';
|
||
|
||
const closeX = document.createElement('div');
|
||
closeX.innerHTML = '×';
|
||
closeX.style.fontSize = '1.8em';
|
||
closeX.style.cursor = 'pointer';
|
||
closeX.style.color = '#ff4444';
|
||
closeX.onclick = () => this.modal.remove();
|
||
|
||
header.appendChild(title);
|
||
header.appendChild(closeX);
|
||
|
||
this.body = document.createElement('div');
|
||
this.body.style.flex = '1';
|
||
this.body.style.display = 'flex';
|
||
this.body.style.overflow = 'hidden';
|
||
|
||
box.appendChild(header);
|
||
box.appendChild(this.body);
|
||
this.modal.appendChild(box);
|
||
document.body.appendChild(this.modal);
|
||
|
||
this.renderList();
|
||
}
|
||
|
||
async renderList() {
|
||
this.body.innerHTML = '';
|
||
|
||
const sidebar = document.createElement('div');
|
||
sidebar.style.width = '250px';
|
||
sidebar.style.borderRight = '1px solid #444';
|
||
sidebar.style.display = 'flex';
|
||
sidebar.style.flexDirection = 'column';
|
||
sidebar.style.background = '#1a1a1a';
|
||
|
||
const topBar = document.createElement('div');
|
||
topBar.style.padding = '10px';
|
||
topBar.style.borderBottom = '1px solid #333';
|
||
const newBtn = document.createElement('button');
|
||
newBtn.innerHTML = '➕ New Notebook';
|
||
newBtn.style.width = '100%';
|
||
newBtn.style.padding = '8px';
|
||
newBtn.style.background = '#4CAF50';
|
||
newBtn.style.color = '#fff';
|
||
newBtn.style.border = 'none';
|
||
newBtn.style.borderRadius = '4px';
|
||
newBtn.style.cursor = 'pointer';
|
||
newBtn.onclick = () => this.openEditor({ name: 'New_Notebook', baseModel: '', checkpoint: '', loras: [], prompt: '' }, true);
|
||
topBar.appendChild(newBtn);
|
||
sidebar.appendChild(topBar);
|
||
|
||
const listCont = document.createElement('div');
|
||
listCont.style.flex = '1';
|
||
listCont.style.overflowY = 'auto';
|
||
|
||
try {
|
||
const res = await fetch('/anomalous/notebooks');
|
||
const data = await res.json();
|
||
if (data.notebooks) {
|
||
data.notebooks.forEach(nb => {
|
||
const item = document.createElement('div');
|
||
item.style.padding = '12px 15px';
|
||
item.style.borderBottom = '1px solid #333';
|
||
item.style.cursor = 'pointer';
|
||
item.style.color = '#ccc';
|
||
item.innerHTML = `<strong>${nb.name}</strong><br><small style="color:#888;">${nb.baseModel || 'No Base'}</small>`;
|
||
item.onmouseover = () => item.style.background = '#333';
|
||
item.onmouseout = () => item.style.background = 'transparent';
|
||
item.onclick = () => this.openEditor(nb, false);
|
||
listCont.appendChild(item);
|
||
});
|
||
}
|
||
} catch(e) {}
|
||
|
||
sidebar.appendChild(listCont);
|
||
this.body.appendChild(sidebar);
|
||
|
||
this.editorArea = document.createElement('div');
|
||
this.editorArea.style.flex = '1';
|
||
this.editorArea.style.display = 'flex';
|
||
this.editorArea.style.flexDirection = 'column';
|
||
this.editorArea.style.background = '#222';
|
||
this.editorArea.style.padding = '20px';
|
||
this.editorArea.style.overflowY = 'auto';
|
||
this.editorArea.innerHTML = '<div style="color:#666; margin:auto;">Select or create a notebook to start</div>';
|
||
this.body.appendChild(this.editorArea);
|
||
}
|
||
|
||
openEditor(nb, isNew) {
|
||
this.currentNotebook = JSON.parse(JSON.stringify(nb));
|
||
this.editorArea.innerHTML = '';
|
||
|
||
// Toolbar
|
||
const tb = document.createElement('div');
|
||
tb.style.display = 'flex';
|
||
tb.style.gap = '10px';
|
||
tb.style.marginBottom = '15px';
|
||
|
||
const nameInput = document.createElement('input');
|
||
nameInput.value = nb.name;
|
||
nameInput.style.flex = '1';
|
||
nameInput.style.padding = '8px';
|
||
nameInput.style.background = '#111';
|
||
nameInput.style.color = '#fff';
|
||
nameInput.style.border = '1px solid #444';
|
||
nameInput.style.borderRadius = '4px';
|
||
|
||
const saveBtn = document.createElement('button');
|
||
saveBtn.innerHTML = '💾 Save';
|
||
saveBtn.style.padding = '8px 15px';
|
||
saveBtn.style.background = '#007bff';
|
||
saveBtn.style.color = '#fff';
|
||
saveBtn.style.border = 'none';
|
||
saveBtn.style.borderRadius = '4px';
|
||
saveBtn.style.cursor = 'pointer';
|
||
saveBtn.onclick = async () => {
|
||
this.currentNotebook.name = nameInput.value;
|
||
this.currentNotebook.prompt = this.promptArea.value;
|
||
await fetch('/anomalous/notebooks/save', {
|
||
method: 'POST', headers: {'Content-Type':'application/json'},
|
||
body: JSON.stringify(this.currentNotebook)
|
||
});
|
||
this.renderList();
|
||
alert('Saved successfully!');
|
||
};
|
||
|
||
const delBtn = document.createElement('button');
|
||
delBtn.innerHTML = '🗑️ Delete';
|
||
delBtn.style.padding = '8px 15px';
|
||
delBtn.style.background = '#ff4444';
|
||
delBtn.style.color = '#fff';
|
||
delBtn.style.border = 'none';
|
||
delBtn.style.borderRadius = '4px';
|
||
delBtn.style.cursor = 'pointer';
|
||
delBtn.onclick = async () => {
|
||
if(!confirm('Delete this notebook?')) return;
|
||
await fetch('/anomalous/notebooks/delete', {
|
||
method: 'POST', headers: {'Content-Type':'application/json'},
|
||
body: JSON.stringify({name: nb.name})
|
||
});
|
||
this.renderList();
|
||
};
|
||
|
||
const applyBtn = document.createElement('button');
|
||
applyBtn.innerHTML = '🚀 Send to Canvas';
|
||
applyBtn.style.padding = '8px 15px';
|
||
applyBtn.style.background = '#28a745';
|
||
applyBtn.style.color = '#fff';
|
||
applyBtn.style.border = 'none';
|
||
applyBtn.style.borderRadius = '4px';
|
||
applyBtn.style.cursor = 'pointer';
|
||
applyBtn.onclick = () => this.sendToCanvas();
|
||
|
||
tb.appendChild(nameInput);
|
||
tb.appendChild(saveBtn);
|
||
if (!isNew) tb.appendChild(delBtn);
|
||
tb.appendChild(applyBtn);
|
||
this.editorArea.appendChild(tb);
|
||
|
||
// Model Selection Gallery
|
||
const modelsContainer = document.createElement('div');
|
||
modelsContainer.style.display = 'flex';
|
||
modelsContainer.style.flexDirection = 'column';
|
||
modelsContainer.style.gap = '15px';
|
||
modelsContainer.style.marginBottom = '15px';
|
||
|
||
const baseRow = document.createElement('div');
|
||
const baseLabel = document.createElement('span');
|
||
baseLabel.innerText = 'Base Model: ';
|
||
baseLabel.style.color = '#ccc';
|
||
|
||
const baseSel = document.createElement('select');
|
||
const defaultBases = ['', 'SD 1.5', 'SDXL 1.0', 'Pony', 'Flux.1 Schnell', 'Flux.1 D', 'Illustrious'];
|
||
const allBases = Array.from(new Set([...defaultBases, ...(this.dynamicBases || [])])).filter(b => b !== 'Flux.1 S'); // Replaced Flux.1 S
|
||
|
||
allBases.forEach(b => {
|
||
const opt = document.createElement('option');
|
||
opt.value = b; opt.innerText = b || 'All Models (No Filter)';
|
||
if (b === nb.baseModel) opt.selected = true;
|
||
baseSel.appendChild(opt);
|
||
});
|
||
|
||
baseRow.appendChild(baseLabel);
|
||
baseRow.appendChild(baseSel);
|
||
modelsContainer.appendChild(baseRow);
|
||
|
||
const mainGallery = document.createElement('div');
|
||
const loraGallery = document.createElement('div');
|
||
modelsContainer.appendChild(mainGallery);
|
||
modelsContainer.appendChild(loraGallery);
|
||
|
||
const createGallery = (titleText, container, models, isMulti, selectedState, onSelectionChange) => {
|
||
container.innerHTML = `
|
||
<div style="color: #aaa; margin-bottom: 5px; font-weight: bold; font-size: 0.9em;">${titleText}</div>
|
||
<div class="nb-gallery-grid" style="display: flex; gap: 10px; overflow-x: auto; padding-bottom: 10px;"></div>
|
||
`;
|
||
const grid = container.querySelector('.nb-gallery-grid');
|
||
if (isMulti) {
|
||
grid.style.flexWrap = 'wrap';
|
||
grid.style.overflowX = 'hidden';
|
||
}
|
||
|
||
// Webkit scrollbar styles via JS injection
|
||
grid.style.cssText += 'scrollbar-width: thin; scrollbar-color: #555 #222;';
|
||
|
||
if (models.length === 0) {
|
||
grid.innerHTML = '<span style="color:#666; font-size: 0.8em;">No models found.</span>';
|
||
return;
|
||
}
|
||
|
||
models.forEach(m => {
|
||
const card = document.createElement('div');
|
||
card.style.minWidth = '100px';
|
||
card.style.width = '100px';
|
||
card.style.height = '130px';
|
||
card.style.background = '#222';
|
||
card.style.borderRadius = '6px';
|
||
card.style.cursor = 'pointer';
|
||
card.style.position = 'relative';
|
||
card.style.overflow = 'hidden';
|
||
card.style.border = '2px solid transparent';
|
||
card.style.transition = 'border-color 0.2s';
|
||
card.style.flexShrink = '0';
|
||
card.title = m.filename;
|
||
|
||
let thumbHTML = '';
|
||
let isVid = false;
|
||
if (m.preview_url) {
|
||
isVid = m.preview_url.toLowerCase().endsWith('.mp4') || m.preview_url.toLowerCase().endsWith('.webm');
|
||
if (isVid) thumbHTML = `<video src="${m.preview_url}" muted loop playsinline style="width:100%; height:90px; object-fit:cover; pointer-events:none;"></video>`;
|
||
else thumbHTML = `<img src="${m.preview_url}" style="width:100%; height:90px; object-fit:cover; pointer-events:none;" />`;
|
||
} else {
|
||
thumbHTML = `<div style="width:100%; height:90px; display:flex; align-items:center; justify-content:center; color:#555; font-size:20px; background:#1a1a1a;">?</div>`;
|
||
}
|
||
|
||
const shortName = m.filename.split('/').pop().split('\\').pop();
|
||
|
||
card.innerHTML = `
|
||
${thumbHTML}
|
||
<div style="font-size: 0.7em; padding: 4px; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; text-align: center;">${shortName}</div>
|
||
<div class="nb-badge" style="display:none; position:absolute; top:4px; right:4px; background:#6f42c1; color:#fff; border-radius:50%; width:18px; height:18px; font-size:10px; text-align:center; line-height:18px; font-weight:bold; pointer-events:none; z-index:2; box-shadow:0 0 5px rgba(0,0,0,0.5);"></div>
|
||
`;
|
||
|
||
let isSelectedLocally = false;
|
||
const updateState = () => {
|
||
const badge = card.querySelector('.nb-badge');
|
||
if (isMulti) {
|
||
const idx = selectedState.indexOf(m.filename);
|
||
if (idx !== -1) {
|
||
card.style.borderColor = '#6f42c1';
|
||
badge.style.display = 'block';
|
||
badge.innerText = (idx + 1).toString();
|
||
isSelectedLocally = true;
|
||
} else {
|
||
card.style.borderColor = 'transparent';
|
||
badge.style.display = 'none';
|
||
isSelectedLocally = false;
|
||
}
|
||
} else {
|
||
if (selectedState === m.filename) {
|
||
card.style.borderColor = '#4CAF50';
|
||
isSelectedLocally = true;
|
||
} else {
|
||
card.style.borderColor = 'transparent';
|
||
isSelectedLocally = false;
|
||
}
|
||
}
|
||
};
|
||
|
||
updateState();
|
||
|
||
if (isVid) {
|
||
card.onmouseenter = () => { const v = card.querySelector('video'); if(v) v.play(); };
|
||
card.onmouseleave = () => { const v = card.querySelector('video'); if(v) { v.pause(); v.currentTime=0; } };
|
||
}
|
||
|
||
card.onclick = () => {
|
||
if (isMulti) {
|
||
const idx = selectedState.indexOf(m.filename);
|
||
if (idx !== -1) {
|
||
selectedState.splice(idx, 1);
|
||
} else {
|
||
selectedState.push(m.filename);
|
||
}
|
||
onSelectionChange(selectedState);
|
||
} else {
|
||
// Single selection logic
|
||
selectedState = isSelectedLocally ? '' : m.filename;
|
||
onSelectionChange(selectedState);
|
||
}
|
||
};
|
||
|
||
grid.appendChild(card);
|
||
});
|
||
};
|
||
|
||
const filterModels = () => {
|
||
const currentBase = baseSel.value;
|
||
this.currentNotebook.baseModel = currentBase;
|
||
|
||
const filteredCkpts = this.checkpoints.filter(c => !currentBase || (c.baseModel === currentBase));
|
||
const filteredLoras = this.loras.filter(l => !currentBase || (l.baseModel === currentBase));
|
||
|
||
if (!this.currentNotebook.loras) this.currentNotebook.loras = [];
|
||
|
||
createGallery('Main Model (Select 1)', mainGallery, filteredCkpts, false, this.currentNotebook.checkpoint, (newVal) => {
|
||
this.currentNotebook.checkpoint = newVal;
|
||
filterModels();
|
||
});
|
||
|
||
createGallery('Loras (Click multiple to chain in order)', loraGallery, filteredLoras, true, this.currentNotebook.loras, (newArr) => {
|
||
this.currentNotebook.loras = newArr;
|
||
filterModels();
|
||
});
|
||
};
|
||
|
||
baseSel.onchange = filterModels;
|
||
filterModels();
|
||
|
||
this.editorArea.appendChild(modelsContainer);
|
||
|
||
// Find & Replace
|
||
const frDiv = document.createElement('div');
|
||
frDiv.style.display = 'flex';
|
||
frDiv.style.gap = '10px';
|
||
frDiv.style.marginBottom = '10px';
|
||
const fInput = document.createElement('input'); fInput.placeholder = 'Find...'; fInput.style.flex = '1'; fInput.style.padding = '6px';
|
||
const rInput = document.createElement('input'); rInput.placeholder = 'Replace...'; rInput.style.flex = '1'; rInput.style.padding = '6px';
|
||
const rBtn = document.createElement('button'); rBtn.innerText = 'Replace All'; rBtn.style.padding = '6px 12px';
|
||
rBtn.onclick = () => {
|
||
if (!fInput.value) return;
|
||
const regex = new RegExp(fInput.value, 'gi');
|
||
this.promptArea.value = this.promptArea.value.replace(regex, rInput.value);
|
||
this.currentNotebook.prompt = this.promptArea.value;
|
||
};
|
||
frDiv.appendChild(fInput); frDiv.appendChild(rInput); frDiv.appendChild(rBtn);
|
||
this.editorArea.appendChild(frDiv);
|
||
|
||
// Split view for translation
|
||
const splitView = document.createElement('div');
|
||
splitView.style.flex = '1';
|
||
splitView.style.display = 'flex';
|
||
splitView.style.gap = '15px';
|
||
splitView.style.overflow = 'hidden';
|
||
splitView.style.minHeight = '400px';
|
||
|
||
const leftCol = document.createElement('div');
|
||
leftCol.style.flex = '1';
|
||
leftCol.style.display = 'flex';
|
||
leftCol.style.flexDirection = 'column';
|
||
leftCol.style.overflowY = 'auto';
|
||
|
||
this.promptArea = document.createElement('textarea');
|
||
this.promptArea.value = nb.prompt || '';
|
||
this.promptArea.style.flex = '1';
|
||
this.promptArea.style.background = '#111';
|
||
this.promptArea.style.color = '#fff';
|
||
this.promptArea.style.border = '1px solid #444';
|
||
this.promptArea.style.padding = '10px';
|
||
this.promptArea.style.resize = 'none';
|
||
this.promptArea.style.fontFamily = 'monospace';
|
||
|
||
const transBtn = document.createElement('button');
|
||
transBtn.innerHTML = '🌐 Translate & Link';
|
||
transBtn.style.padding = '8px';
|
||
transBtn.style.marginTop = '10px';
|
||
transBtn.style.background = '#6f42c1';
|
||
transBtn.style.color = '#fff';
|
||
transBtn.style.border = 'none';
|
||
transBtn.style.borderRadius = '4px';
|
||
transBtn.style.cursor = 'pointer';
|
||
|
||
leftCol.appendChild(this.promptArea);
|
||
leftCol.appendChild(transBtn);
|
||
|
||
const rightCol = document.createElement('div');
|
||
rightCol.style.flex = '1';
|
||
rightCol.style.background = '#111';
|
||
rightCol.style.border = '1px solid #444';
|
||
rightCol.style.borderRadius = '4px';
|
||
rightCol.style.padding = '10px';
|
||
rightCol.style.overflowY = 'auto';
|
||
|
||
splitView.appendChild(leftCol);
|
||
splitView.appendChild(rightCol);
|
||
this.editorArea.appendChild(splitView);
|
||
|
||
// Translate Logic
|
||
transBtn.onclick = async () => {
|
||
const text = this.promptArea.value.trim();
|
||
if (!text) return;
|
||
|
||
// Tag split mode
|
||
const tags = text.split(',').map(t => t.trim());
|
||
const payload = tags.join('\n');
|
||
|
||
transBtn.innerHTML = '⏳ Translating...';
|
||
try {
|
||
const res = await fetch('/anomalous/translate', {
|
||
method: 'POST', headers: {'Content-Type':'application/json'},
|
||
body: JSON.stringify({text: payload, target: 'zh-CN'})
|
||
});
|
||
const data = await res.json();
|
||
|
||
const zhTags = data.translated.split('\n').map(t => t.trim());
|
||
|
||
// Render interactive viewer
|
||
leftCol.innerHTML = '';
|
||
rightCol.innerHTML = '';
|
||
|
||
const backBtn = document.createElement('button');
|
||
backBtn.innerHTML = '✏️ Edit Text';
|
||
backBtn.style.padding = '8px';
|
||
backBtn.style.marginBottom = '10px';
|
||
backBtn.onclick = () => {
|
||
leftCol.innerHTML = ''; rightCol.innerHTML = '';
|
||
leftCol.appendChild(this.promptArea); leftCol.appendChild(transBtn);
|
||
};
|
||
leftCol.appendChild(backBtn);
|
||
|
||
const leftTags = document.createElement('div');
|
||
const rightTags = document.createElement('div');
|
||
|
||
tags.forEach((enTag, i) => {
|
||
if(!enTag) return;
|
||
const zhTag = zhTags[i] || '';
|
||
|
||
const el = document.createElement('span');
|
||
el.innerText = enTag + ', ';
|
||
el.style.display = 'inline-block';
|
||
el.style.padding = '2px';
|
||
el.style.margin = '2px';
|
||
el.style.borderRadius = '3px';
|
||
el.style.cursor = 'pointer';
|
||
|
||
const zl = document.createElement('span');
|
||
zl.innerText = zhTag + '，';
|
||
zl.style.display = 'inline-block';
|
||
zl.style.padding = '2px';
|
||
zl.style.margin = '2px';
|
||
zl.style.borderRadius = '3px';
|
||
|
||
el.onmouseover = () => { el.style.background = '#4CAF50'; zl.style.background = '#4CAF50'; };
|
||
el.onmouseout = () => { el.style.background = 'transparent'; zl.style.background = 'transparent'; };
|
||
zl.onmouseover = el.onmouseover;
|
||
zl.onmouseout = el.onmouseout;
|
||
|
||
leftTags.appendChild(el);
|
||
rightTags.appendChild(zl);
|
||
});
|
||
|
||
leftCol.appendChild(leftTags);
|
||
rightCol.appendChild(rightTags);
|
||
|
||
} catch(e) {
|
||
alert('Translation failed.');
|
||
}
|
||
transBtn.innerHTML = '🌐 Translate & Link';
|
||
};
|
||
}
|
||
|
||
sendToCanvas() {
|
||
if (!window.LiteGraph || !app.graph) return;
|
||
|
||
let x = window.innerWidth / 2;
|
||
let y = window.innerHeight / 2;
|
||
|
||
if (app.canvas && app.canvas.graph_mouse) {
|
||
x = app.canvas.graph_mouse[0] || x;
|
||
y = app.canvas.graph_mouse[1] || y;
|
||
}
|
||
|
||
let ckptNode = null;
|
||
let currentModelNode = null;
|
||
let currentModelSlot = 0;
|
||
let currentClipNode = null;
|
||
let currentClipSlot = 1;
|
||
|
||
// 1. Create Checkpoint Node
|
||
if (this.currentNotebook.checkpoint) {
|
||
ckptNode = LiteGraph.createNode("CheckpointLoaderSimple");
|
||
if (ckptNode) {
|
||
ckptNode.pos = [x, y];
|
||
app.graph.add(ckptNode);
|
||
if (ckptNode.widgets && ckptNode.widgets.length > 0) {
|
||
ckptNode.widgets[0].value = this.currentNotebook.checkpoint;
|
||
}
|
||
currentModelNode = ckptNode;
|
||
currentModelSlot = 0;
|
||
currentClipNode = ckptNode;
|
||
currentClipSlot = 1;
|
||
}
|
||
}
|
||
|
||
// 2. Add Loras
|
||
let loraX = x + 300;
|
||
if (this.currentNotebook.loras && this.currentNotebook.loras.length > 0) {
|
||
this.currentNotebook.loras.forEach(lora => {
|
||
if (!lora) return;
|
||
const loraNode = LiteGraph.createNode("LoraLoader");
|
||
if (loraNode) {
|
||
loraNode.pos = [loraX, y];
|
||
app.graph.add(loraNode);
|
||
if (loraNode.widgets && loraNode.widgets.length > 0) {
|
||
loraNode.widgets[0].value = lora;
|
||
}
|
||
|
||
// Connect from previous
|
||
if (currentModelNode) {
|
||
currentModelNode.connect(currentModelSlot, loraNode, 0); // MODEL
|
||
}
|
||
if (currentClipNode) {
|
||
currentClipNode.connect(currentClipSlot, loraNode, 1); // CLIP
|
||
}
|
||
|
||
// Update current pointers
|
||
currentModelNode = loraNode;
|
||
currentModelSlot = 0;
|
||
currentClipNode = loraNode;
|
||
currentClipSlot = 1;
|
||
|
||
loraX += 300;
|
||
}
|
||
});
|
||
}
|
||
|
||
let dragNode = currentClipNode || ckptNode;
|
||
|
||
// 3. Create Prompt Node
|
||
if (this.currentNotebook.prompt) {
|
||
const promptNode = LiteGraph.createNode("CLIPTextEncode");
|
||
if (promptNode) {
|
||
promptNode.pos = [loraX, y + 200];
|
||
app.graph.add(promptNode);
|
||
if (promptNode.widgets && promptNode.widgets.length > 0) {
|
||
promptNode.widgets[0].value = this.currentNotebook.prompt;
|
||
}
|
||
if (currentClipNode) {
|
||
currentClipNode.connect(currentClipSlot, promptNode, 0); // connect CLIP
|
||
}
|
||
dragNode = promptNode;
|
||
}
|
||
}
|
||
|
||
this.modal.remove();
|
||
|
||
// Make it stick to mouse if app.canvas exists
|
||
if (lastNode && app.canvas) {
|
||
let isSticking = true;
|
||
const stickHandler = (e) => {
|
||
if (!isSticking || !app.canvas) return;
|
||
const canvas = app.canvas;
|
||
let canvasX, canvasY;
|
||
if (canvas.convertEventToCanvasOffset) {
|
||
const pos = canvas.convertEventToCanvasOffset(e);
|
||
canvasX = pos[0]; canvasY = pos[1];
|
||
} else {
|
||
const rect = canvas.canvas.getBoundingClientRect();
|
||
canvasX = (e.clientX - rect.left - canvas.ds.offset[0]) / canvas.ds.scale;
|
||
canvasY = (e.clientY - rect.top - canvas.ds.offset[1]) / canvas.ds.scale;
|
||
}
|
||
dragNode.pos = [canvasX - dragNode.size[0] / 2, canvasY - 20];
|
||
canvas.setDirty(true, true);
|
||
};
|
||
const dropHandler = (e) => {
|
||
isSticking = false;
|
||
window.removeEventListener('mousemove', stickHandler, true);
|
||
window.removeEventListener('pointerdown', dropHandler, true);
|
||
window.removeEventListener('mousedown', dropHandler, true);
|
||
window.removeEventListener('click', dropHandler, true);
|
||
};
|
||
window.addEventListener('mousemove', stickHandler, true);
|
||
setTimeout(() => {
|
||
window.addEventListener('pointerdown', dropHandler, true);
|
||
window.addEventListener('mousedown', dropHandler, true);
|
||
window.addEventListener('click', dropHandler, true);
|
||
}, 100);
|
||
}
|
||
}
|
||
}
|