mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-07-05 17:01:16 -03:00
Compare commits
8 Commits
v1.0.11
...
5a4664fa12
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5a4664fa12 | ||
|
|
dd5b213adc | ||
|
|
d9ee9b3155 | ||
|
|
01dac57c35 | ||
|
|
7f92d09239 | ||
|
|
62f9e3f44a | ||
|
|
e55895786d | ||
|
|
4e3ede23b7 |
@@ -81,7 +81,7 @@ def read_safetensors_metadata(file_path: str) -> dict[str, Any]:
|
|||||||
return {}
|
return {}
|
||||||
header = json.loads(header_bytes.decode("utf-8"))
|
header = json.loads(header_bytes.decode("utf-8"))
|
||||||
return header.get("__metadata__", {})
|
return header.get("__metadata__", {})
|
||||||
except (OSError, json.JSONDecodeError, UnicodeDecodeError, struct.error):
|
except (OSError, json.JSONDecodeError, UnicodeDecodeError, struct.error, MemoryError, Exception):
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -34,6 +34,8 @@ import sys
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
|
from platformdirs import user_config_dir
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
format="%(asctime)s - %(levelname)s - %(message)s",
|
format="%(asctime)s - %(levelname)s - %(message)s",
|
||||||
@@ -53,10 +55,7 @@ def resolve_settings_path() -> Path:
|
|||||||
if isinstance(payload, dict) and payload.get("use_portable_settings") is True:
|
if isinstance(payload, dict) and payload.get("use_portable_settings") is True:
|
||||||
return portable
|
return portable
|
||||||
|
|
||||||
config_home = os.environ.get("XDG_CONFIG_HOME")
|
return Path(user_config_dir(APP_NAME, appauthor=False)) / "settings.json"
|
||||||
if config_home:
|
|
||||||
return Path(config_home).expanduser() / APP_NAME / "settings.json"
|
|
||||||
return Path.home() / ".config" / APP_NAME / "settings.json"
|
|
||||||
|
|
||||||
|
|
||||||
def load_json(path: Path) -> dict[str, Any]:
|
def load_json(path: Path) -> dict[str, Any]:
|
||||||
|
|||||||
@@ -39,6 +39,8 @@ import sys
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
|
from platformdirs import user_config_dir
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
format="%(message)s",
|
format="%(message)s",
|
||||||
@@ -68,10 +70,7 @@ def resolve_settings_path() -> Path:
|
|||||||
if isinstance(payload, dict) and payload.get("use_portable_settings") is True:
|
if isinstance(payload, dict) and payload.get("use_portable_settings") is True:
|
||||||
return portable
|
return portable
|
||||||
|
|
||||||
config_home = os.environ.get("XDG_CONFIG_HOME")
|
return Path(user_config_dir(APP_NAME, appauthor=False)) / "settings.json"
|
||||||
if config_home:
|
|
||||||
return Path(config_home).expanduser() / APP_NAME / "settings.json"
|
|
||||||
return Path.home() / ".config" / APP_NAME / "settings.json"
|
|
||||||
|
|
||||||
|
|
||||||
def _load_json(path: Path) -> dict[str, Any]:
|
def _load_json(path: Path) -> dict[str, Any]:
|
||||||
|
|||||||
@@ -668,4 +668,143 @@
|
|||||||
[data-theme="dark"] .file-tag.size {
|
[data-theme="dark"] .file-tag.size {
|
||||||
background: oklch(0.5 0.08 160 / 0.15);
|
background: oklch(0.5 0.08 160 / 0.15);
|
||||||
color: oklch(0.65 0.08 160);
|
color: oklch(0.65 0.08 160);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Textarea for multi-URL input */
|
||||||
|
#modelUrl {
|
||||||
|
width: 100%;
|
||||||
|
padding: 8px;
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: var(--border-radius-xs);
|
||||||
|
background: var(--bg-color);
|
||||||
|
color: var(--text-color);
|
||||||
|
font-family: monospace;
|
||||||
|
font-size: 0.9em;
|
||||||
|
resize: vertical;
|
||||||
|
line-height: 1.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Batch Preview List */
|
||||||
|
.batch-preview-list {
|
||||||
|
max-height: 400px;
|
||||||
|
overflow-y: auto;
|
||||||
|
margin: var(--space-2) 0;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 1px;
|
||||||
|
background: var(--border-color);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: var(--border-radius-sm);
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 12px;
|
||||||
|
padding: 10px 12px;
|
||||||
|
background: var(--bg-color);
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-item:first-child {
|
||||||
|
border-radius: var(--border-radius-sm) var(--border-radius-sm) 0 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-item:last-child {
|
||||||
|
border-radius: 0 0 var(--border-radius-sm) var(--border-radius-sm);
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-item:only-child {
|
||||||
|
border-radius: var(--border-radius-sm);
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-thumbnail {
|
||||||
|
width: 48px;
|
||||||
|
height: 48px;
|
||||||
|
flex-shrink: 0;
|
||||||
|
border-radius: var(--border-radius-xs);
|
||||||
|
overflow: hidden;
|
||||||
|
background: var(--lora-surface);
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-thumbnail img {
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: cover;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-icon {
|
||||||
|
width: 48px;
|
||||||
|
height: 48px;
|
||||||
|
flex-shrink: 0;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
color: #e74c3c;
|
||||||
|
font-size: 1.2em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-info {
|
||||||
|
flex: 1;
|
||||||
|
min-width: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-name {
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-color);
|
||||||
|
white-space: nowrap;
|
||||||
|
overflow: hidden;
|
||||||
|
text-overflow: ellipsis;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-meta {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
font-size: 0.85em;
|
||||||
|
color: var(--text-color);
|
||||||
|
opacity: 0.7;
|
||||||
|
margin-top: 2px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-error-text {
|
||||||
|
color: #e74c3c;
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-local-badge {
|
||||||
|
color: var(--lora-accent);
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-local {
|
||||||
|
opacity: 0.6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-change-version {
|
||||||
|
flex-shrink: 0;
|
||||||
|
font-size: 0.85em;
|
||||||
|
padding: 4px 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-remove {
|
||||||
|
flex-shrink: 0;
|
||||||
|
background: none;
|
||||||
|
border: none;
|
||||||
|
color: var(--text-color);
|
||||||
|
opacity: 0.5;
|
||||||
|
cursor: pointer;
|
||||||
|
padding: 4px 8px;
|
||||||
|
font-size: 1em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-remove:hover {
|
||||||
|
opacity: 1;
|
||||||
|
color: #e74c3c;
|
||||||
|
}
|
||||||
|
|
||||||
|
.batch-preview-error {
|
||||||
|
background: oklch(0.5 0.15 25 / 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
[data-theme="dark"] .batch-preview-item {
|
||||||
|
background: var(--lora-surface);
|
||||||
}
|
}
|
||||||
@@ -22,6 +22,11 @@ export class DownloadManager {
|
|||||||
this.apiClient = null;
|
this.apiClient = null;
|
||||||
this.useDefaultPath = false;
|
this.useDefaultPath = false;
|
||||||
|
|
||||||
|
// Batch mode state
|
||||||
|
this.batchModels = [];
|
||||||
|
this.isBatchMode = false;
|
||||||
|
this.editingBatchIndex = -1;
|
||||||
|
|
||||||
this.loadingManager = new LoadingManager();
|
this.loadingManager = new LoadingManager();
|
||||||
this.folderTreeManager = new FolderTreeManager();
|
this.folderTreeManager = new FolderTreeManager();
|
||||||
this.folderClickHandler = null;
|
this.folderClickHandler = null;
|
||||||
@@ -37,6 +42,8 @@ export class DownloadManager {
|
|||||||
this.handleConfirmFileSelection = this.confirmFileSelection.bind(this);
|
this.handleConfirmFileSelection = this.confirmFileSelection.bind(this);
|
||||||
this.handleCloseModal = this.closeModal.bind(this);
|
this.handleCloseModal = this.closeModal.bind(this);
|
||||||
this.handleToggleDefaultPath = this.toggleDefaultPath.bind(this);
|
this.handleToggleDefaultPath = this.toggleDefaultPath.bind(this);
|
||||||
|
this.handleBackToUrlFromBatch = this.backToUrlFromBatch.bind(this);
|
||||||
|
this.handleNextFromBatch = this.nextFromBatch.bind(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
showDownloadModal() {
|
showDownloadModal() {
|
||||||
@@ -86,6 +93,10 @@ export class DownloadManager {
|
|||||||
document.getElementById('backToVersionFromFilesBtn').addEventListener('click', this.handleBackToVersionFromFiles);
|
document.getElementById('backToVersionFromFilesBtn').addEventListener('click', this.handleBackToVersionFromFiles);
|
||||||
document.getElementById('confirmFileSelection').addEventListener('click', this.handleConfirmFileSelection);
|
document.getElementById('confirmFileSelection').addEventListener('click', this.handleConfirmFileSelection);
|
||||||
|
|
||||||
|
// Batch preview buttons
|
||||||
|
document.getElementById('backToUrlFromBatchBtn').addEventListener('click', this.handleBackToUrlFromBatch);
|
||||||
|
document.getElementById('nextFromBatchBtn').addEventListener('click', this.handleNextFromBatch);
|
||||||
|
|
||||||
// Default path toggle handler
|
// Default path toggle handler
|
||||||
document.getElementById('useDefaultPath').addEventListener('change', this.handleToggleDefaultPath);
|
document.getElementById('useDefaultPath').addEventListener('change', this.handleToggleDefaultPath);
|
||||||
}
|
}
|
||||||
@@ -138,6 +149,9 @@ export class DownloadManager {
|
|||||||
this.selectedFile = null;
|
this.selectedFile = null;
|
||||||
|
|
||||||
this.selectedFolder = '';
|
this.selectedFolder = '';
|
||||||
|
this.batchModels = [];
|
||||||
|
this.isBatchMode = false;
|
||||||
|
this.editingBatchIndex = -1;
|
||||||
|
|
||||||
// Clear folder tree selection
|
// Clear folder tree selection
|
||||||
if (this.folderTreeManager) {
|
if (this.folderTreeManager) {
|
||||||
@@ -157,30 +171,99 @@ export class DownloadManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async validateAndFetchVersions() {
|
async validateAndFetchVersions() {
|
||||||
const url = document.getElementById('modelUrl').value.trim();
|
const rawText = document.getElementById('modelUrl').value.trim();
|
||||||
const errorElement = document.getElementById('urlError');
|
const errorElement = document.getElementById('urlError');
|
||||||
|
const urls = rawText.split('\n').map(l => l.trim()).filter(Boolean);
|
||||||
|
|
||||||
try {
|
if (urls.length === 0) {
|
||||||
this.loadingManager.showSimpleLoading(translate('modals.download.fetchingVersions'));
|
errorElement.textContent = translate('modals.download.errors.invalidUrl');
|
||||||
|
return;
|
||||||
this.modelId = this.extractModelId(url);
|
|
||||||
if (!this.modelId) {
|
|
||||||
throw new Error(translate('modals.download.errors.invalidUrl'));
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.retrieveVersionsForModel(this.modelId, this.source);
|
|
||||||
|
|
||||||
// If we have a version ID from URL, pre-select it
|
|
||||||
if (this.modelVersionId) {
|
|
||||||
this.currentVersion = this.versions.find(v => v.id.toString() === this.modelVersionId);
|
|
||||||
}
|
|
||||||
|
|
||||||
this.showVersionStep();
|
|
||||||
} catch (error) {
|
|
||||||
errorElement.textContent = error.message;
|
|
||||||
} finally {
|
|
||||||
this.loadingManager.hide();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (urls.length === 1) {
|
||||||
|
this.isBatchMode = false;
|
||||||
|
try {
|
||||||
|
this.loadingManager.showSimpleLoading(translate('modals.download.fetchingVersions'));
|
||||||
|
|
||||||
|
this.modelId = this.extractModelId(urls[0]);
|
||||||
|
if (!this.modelId) {
|
||||||
|
throw new Error(translate('modals.download.errors.invalidUrl'));
|
||||||
|
}
|
||||||
|
|
||||||
|
await this.retrieveVersionsForModel(this.modelId, this.source);
|
||||||
|
|
||||||
|
if (this.modelVersionId) {
|
||||||
|
this.currentVersion = this.versions.find(v => v.id.toString() === this.modelVersionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
this.showVersionStep();
|
||||||
|
} catch (error) {
|
||||||
|
errorElement.textContent = error.message;
|
||||||
|
} finally {
|
||||||
|
this.loadingManager.hide();
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Multi-URL batch mode
|
||||||
|
this.isBatchMode = true;
|
||||||
|
this.batchModels = [];
|
||||||
|
errorElement.textContent = '';
|
||||||
|
|
||||||
|
const seen = new Set();
|
||||||
|
const parsed = [];
|
||||||
|
for (const url of urls) {
|
||||||
|
const result = DownloadManager.parseModelUrl(url);
|
||||||
|
if (!result.modelId) {
|
||||||
|
parsed.push({ url, error: translate('modals.download.errors.invalidUrl') });
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (seen.has(result.modelId)) continue;
|
||||||
|
seen.add(result.modelId);
|
||||||
|
parsed.push({ url, ...result, error: null });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (parsed.length === 0) {
|
||||||
|
errorElement.textContent = translate('modals.download.errors.invalidUrl');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.loadingManager.showSimpleLoading(translate('modals.download.fetchingVersions'));
|
||||||
|
|
||||||
|
let fetched = 0;
|
||||||
|
const total = parsed.filter(p => !p.error).length;
|
||||||
|
|
||||||
|
this.batchModels = new Array(parsed.length);
|
||||||
|
|
||||||
|
const fetchPromises = parsed.map(async (item, index) => {
|
||||||
|
if (item.error) {
|
||||||
|
this.batchModels[index] = { ...item, versions: [], selectedVersion: null };
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const versions = await this.apiClient.fetchCivitaiVersions(item.modelId, item.source);
|
||||||
|
fetched++;
|
||||||
|
this.loadingManager.setStatus(`${fetched}/${total}`);
|
||||||
|
|
||||||
|
let selectedVersion = null;
|
||||||
|
if (versions && versions.length > 0) {
|
||||||
|
if (item.modelVersionId) {
|
||||||
|
selectedVersion = versions.find(v => v.id.toString() === item.modelVersionId) || versions[0];
|
||||||
|
} else {
|
||||||
|
selectedVersion = versions[0];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
this.batchModels[index] = { ...item, versions: versions || [], selectedVersion };
|
||||||
|
} catch (err) {
|
||||||
|
this.batchModels[index] = { ...item, versions: [], selectedVersion: null, error: err.message };
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
await Promise.all(fetchPromises);
|
||||||
|
this.loadingManager.hide();
|
||||||
|
|
||||||
|
this.showBatchPreviewStep();
|
||||||
}
|
}
|
||||||
|
|
||||||
async fetchVersionsForCurrentModel() {
|
async fetchVersionsForCurrentModel() {
|
||||||
@@ -204,25 +287,30 @@ export class DownloadManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
extractModelId(url) {
|
static parseModelUrl(url) {
|
||||||
const civarchiveMatch = url.match(/https?:\/\/(?:www\.)?(?:civitaiarchive|civarchive)\.com\/models\/(\d+)/i);
|
const civarchiveMatch = url.match(/https?:\/\/(?:www\.)?(?:civitaiarchive|civarchive)\.com\/models\/(\d+)/i);
|
||||||
if (civarchiveMatch) {
|
if (civarchiveMatch) {
|
||||||
const versionMatch = url.match(/modelVersionId=(\d+)/i);
|
const versionMatch = url.match(/modelVersionId=(\d+)/i);
|
||||||
this.modelVersionId = versionMatch ? versionMatch[1] : null;
|
return {
|
||||||
this.source = 'civarchive';
|
modelId: civarchiveMatch[1],
|
||||||
return civarchiveMatch[1];
|
modelVersionId: versionMatch ? versionMatch[1] : null,
|
||||||
|
source: 'civarchive',
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
const { modelId, modelVersionId } = extractCivitaiModelUrlParts(url);
|
const { modelId, modelVersionId } = extractCivitaiModelUrlParts(url);
|
||||||
if (modelId) {
|
if (modelId) {
|
||||||
this.modelVersionId = modelVersionId;
|
return { modelId, modelVersionId, source: null };
|
||||||
this.source = null;
|
|
||||||
return modelId;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
this.modelVersionId = null;
|
return { modelId: null, modelVersionId: null, source: null };
|
||||||
this.source = null;
|
}
|
||||||
return null;
|
|
||||||
|
extractModelId(url) {
|
||||||
|
const result = DownloadManager.parseModelUrl(url);
|
||||||
|
this.modelVersionId = result.modelVersionId;
|
||||||
|
this.source = result.source;
|
||||||
|
return result.modelId;
|
||||||
}
|
}
|
||||||
|
|
||||||
async openForModelVersion(modelType, modelId, versionId = null) {
|
async openForModelVersion(modelType, modelId, versionId = null) {
|
||||||
@@ -250,7 +338,10 @@ export class DownloadManager {
|
|||||||
document.getElementById('versionStep').style.display = 'block';
|
document.getElementById('versionStep').style.display = 'block';
|
||||||
|
|
||||||
const versionList = document.getElementById('versionList');
|
const versionList = document.getElementById('versionList');
|
||||||
versionList.innerHTML = this.versions.map(version => {
|
const newList = versionList.cloneNode(false);
|
||||||
|
versionList.parentNode.replaceChild(newList, versionList);
|
||||||
|
|
||||||
|
newList.innerHTML = this.versions.map(version => {
|
||||||
const firstImage = version.images?.find(img => !img.url.endsWith('.mp4'));
|
const firstImage = version.images?.find(img => !img.url.endsWith('.mp4'));
|
||||||
const thumbnailUrl = firstImage ? firstImage.url : '/loras_static/images/no-preview.png';
|
const thumbnailUrl = firstImage ? firstImage.url : '/loras_static/images/no-preview.png';
|
||||||
|
|
||||||
@@ -326,7 +417,7 @@ export class DownloadManager {
|
|||||||
}).join('');
|
}).join('');
|
||||||
|
|
||||||
// Add click handlers for version selection and file badge
|
// Add click handlers for version selection and file badge
|
||||||
versionList.addEventListener('click', (event) => {
|
newList.addEventListener('click', (event) => {
|
||||||
const badge = event.target.closest('.file-select-badge');
|
const badge = event.target.closest('.file-select-badge');
|
||||||
if (badge) {
|
if (badge) {
|
||||||
event.stopPropagation();
|
event.stopPropagation();
|
||||||
@@ -452,18 +543,30 @@ export class DownloadManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async proceedToLocation() {
|
async proceedToLocation() {
|
||||||
if (!this.currentVersion) {
|
// If editing a batch item's version, save and return to batch preview
|
||||||
showToast('toast.loras.pleaseSelectVersion', {}, 'error');
|
if (this.isBatchMode && this.editingBatchIndex >= 0) {
|
||||||
|
if (this.currentVersion) {
|
||||||
|
this.batchModels[this.editingBatchIndex].selectedVersion = this.currentVersion;
|
||||||
|
}
|
||||||
|
this.editingBatchIndex = -1;
|
||||||
|
document.getElementById('versionStep').style.display = 'none';
|
||||||
|
this.showBatchPreviewStep();
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const existsLocally = this.currentVersion.existsLocally;
|
// In single-URL mode, validate version selection
|
||||||
if (existsLocally) {
|
if (!this.isBatchMode) {
|
||||||
showToast('toast.loras.versionExists', {}, 'info');
|
if (!this.currentVersion) {
|
||||||
return;
|
showToast('toast.loras.pleaseSelectVersion', {}, 'error');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (this.currentVersion.existsLocally) {
|
||||||
|
showToast('toast.loras.versionExists', {}, 'info');
|
||||||
|
return;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
document.getElementById('versionStep').style.display = 'none';
|
document.querySelectorAll('.download-step').forEach(step => step.style.display = 'none');
|
||||||
document.getElementById('locationStep').style.display = 'block';
|
document.getElementById('locationStep').style.display = 'block';
|
||||||
await this.proceedToLocationContent();
|
await this.proceedToLocationContent();
|
||||||
}
|
}
|
||||||
@@ -700,14 +803,123 @@ export class DownloadManager {
|
|||||||
this.updateTargetPath();
|
this.updateTargetPath();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
showBatchPreviewStep() {
|
||||||
|
document.querySelectorAll('.download-step').forEach(step => step.style.display = 'none');
|
||||||
|
document.getElementById('batchPreviewStep').style.display = 'block';
|
||||||
|
|
||||||
|
const validCount = this.batchModels.filter(m => !m.error && m.selectedVersion).length;
|
||||||
|
document.getElementById('downloadModalTitle').textContent =
|
||||||
|
translate('modals.download.titleWithType', { type: this.apiClient.apiConfig.config.displayName }) +
|
||||||
|
` (${validCount})`;
|
||||||
|
|
||||||
|
const list = document.getElementById('batchPreviewList');
|
||||||
|
list.innerHTML = this.batchModels.map((item, index) => {
|
||||||
|
if (item.error) {
|
||||||
|
return `
|
||||||
|
<div class="batch-preview-item batch-preview-error" data-index="${index}">
|
||||||
|
<div class="batch-preview-icon">
|
||||||
|
<i class="fas fa-exclamation-triangle"></i>
|
||||||
|
</div>
|
||||||
|
<div class="batch-preview-info">
|
||||||
|
<div class="batch-preview-name">${item.url}</div>
|
||||||
|
<div class="batch-preview-meta batch-preview-error-text">${item.error}</div>
|
||||||
|
</div>
|
||||||
|
<button class="batch-preview-remove" data-index="${index}" title="${translate('common.actions.remove', {}, 'Remove')}">
|
||||||
|
<i class="fas fa-times"></i>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ver = item.selectedVersion;
|
||||||
|
const firstImage = ver?.images?.find(img => !img.url.endsWith('.mp4'));
|
||||||
|
const thumbnailUrl = firstImage ? firstImage.url : '/loras_static/images/no-preview.png';
|
||||||
|
const fileSize = ver?.modelSizeKB
|
||||||
|
? (ver.modelSizeKB / 1024).toFixed(1)
|
||||||
|
: (ver?.files?.[0]?.sizeKB ? (ver.files[0].sizeKB / 1024).toFixed(1) : '?');
|
||||||
|
const existsLocally = ver?.existsLocally;
|
||||||
|
|
||||||
|
return `
|
||||||
|
<div class="batch-preview-item ${existsLocally ? 'batch-preview-local' : ''}" data-index="${index}">
|
||||||
|
<div class="batch-preview-thumbnail">
|
||||||
|
<img src="${thumbnailUrl}" alt="">
|
||||||
|
</div>
|
||||||
|
<div class="batch-preview-info">
|
||||||
|
<div class="batch-preview-name">${ver?.name || `Model #${item.modelId}`}</div>
|
||||||
|
<div class="batch-preview-meta">
|
||||||
|
${ver?.baseModel ? `<span>${ver.baseModel}</span>` : ''}
|
||||||
|
<span>${fileSize} MB</span>
|
||||||
|
${existsLocally ? `<span class="batch-preview-local-badge"><i class="fas fa-check"></i> ${translate('modals.download.inLibrary')}</span>` : ''}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
${item.versions.length > 1 ? `
|
||||||
|
<button class="batch-preview-change-version secondary-btn" data-index="${index}">
|
||||||
|
${translate('common.actions.change', {}, 'Change')}
|
||||||
|
</button>
|
||||||
|
` : ''}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}).join('');
|
||||||
|
|
||||||
|
list.onclick = (e) => {
|
||||||
|
const removeBtn = e.target.closest('.batch-preview-remove');
|
||||||
|
if (removeBtn) {
|
||||||
|
const idx = parseInt(removeBtn.dataset.index);
|
||||||
|
this.batchModels.splice(idx, 1);
|
||||||
|
this.showBatchPreviewStep();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const changeBtn = e.target.closest('.batch-preview-change-version');
|
||||||
|
if (changeBtn) {
|
||||||
|
const idx = parseInt(changeBtn.dataset.index);
|
||||||
|
this.openBatchVersionEditor(idx);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const nextBtn = document.getElementById('nextFromBatchBtn');
|
||||||
|
nextBtn.disabled = validCount === 0;
|
||||||
|
nextBtn.classList.toggle('disabled', validCount === 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
openBatchVersionEditor(index) {
|
||||||
|
this.editingBatchIndex = index;
|
||||||
|
const item = this.batchModels[index];
|
||||||
|
|
||||||
|
this.versions = item.versions;
|
||||||
|
this.currentVersion = item.selectedVersion;
|
||||||
|
|
||||||
|
document.getElementById('batchPreviewStep').style.display = 'none';
|
||||||
|
this.showVersionStep();
|
||||||
|
}
|
||||||
|
|
||||||
|
backToUrlFromBatch() {
|
||||||
|
document.getElementById('batchPreviewStep').style.display = 'none';
|
||||||
|
document.getElementById('urlStep').style.display = 'block';
|
||||||
|
}
|
||||||
|
|
||||||
|
nextFromBatch() {
|
||||||
|
const validModels = this.batchModels.filter(m => !m.error && m.selectedVersion);
|
||||||
|
if (validModels.length === 0) return;
|
||||||
|
this.proceedToLocation();
|
||||||
|
}
|
||||||
|
|
||||||
backToUrl() {
|
backToUrl() {
|
||||||
document.getElementById('versionStep').style.display = 'none';
|
document.getElementById('versionStep').style.display = 'none';
|
||||||
document.getElementById('urlStep').style.display = 'block';
|
if (this.isBatchMode && this.editingBatchIndex >= 0) {
|
||||||
|
this.editingBatchIndex = -1;
|
||||||
|
this.showBatchPreviewStep();
|
||||||
|
} else {
|
||||||
|
document.getElementById('urlStep').style.display = 'block';
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
backToVersions() {
|
backToVersions() {
|
||||||
document.getElementById('locationStep').style.display = 'none';
|
document.getElementById('locationStep').style.display = 'none';
|
||||||
document.getElementById('versionStep').style.display = 'block';
|
if (this.isBatchMode) {
|
||||||
|
document.getElementById('batchPreviewStep').style.display = 'block';
|
||||||
|
} else {
|
||||||
|
document.getElementById('versionStep').style.display = 'block';
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
closeModal() {
|
closeModal() {
|
||||||
@@ -727,34 +939,120 @@ export class DownloadManager {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Determine target folder and use_default_paths parameter
|
|
||||||
let targetFolder = '';
|
let targetFolder = '';
|
||||||
let useDefaultPaths = false;
|
let useDefaultPaths = false;
|
||||||
|
|
||||||
if (this.useDefaultPath) {
|
if (this.useDefaultPath) {
|
||||||
useDefaultPaths = true;
|
useDefaultPaths = true;
|
||||||
targetFolder = ''; // Not needed when using default paths
|
|
||||||
} else {
|
} else {
|
||||||
targetFolder = this.folderTreeManager.getSelectedPath();
|
targetFolder = this.folderTreeManager.getSelectedPath();
|
||||||
}
|
}
|
||||||
const fileParams = this.selectedFile ? {
|
if (!this.isBatchMode) {
|
||||||
type: 'Model',
|
const fileParams = this.selectedFile ? {
|
||||||
format: this.selectedFile.metadata?.format || 'SafeTensor',
|
type: 'Model',
|
||||||
size: this.selectedFile.metadata?.size || 'full',
|
format: this.selectedFile.metadata?.format || 'SafeTensor',
|
||||||
fp: this.selectedFile.metadata?.fp,
|
size: this.selectedFile.metadata?.size || 'full',
|
||||||
} : null;
|
fp: this.selectedFile.metadata?.fp,
|
||||||
|
} : null;
|
||||||
|
|
||||||
return this.executeDownloadWithProgress({
|
return this.executeDownloadWithProgress({
|
||||||
modelId: this.modelId,
|
modelId: this.modelId,
|
||||||
versionId: this.currentVersion.id,
|
versionId: this.currentVersion.id,
|
||||||
versionName: this.currentVersion.name,
|
versionName: this.currentVersion.name,
|
||||||
modelRoot,
|
modelRoot,
|
||||||
targetFolder,
|
targetFolder,
|
||||||
useDefaultPaths,
|
useDefaultPaths,
|
||||||
source: this.source,
|
source: this.source,
|
||||||
fileParams,
|
fileParams,
|
||||||
closeModal: true,
|
closeModal: true,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Batch download mode
|
||||||
|
const downloadItems = this.batchModels.filter(m => !m.error && m.selectedVersion && !m.selectedVersion.existsLocally);
|
||||||
|
if (downloadItems.length === 0) {
|
||||||
|
showToast('toast.loras.downloadCompleted', {}, 'info');
|
||||||
|
modalManager.closeModal('downloadModal');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
modalManager.closeModal('downloadModal');
|
||||||
|
|
||||||
|
const batchDownloadId = Date.now().toString();
|
||||||
|
const wsProtocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
|
||||||
|
const ws = new WebSocket(`${wsProtocol}${window.location.host}/ws/download-progress?id=${batchDownloadId}`);
|
||||||
|
|
||||||
|
const loadingManager = state.loadingManager || this.loadingManager;
|
||||||
|
const updateProgress = loadingManager.showDownloadProgress(downloadItems.length);
|
||||||
|
|
||||||
|
let completedDownloads = 0;
|
||||||
|
let failedDownloads = 0;
|
||||||
|
|
||||||
|
ws.onmessage = (event) => {
|
||||||
|
const data = JSON.parse(event.data);
|
||||||
|
if (data.type === 'download_id') return;
|
||||||
|
|
||||||
|
if (data.status === 'progress' && data.download_id?.startsWith(batchDownloadId)) {
|
||||||
|
const current = downloadItems[completedDownloads + failedDownloads];
|
||||||
|
const name = current?.selectedVersion?.name || `#${completedDownloads + failedDownloads + 1}`;
|
||||||
|
const metrics = {
|
||||||
|
bytesDownloaded: data.bytes_downloaded,
|
||||||
|
totalBytes: data.total_bytes,
|
||||||
|
bytesPerSecond: data.bytes_per_second,
|
||||||
|
};
|
||||||
|
updateProgress(data.progress, completedDownloads, name, metrics);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
await new Promise((resolve, reject) => {
|
||||||
|
ws.onopen = resolve;
|
||||||
|
ws.onerror = reject;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
for (let i = 0; i < downloadItems.length; i++) {
|
||||||
|
const item = downloadItems[i];
|
||||||
|
const ver = item.selectedVersion;
|
||||||
|
const name = ver?.name || `Model #${item.modelId}`;
|
||||||
|
|
||||||
|
updateProgress(0, completedDownloads, name);
|
||||||
|
loadingManager.setStatus(`${i + 1}/${downloadItems.length}: ${name}`);
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await this.apiClient.downloadModel(
|
||||||
|
item.modelId,
|
||||||
|
ver.id,
|
||||||
|
modelRoot,
|
||||||
|
targetFolder,
|
||||||
|
useDefaultPaths,
|
||||||
|
batchDownloadId,
|
||||||
|
item.source
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!response.success) {
|
||||||
|
failedDownloads++;
|
||||||
|
} else {
|
||||||
|
completedDownloads++;
|
||||||
|
updateProgress(100, completedDownloads, '');
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
console.error(`Failed to download ${name}:`, err);
|
||||||
|
failedDownloads++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ws.close();
|
||||||
|
loadingManager.hide();
|
||||||
|
|
||||||
|
if (failedDownloads === 0) {
|
||||||
|
showToast('toast.loras.allDownloadSuccessful', { count: completedDownloads }, 'success');
|
||||||
|
} else {
|
||||||
|
showToast('toast.loras.downloadPartialSuccess', {
|
||||||
|
completed: completedDownloads,
|
||||||
|
total: downloadItems.length,
|
||||||
|
}, 'warning');
|
||||||
|
}
|
||||||
|
|
||||||
|
await resetAndReload(true);
|
||||||
}
|
}
|
||||||
|
|
||||||
async downloadVersionWithDefaults(modelType, modelId, versionId, {
|
async downloadVersionWithDefaults(modelType, modelId, versionId, {
|
||||||
|
|||||||
@@ -10,7 +10,7 @@
|
|||||||
<div class="download-step" id="urlStep">
|
<div class="download-step" id="urlStep">
|
||||||
<div class="input-group">
|
<div class="input-group">
|
||||||
<label for="modelUrl" id="modelUrlLabel">{{ t('modals.download.url') }}:</label>
|
<label for="modelUrl" id="modelUrlLabel">{{ t('modals.download.url') }}:</label>
|
||||||
<input type="text" id="modelUrl" placeholder="{{ t('modals.download.placeholder') }}" />
|
<textarea id="modelUrl" rows="5" placeholder="{{ t('modals.download.placeholder') }}"></textarea>
|
||||||
<div class="error-message" id="urlError"></div>
|
<div class="error-message" id="urlError"></div>
|
||||||
</div>
|
</div>
|
||||||
<div class="modal-actions">
|
<div class="modal-actions">
|
||||||
@@ -18,7 +18,18 @@
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- Step 2: Version Selection -->
|
<!-- Step 2: Batch Preview (multi-URL mode) -->
|
||||||
|
<div class="download-step" id="batchPreviewStep" style="display: none;">
|
||||||
|
<div class="batch-preview-list" id="batchPreviewList">
|
||||||
|
<!-- Batch items will be inserted here dynamically -->
|
||||||
|
</div>
|
||||||
|
<div class="modal-actions">
|
||||||
|
<button class="secondary-btn" id="backToUrlFromBatchBtn">{{ t('common.actions.back') }}</button>
|
||||||
|
<button class="primary-btn" id="nextFromBatchBtn">{{ t('common.actions.next') }}</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- Step 3: Version Selection (single-URL or per-item editing) -->
|
||||||
<div class="download-step" id="versionStep" style="display: none;">
|
<div class="download-step" id="versionStep" style="display: none;">
|
||||||
<div class="version-list" id="versionList">
|
<div class="version-list" id="versionList">
|
||||||
<!-- Versions will be inserted here dynamically -->
|
<!-- Versions will be inserted here dynamically -->
|
||||||
|
|||||||
@@ -5,7 +5,8 @@
|
|||||||
ref="textareaRef"
|
ref="textareaRef"
|
||||||
:placeholder="placeholder"
|
:placeholder="placeholder"
|
||||||
:spellcheck="spellcheck ?? false"
|
:spellcheck="spellcheck ?? false"
|
||||||
:class="['text-input', { 'vue-dom-mode': isVueDomMode }]"
|
:class="['text-input', { 'vue-dom-mode': isVueDomMode, 'lm-wheel-scrollable': isVueDomMode }]"
|
||||||
|
:style="maxHeight && isVueDomMode ? { maxHeight: maxHeight + 'px' } : undefined"
|
||||||
@input="onInput"
|
@input="onInput"
|
||||||
@wheel="onWheel"
|
@wheel="onWheel"
|
||||||
/>
|
/>
|
||||||
@@ -47,6 +48,7 @@ const props = defineProps<{
|
|||||||
placeholder?: string
|
placeholder?: string
|
||||||
showPreview?: boolean
|
showPreview?: boolean
|
||||||
spellcheck?: boolean
|
spellcheck?: boolean
|
||||||
|
maxHeight?: number
|
||||||
}>()
|
}>()
|
||||||
|
|
||||||
// Reactive ref for Vue DOM mode
|
// Reactive ref for Vue DOM mode
|
||||||
|
|||||||
@@ -655,13 +655,16 @@ function createAutocompleteTextWidgetFactory(
|
|||||||
// Get spellcheck setting from ComfyUI settings (default: false)
|
// Get spellcheck setting from ComfyUI settings (default: false)
|
||||||
const spellcheck = app.ui?.settings?.getSettingValue?.('Comfy.TextareaWidget.Spellcheck') ?? false
|
const spellcheck = app.ui?.settings?.getSettingValue?.('Comfy.TextareaWidget.Spellcheck') ?? false
|
||||||
|
|
||||||
|
const maxHeight = modelType === 'loras' ? AUTOCOMPLETE_TEXT_WIDGET_MAX_HEIGHT : undefined
|
||||||
|
|
||||||
const vueApp = createApp(AutocompleteTextWidget, {
|
const vueApp = createApp(AutocompleteTextWidget, {
|
||||||
widget,
|
widget,
|
||||||
node,
|
node,
|
||||||
modelType,
|
modelType,
|
||||||
placeholder: inputOptions.placeholder || widgetName,
|
placeholder: inputOptions.placeholder || widgetName,
|
||||||
showPreview: true,
|
showPreview: true,
|
||||||
spellcheck
|
spellcheck,
|
||||||
|
maxHeight
|
||||||
})
|
})
|
||||||
|
|
||||||
vueApp.use(PrimeVue, {
|
vueApp.use(PrimeVue, {
|
||||||
@@ -673,6 +676,10 @@ function createAutocompleteTextWidgetFactory(
|
|||||||
const appKey = instanceId
|
const appKey = instanceId
|
||||||
vueApps.set(appKey, vueApp)
|
vueApps.set(appKey, vueApp)
|
||||||
|
|
||||||
|
if (maxHeight) {
|
||||||
|
container.style.maxHeight = `${maxHeight}px`
|
||||||
|
}
|
||||||
|
|
||||||
widget.onRemove = createVueWidgetCleanup(vueApp, () => {
|
widget.onRemove = createVueWidgetCleanup(vueApp, () => {
|
||||||
vueApps.delete(appKey)
|
vueApps.delete(appKey)
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -5,13 +5,13 @@ import {
|
|||||||
updateWidgetHeight,
|
updateWidgetHeight,
|
||||||
shouldShowClipEntry,
|
shouldShowClipEntry,
|
||||||
syncClipStrengthIfCollapsed,
|
syncClipStrengthIfCollapsed,
|
||||||
LORA_ENTRY_HEIGHT,
|
LORA_ENTRY_HEIGHT,
|
||||||
HEADER_HEIGHT,
|
HEADER_HEIGHT,
|
||||||
CONTAINER_PADDING,
|
CONTAINER_PADDING,
|
||||||
EMPTY_CONTAINER_HEIGHT
|
EMPTY_CONTAINER_HEIGHT
|
||||||
} from "./loras_widget_utils.js";
|
} from "./loras_widget_utils.js";
|
||||||
import { initDrag, createContextMenu, initHeaderDrag, initReorderDrag, handleKeyboardNavigation } from "./loras_widget_events.js";
|
import { initDrag, createContextMenu, initHeaderDrag, initReorderDrag, handleKeyboardNavigation } from "./loras_widget_events.js";
|
||||||
import { forwardMiddleMouseToCanvas, forwardWheelToCanvas } from "./utils.js";
|
import { forwardMiddleMouseToCanvas, forwardWheelToCanvas, enableListWheelScroll } from "./utils.js";
|
||||||
import { PreviewTooltip } from "./preview_tooltip.js";
|
import { PreviewTooltip } from "./preview_tooltip.js";
|
||||||
import { ensureLmStyles } from "./lm_styles_loader.js";
|
import { ensureLmStyles } from "./lm_styles_loader.js";
|
||||||
import { getStrengthStepPreference } from "./settings.js";
|
import { getStrengthStepPreference } from "./settings.js";
|
||||||
@@ -24,11 +24,18 @@ export function addLorasWidget(node, name, opts, callback) {
|
|||||||
container.className = "lm-loras-container";
|
container.className = "lm-loras-container";
|
||||||
|
|
||||||
forwardMiddleMouseToCanvas(container);
|
forwardMiddleMouseToCanvas(container);
|
||||||
|
// Capture-phase handler: scroll the list with the wheel when it overflows.
|
||||||
|
// Falls through to forwardWheelToCanvas (canvas zoom) when the list is short.
|
||||||
|
enableListWheelScroll(container);
|
||||||
forwardWheelToCanvas(container);
|
forwardWheelToCanvas(container);
|
||||||
|
|
||||||
// Set initial height using CSS variables approach
|
// Set initial height using CSS variables approach
|
||||||
const defaultHeight = 200;
|
const defaultHeight = 200;
|
||||||
|
|
||||||
|
// Content height (capped at 12 rows by renderLoras), kept up to date and used
|
||||||
|
// to fix the widget/node height in both Canvas and Nodes 2.0 (Vue) modes.
|
||||||
|
let currentContentHeight = defaultHeight;
|
||||||
|
|
||||||
// Check if this is a randomizer node (lock button instead of drag handle)
|
// Check if this is a randomizer node (lock button instead of drag handle)
|
||||||
const isRandomizerNode = opts?.isRandomizerNode === true;
|
const isRandomizerNode = opts?.isRandomizerNode === true;
|
||||||
|
|
||||||
@@ -198,7 +205,7 @@ export function addLorasWidget(node, name, opts, callback) {
|
|||||||
container.appendChild(emptyMessage);
|
container.appendChild(emptyMessage);
|
||||||
|
|
||||||
// Set fixed height for empty state
|
// Set fixed height for empty state
|
||||||
updateWidgetHeight(container, EMPTY_CONTAINER_HEIGHT, defaultHeight, node);
|
currentContentHeight = updateWidgetHeight(container, EMPTY_CONTAINER_HEIGHT, defaultHeight, node);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -645,7 +652,7 @@ export function addLorasWidget(node, name, opts, callback) {
|
|||||||
|
|
||||||
// Calculate height based on number of loras and fixed sizes
|
// Calculate height based on number of loras and fixed sizes
|
||||||
const calculatedHeight = CONTAINER_PADDING + HEADER_HEIGHT + (Math.min(totalVisibleEntries, 12) * LORA_ENTRY_HEIGHT);
|
const calculatedHeight = CONTAINER_PADDING + HEADER_HEIGHT + (Math.min(totalVisibleEntries, 12) * LORA_ENTRY_HEIGHT);
|
||||||
updateWidgetHeight(container, calculatedHeight, defaultHeight, node);
|
currentContentHeight = updateWidgetHeight(container, calculatedHeight, defaultHeight, node);
|
||||||
|
|
||||||
// After all LoRA elements are created, apply selection state as the last step
|
// After all LoRA elements are created, apply selection state as the last step
|
||||||
// This ensures the selection state is not overwritten
|
// This ensures the selection state is not overwritten
|
||||||
@@ -727,12 +734,31 @@ export function addLorasWidget(node, name, opts, callback) {
|
|||||||
widgetValue = updatedValue;
|
widgetValue = updatedValue;
|
||||||
renderLoras(widgetValue, widget);
|
renderLoras(widgetValue, widget);
|
||||||
},
|
},
|
||||||
|
// The list area is capped at 12 rows (see calculatedHeight); beyond that the
|
||||||
|
// container scrolls. Report that capped height as both the min and preferred
|
||||||
|
// size so the node height stays fixed to the list, matching Canvas mode.
|
||||||
|
getMinHeight: () => currentContentHeight,
|
||||||
|
getHeight: () => currentContentHeight,
|
||||||
hideOnZoom: true,
|
hideOnZoom: true,
|
||||||
selectOn: ['click', 'focus']
|
selectOn: ['click', 'focus']
|
||||||
});
|
});
|
||||||
|
|
||||||
widget.value = defaultValue;
|
widget.value = defaultValue;
|
||||||
|
|
||||||
|
// Canonical LiteGraph sizing hook (Canvas mode): fix the widget to the capped
|
||||||
|
// content height. Rows beyond the 12-row cap scroll inside the container.
|
||||||
|
widget.computeSize = (width) => [width, currentContentHeight];
|
||||||
|
|
||||||
|
// Nodes 2.0 / Vue mode reads computeLayoutSize for the node's size. Pin both
|
||||||
|
// the min and max to the capped content height so the list area is fixed to
|
||||||
|
// 12 rows (scrolling beyond), matching Canvas mode, instead of the layout
|
||||||
|
// engine measuring the full DOM and locking the node fully expanded.
|
||||||
|
widget.computeLayoutSize = () => ({
|
||||||
|
minHeight: currentContentHeight,
|
||||||
|
maxHeight: currentContentHeight,
|
||||||
|
minWidth: 400,
|
||||||
|
});
|
||||||
|
|
||||||
widget.callback = callback;
|
widget.callback = callback;
|
||||||
|
|
||||||
widget.onRemove = () => {
|
widget.onRemove = () => {
|
||||||
|
|||||||
@@ -18,21 +18,28 @@ export function formatLoraValue(loras) {
|
|||||||
return loras;
|
return loras;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Function to update widget height consistently
|
// Resolve the capped (12-row) height of the widget and physically cap the
|
||||||
|
// container so the list area never grows past it.
|
||||||
|
// `height` is the raw content height (already capped at 12 rows by the caller);
|
||||||
|
// the result never drops below defaultHeight.
|
||||||
|
// The max-height is the reliable lever: Nodes 2.0 / Vue mode measures the
|
||||||
|
// rendered DOM to size the node, so without an actual height cap on the element
|
||||||
|
// the list always shows every row. max-height bounds the element regardless of
|
||||||
|
// what the layout engine measures, and the overflow makes the extra rows scroll.
|
||||||
|
// Returns the resolved height so callers can also report it to ComfyUI.
|
||||||
export function updateWidgetHeight(container, height, defaultHeight, node) {
|
export function updateWidgetHeight(container, height, defaultHeight, node) {
|
||||||
// Ensure minimum height
|
const cappedHeight = Math.max(defaultHeight, height);
|
||||||
const finalHeight = Math.max(defaultHeight, height);
|
|
||||||
|
container.style.maxHeight = `${cappedHeight}px`;
|
||||||
// Update CSS variables
|
|
||||||
container.style.setProperty('--comfy-widget-min-height', `${finalHeight}px`);
|
// Force node to redraw after a short delay to ensure the DOM is updated.
|
||||||
container.style.setProperty('--comfy-widget-height', `${finalHeight}px`);
|
|
||||||
|
|
||||||
// Force node to update size after a short delay to ensure DOM is updated
|
|
||||||
if (node) {
|
if (node) {
|
||||||
setTimeout(() => {
|
setTimeout(() => {
|
||||||
node.setDirtyCanvas(true, true);
|
node.setDirtyCanvas(true, true);
|
||||||
}, 10);
|
}, 10);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return cappedHeight;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Determine if clip entry should be shown - now based on expanded property or initial diff values
|
// Determine if clip entry should be shown - now based on expanded property or initial diff values
|
||||||
|
|||||||
@@ -784,6 +784,59 @@ export function forwardWheelToCanvas(container, options = {}) {
|
|||||||
}, { passive: false });
|
}, { passive: false });
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Marks elements whose wheel scrolling must win over the canvas zoom.
|
||||||
|
const LIST_WHEEL_SCROLL_CLASS = 'lm-wheel-scrollable';
|
||||||
|
let listWheelScrollHookInstalled = false;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Keep vertical wheel scrolling inside a scrollable widget container instead of
|
||||||
|
* letting ComfyUI zoom the canvas.
|
||||||
|
*
|
||||||
|
* In Nodes 2.0 / Vue mode ComfyUI's wheel→zoom handler runs on the document /
|
||||||
|
* canvas in the capture phase, which is *outer* than the widget, so a listener
|
||||||
|
* on the container (even in capture) fires too late. The reliable place to win
|
||||||
|
* is a single hook on `window` in the capture phase — the very first step of
|
||||||
|
* event dispatch. When the wheel is over a marked, scrollable element we scroll
|
||||||
|
* it manually and fully consume the event; otherwise we leave it alone so canvas
|
||||||
|
* zoom keeps working.
|
||||||
|
* @param {HTMLElement} container - The scrollable element (overflow: auto)
|
||||||
|
*/
|
||||||
|
export function enableListWheelScroll(container) {
|
||||||
|
if (!container) return;
|
||||||
|
container.classList.add(LIST_WHEEL_SCROLL_CLASS);
|
||||||
|
|
||||||
|
if (listWheelScrollHookInstalled) return;
|
||||||
|
listWheelScrollHookInstalled = true;
|
||||||
|
|
||||||
|
window.addEventListener('wheel', (event) => {
|
||||||
|
// Let pinch/zoom and horizontal gestures fall through to the canvas.
|
||||||
|
if (event.ctrlKey) return;
|
||||||
|
if (Math.abs(event.deltaX) > Math.abs(event.deltaY)) return;
|
||||||
|
|
||||||
|
const target = event.target;
|
||||||
|
if (!target || typeof target.closest !== 'function') return;
|
||||||
|
const el = target.closest(`.${LIST_WHEEL_SCROLL_CLASS}`);
|
||||||
|
if (!el) return;
|
||||||
|
|
||||||
|
const canScrollY = el.scrollHeight > el.clientHeight + 1;
|
||||||
|
if (!canScrollY) return; // Nothing to scroll → allow canvas zoom.
|
||||||
|
|
||||||
|
// Translate the wheel delta to pixels (line / page modes → approx px).
|
||||||
|
const unit = event.deltaMode === 1
|
||||||
|
? 16
|
||||||
|
: event.deltaMode === 2
|
||||||
|
? el.clientHeight
|
||||||
|
: 1;
|
||||||
|
|
||||||
|
el.scrollTop += event.deltaY * unit;
|
||||||
|
|
||||||
|
// Consume the event so neither ComfyUI's zoom nor forwardWheelToCanvas react.
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
event.stopImmediatePropagation();
|
||||||
|
}, { capture: true, passive: false });
|
||||||
|
}
|
||||||
|
|
||||||
// Get connected Lora Pool node from pool_config input
|
// Get connected Lora Pool node from pool_config input
|
||||||
export function getConnectedPoolConfigNode(node) {
|
export function getConnectedPoolConfigNode(node) {
|
||||||
if (!node?.inputs) {
|
if (!node?.inputs) {
|
||||||
|
|||||||
@@ -2118,14 +2118,14 @@ to { transform: rotate(360deg);
|
|||||||
padding: 20px 0;
|
padding: 20px 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
.autocomplete-text-widget[data-v-5514bf46] {
|
.autocomplete-text-widget[data-v-1c610e5d] {
|
||||||
background: transparent;
|
background: transparent;
|
||||||
height: 100%;
|
height: 100%;
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
box-sizing: border-box;
|
box-sizing: border-box;
|
||||||
}
|
}
|
||||||
.input-wrapper[data-v-5514bf46] {
|
.input-wrapper[data-v-1c610e5d] {
|
||||||
position: relative;
|
position: relative;
|
||||||
flex: 1;
|
flex: 1;
|
||||||
display: flex;
|
display: flex;
|
||||||
@@ -2133,7 +2133,7 @@ to { transform: rotate(360deg);
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Canvas mode styles (default) - matches built-in comfy-multiline-input */
|
/* Canvas mode styles (default) - matches built-in comfy-multiline-input */
|
||||||
.text-input[data-v-5514bf46] {
|
.text-input[data-v-1c610e5d] {
|
||||||
flex: 1;
|
flex: 1;
|
||||||
width: 100%;
|
width: 100%;
|
||||||
background-color: var(--comfy-input-bg, #222);
|
background-color: var(--comfy-input-bg, #222);
|
||||||
@@ -2150,7 +2150,7 @@ to { transform: rotate(360deg);
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Vue DOM mode styles - matches built-in p-textarea in Vue DOM mode */
|
/* Vue DOM mode styles - matches built-in p-textarea in Vue DOM mode */
|
||||||
.text-input.vue-dom-mode[data-v-5514bf46] {
|
.text-input.vue-dom-mode[data-v-1c610e5d] {
|
||||||
background-color: var(--color-charcoal-400, #313235);
|
background-color: var(--color-charcoal-400, #313235);
|
||||||
color: #fff;
|
color: #fff;
|
||||||
padding: 8px 12px 30px 12px; /* Reserve bottom space for clear button */
|
padding: 8px 12px 30px 12px; /* Reserve bottom space for clear button */
|
||||||
@@ -2159,12 +2159,12 @@ to { transform: rotate(360deg);
|
|||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
font-family: inherit;
|
font-family: inherit;
|
||||||
}
|
}
|
||||||
.text-input[data-v-5514bf46]:focus {
|
.text-input[data-v-1c610e5d]:focus {
|
||||||
outline: none;
|
outline: none;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Clear button styles */
|
/* Clear button styles */
|
||||||
.clear-button[data-v-5514bf46] {
|
.clear-button[data-v-1c610e5d] {
|
||||||
position: absolute;
|
position: absolute;
|
||||||
right: 6px;
|
right: 6px;
|
||||||
bottom: 6px; /* Changed from top to bottom */
|
bottom: 6px; /* Changed from top to bottom */
|
||||||
@@ -2187,31 +2187,31 @@ to { transform: rotate(360deg);
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Show clear button when hovering over input wrapper */
|
/* Show clear button when hovering over input wrapper */
|
||||||
.input-wrapper:hover .clear-button[data-v-5514bf46] {
|
.input-wrapper:hover .clear-button[data-v-1c610e5d] {
|
||||||
opacity: 0.7;
|
opacity: 0.7;
|
||||||
pointer-events: auto;
|
pointer-events: auto;
|
||||||
}
|
}
|
||||||
.clear-button[data-v-5514bf46]:hover {
|
.clear-button[data-v-1c610e5d]:hover {
|
||||||
opacity: 1;
|
opacity: 1;
|
||||||
background: rgba(255, 100, 100, 0.8);
|
background: rgba(255, 100, 100, 0.8);
|
||||||
}
|
}
|
||||||
.clear-button svg[data-v-5514bf46] {
|
.clear-button svg[data-v-1c610e5d] {
|
||||||
width: 12px;
|
width: 12px;
|
||||||
height: 12px;
|
height: 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Vue DOM mode adjustments for clear button */
|
/* Vue DOM mode adjustments for clear button */
|
||||||
.text-input.vue-dom-mode ~ .clear-button[data-v-5514bf46] {
|
.text-input.vue-dom-mode ~ .clear-button[data-v-1c610e5d] {
|
||||||
right: 8px;
|
right: 8px;
|
||||||
bottom: 10px; /* Changed from top to bottom, adjusted for Vue DOM padding */
|
bottom: 10px; /* Changed from top to bottom, adjusted for Vue DOM padding */
|
||||||
width: 20px;
|
width: 20px;
|
||||||
height: 20px;
|
height: 20px;
|
||||||
background: rgba(107, 114, 128, 0.6);
|
background: rgba(107, 114, 128, 0.6);
|
||||||
}
|
}
|
||||||
.text-input.vue-dom-mode ~ .clear-button[data-v-5514bf46]:hover {
|
.text-input.vue-dom-mode ~ .clear-button[data-v-1c610e5d]:hover {
|
||||||
background: oklch(62% 0.18 25);
|
background: oklch(62% 0.18 25);
|
||||||
}
|
}
|
||||||
.text-input.vue-dom-mode ~ .clear-button svg[data-v-5514bf46] {
|
.text-input.vue-dom-mode ~ .clear-button svg[data-v-1c610e5d] {
|
||||||
width: 14px;
|
width: 14px;
|
||||||
height: 14px;
|
height: 14px;
|
||||||
}`));
|
}`));
|
||||||
@@ -14783,7 +14783,8 @@ const _sfc_main = /* @__PURE__ */ defineComponent({
|
|||||||
modelType: {},
|
modelType: {},
|
||||||
placeholder: {},
|
placeholder: {},
|
||||||
showPreview: { type: Boolean },
|
showPreview: { type: Boolean },
|
||||||
spellcheck: { type: Boolean }
|
spellcheck: { type: Boolean },
|
||||||
|
maxHeight: {}
|
||||||
},
|
},
|
||||||
setup(__props) {
|
setup(__props) {
|
||||||
const props = __props;
|
const props = __props;
|
||||||
@@ -14913,10 +14914,11 @@ const _sfc_main = /* @__PURE__ */ defineComponent({
|
|||||||
ref: textareaRef,
|
ref: textareaRef,
|
||||||
placeholder: __props.placeholder,
|
placeholder: __props.placeholder,
|
||||||
spellcheck: __props.spellcheck ?? false,
|
spellcheck: __props.spellcheck ?? false,
|
||||||
class: normalizeClass(["text-input", { "vue-dom-mode": isVueDomMode.value }]),
|
class: normalizeClass(["text-input", { "vue-dom-mode": isVueDomMode.value, "lm-wheel-scrollable": isVueDomMode.value }]),
|
||||||
|
style: normalizeStyle(__props.maxHeight && isVueDomMode.value ? { maxHeight: __props.maxHeight + "px" } : void 0),
|
||||||
onInput,
|
onInput,
|
||||||
onWheel
|
onWheel
|
||||||
}, null, 42, _hoisted_3),
|
}, null, 46, _hoisted_3),
|
||||||
showClearButton.value ? (openBlock(), createElementBlock("button", {
|
showClearButton.value ? (openBlock(), createElementBlock("button", {
|
||||||
key: 0,
|
key: 0,
|
||||||
type: "button",
|
type: "button",
|
||||||
@@ -14949,7 +14951,7 @@ const _sfc_main = /* @__PURE__ */ defineComponent({
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
const AutocompleteTextWidget = /* @__PURE__ */ _export_sfc(_sfc_main, [["__scopeId", "data-v-5514bf46"]]);
|
const AutocompleteTextWidget = /* @__PURE__ */ _export_sfc(_sfc_main, [["__scopeId", "data-v-1c610e5d"]]);
|
||||||
function createVueWidgetCleanup(vueApp, onCleanup) {
|
function createVueWidgetCleanup(vueApp, onCleanup) {
|
||||||
let didUnmount = false;
|
let didUnmount = false;
|
||||||
return () => {
|
return () => {
|
||||||
@@ -15799,13 +15801,15 @@ function createAutocompleteTextWidgetFactory(node, widgetName, modelType, inputO
|
|||||||
);
|
);
|
||||||
widget.metadataWidget = metadataWidget;
|
widget.metadataWidget = metadataWidget;
|
||||||
const spellcheck = ((_c = (_b = (_a2 = app$1.ui) == null ? void 0 : _a2.settings) == null ? void 0 : _b.getSettingValue) == null ? void 0 : _c.call(_b, "Comfy.TextareaWidget.Spellcheck")) ?? false;
|
const spellcheck = ((_c = (_b = (_a2 = app$1.ui) == null ? void 0 : _a2.settings) == null ? void 0 : _b.getSettingValue) == null ? void 0 : _c.call(_b, "Comfy.TextareaWidget.Spellcheck")) ?? false;
|
||||||
|
const maxHeight = modelType === "loras" ? AUTOCOMPLETE_TEXT_WIDGET_MAX_HEIGHT : void 0;
|
||||||
const vueApp = createApp(AutocompleteTextWidget, {
|
const vueApp = createApp(AutocompleteTextWidget, {
|
||||||
widget,
|
widget,
|
||||||
node,
|
node,
|
||||||
modelType,
|
modelType,
|
||||||
placeholder: inputOptions.placeholder || widgetName,
|
placeholder: inputOptions.placeholder || widgetName,
|
||||||
showPreview: true,
|
showPreview: true,
|
||||||
spellcheck
|
spellcheck,
|
||||||
|
maxHeight
|
||||||
});
|
});
|
||||||
vueApp.use(PrimeVue, {
|
vueApp.use(PrimeVue, {
|
||||||
unstyled: true,
|
unstyled: true,
|
||||||
@@ -15814,6 +15818,9 @@ function createAutocompleteTextWidgetFactory(node, widgetName, modelType, inputO
|
|||||||
vueApp.mount(container);
|
vueApp.mount(container);
|
||||||
const appKey = instanceId;
|
const appKey = instanceId;
|
||||||
vueApps.set(appKey, vueApp);
|
vueApps.set(appKey, vueApp);
|
||||||
|
if (maxHeight) {
|
||||||
|
container.style.maxHeight = `${maxHeight}px`;
|
||||||
|
}
|
||||||
widget.onRemove = createVueWidgetCleanup(vueApp, () => {
|
widget.onRemove = createVueWidgetCleanup(vueApp, () => {
|
||||||
vueApps.delete(appKey);
|
vueApps.delete(appKey);
|
||||||
});
|
});
|
||||||
|
|||||||
File diff suppressed because one or more lines are too long
Reference in New Issue
Block a user