alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

Update unauthorized access error message

Browse Source
This commit is contained in:
Will Miao
2025-03-30 20:00:23 +08:00
parent f04af2de21
commit 3f97087abb
2 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
py/services/civitai_client.py
View File

@@ -80,12 +80,7 @@ class CivitaiClient:
if response.status == 401:
logger.warning(f"Unauthorized access to resource: {url} (Status 401)")
# Check if this is an API key issue (has Set-Cookie headers)
if 'Set-Cookie' in response.headers:
return False, "Invalid or missing CivitAI API key. Please check your API key in settings."
# Otherwise it's an early access restriction
return False, "Early access restriction: You must purchase early access to download this LoRA."
return False, "Invalid or missing CivitAI API key, or early access restriction."
# Handle other client errors that might be permission-related
if response.status == 403:

22
static/js/managers/ImportManager.js
View File

@@ -893,7 +893,7 @@ export class ImportManager {
});
const result = await response.json();
if (!result.success) {
// Handle save error
console.error("Failed to save recipe:", result.error);
@@ -905,6 +905,7 @@ export class ImportManager {
}
// Check if we need to download LoRAs
let failedDownloads = 0;
if (this.downloadableLoRAs && this.downloadableLoRAs.length > 0) {
// For download, we need to validate the target path
const loraRoot = document.getElementById('importLoraRoot')?.value;
@@ -931,8 +932,7 @@ export class ImportManager {
const updateProgress = this.loadingManager.showDownloadProgress(this.downloadableLoRAs.length);
let completedDownloads = 0;
let failedDownloads = 0;
let earlyAccessFailures = 0;
let accessFailures = 0;
let currentLoraProgress = 0;
// Set up progress tracking for current download
@@ -999,12 +999,10 @@ export class ImportManager {
console.error(`Failed to download LoRA ${lora.name}: ${errorText}`);
// Check if this is an early access error (status 401 is the key indicator)
if (response.status === 401 ||
(errorText.toLowerCase().includes('early access') ||
errorText.toLowerCase().includes('purchase'))) {
earlyAccessFailures++;
if (response.status === 401) {
accessFailures++;
this.loadingManager.setStatus(
`Failed to download ${lora.name}: Early Access required`
`Failed to download ${lora.name}: Access restricted`
);
}
@@ -1036,9 +1034,9 @@ export class ImportManager {
if (failedDownloads === 0) {
showToast(`All ${completedDownloads} LoRAs downloaded successfully`, 'success');
} else {
if (earlyAccessFailures > 0) {
if (accessFailures > 0) {
showToast(
`Downloaded ${completedDownloads} of ${this.downloadableLoRAs.length} LoRAs. ${earlyAccessFailures} failed due to Early Access restrictions.`,
`Downloaded ${completedDownloads} of ${this.downloadableLoRAs.length} LoRAs. ${accessFailures} failed due to access restrictions. Check your API key in settings or early access status.`,
'error'
);
} else {
@@ -1049,7 +1047,9 @@ export class ImportManager {
// Show success message
if (isDownloadOnly) {
showToast('LoRAs downloaded successfully', 'success');
if (failedDownloads === 0) {
showToast('LoRAs downloaded successfully', 'success');
}
} else {
showToast(`Recipe "${this.recipeName}" saved successfully`, 'success');
}