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

Enhance migration and download processes: add backend path update and prevent duplicate completion toasts

Browse Source
This commit is contained in:
Will Miao
2025-06-07 11:29:53 +08:00
parent b1015ecdc5
commit c96550cc68
2 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
py/routes/misc_routes.py
View File

@@ -1158,8 +1158,6 @@ class MiscRoutes:
"""
matching_files = []
model_name = os.path.splitext(model_filename)[0]
if model_name == "FluxMechaKnights":
logger.info(f"Processing model: {model_name}")
# Check if pattern contains a directory separator
has_subdirs = '/' in regex_pattern.pattern or '\\\\' in regex_pattern.pattern

26
static/js/managers/ExampleImagesManager.js
View File

@@ -12,6 +12,7 @@ class ExampleImagesManager {
this.isProgressPanelCollapsed = false;
this.pauseButton = null; // Store reference to the pause button
this.isMigrating = false; // Track migration state separately from downloading
this.hasShownCompletionToast = false; // Flag to track if completion toast has been shown
// Initialize download path field and check download status
this.initializePathOptions();
@@ -173,6 +174,25 @@ class ExampleImagesManager {
return;
}
// Update path in backend settings before starting migration
try {
const pathUpdateResponse = await fetch('/api/settings', {
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({
example_images_path: outputDir
})
});
if (!pathUpdateResponse.ok) {
throw new Error(`HTTP error! Status: ${pathUpdateResponse.status}`);
}
} catch (error) {
console.error('Failed to update example images path:', error);
}
const pattern = document.getElementById('exampleImagesMigratePattern').value || '{model}.example.{index}.{ext}';
const optimize = document.getElementById('optimizeExampleImages').checked;
@@ -195,6 +215,7 @@ class ExampleImagesManager {
this.isDownloading = true;
this.isMigrating = true;
this.isPaused = false;
this.hasShownCompletionToast = false; // Reset toast flag when starting new migration
this.startTime = new Date();
this.updateUI(data.status);
this.showProgressPanel();
@@ -300,6 +321,7 @@ class ExampleImagesManager {
if (data.success) {
this.isDownloading = true;
this.isPaused = false;
this.hasShownCompletionToast = false; // Reset toast flag when starting new download
this.startTime = new Date();
this.updateUI(data.status);
this.showProgressPanel();
@@ -422,9 +444,11 @@ class ExampleImagesManager {
clearInterval(this.progressUpdateInterval);
this.progressUpdateInterval = null;
if (data.status.status === 'completed') {
if (data.status.status === 'completed' && !this.hasShownCompletionToast) {
const actionType = this.isMigrating ? 'migration' : 'download';
showToast(`Example images ${actionType} completed`, 'success');
// Mark as shown to prevent duplicate toasts
this.hasShownCompletionToast = true;
// Reset migration flag
this.isMigrating = false;
// Hide the panel after a delay