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

Refactor progress reporting in DownloadManager and improve selection logic in MoveManager

Browse Source
This commit is contained in:
Will Miao
2025-03-03 18:51:35 +08:00
parent 81b401106f
commit a38757fee6
2 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
py/services/download_manager.py
View File

@@ -80,7 +80,7 @@ class DownloadManager:
if images: if images:
# Report preview download progress # Report preview download progress
if progress_callback: if progress_callback:
await progress_callback(5) # 5% progress for starting preview download await progress_callback(1) # 1% progress for starting preview download
preview_ext = '.mp4' if images[0].get('type') == 'video' else '.png' preview_ext = '.mp4' if images[0].get('type') == 'video' else '.png'
preview_path = os.path.splitext(save_path)[0] + '.preview' + preview_ext preview_path = os.path.splitext(save_path)[0] + '.preview' + preview_ext
@@ -91,7 +91,7 @@ class DownloadManager:
# Report preview download completion # Report preview download completion
if progress_callback: if progress_callback:
await progress_callback(10) # 10% progress after preview download await progress_callback(3) # 3% progress after preview download
# Download model file with progress tracking # Download model file with progress tracking
success, result = await self.civitai_client._download_file( success, result = await self.civitai_client._download_file(
@@ -149,6 +149,6 @@ class DownloadManager:
progress_callback: Callback function for progress updates progress_callback: Callback function for progress updates
""" """
if progress_callback: if progress_callback:
# Scale file progress to 10-100 range (after preview download) # Scale file progress to 3-100 range (after preview download)
overall_progress = 10 + (file_progress * 0.9) # 90% of progress for file download overall_progress = 3 + (file_progress * 0.97) # 97% of progress for file download
await progress_callback(round(overall_progress)) await progress_callback(round(overall_progress))

7
static/js/managers/MoveManager.js
View File

@@ -23,13 +23,18 @@ class MoveManager {
const folderItem = e.target.closest('.folder-item'); const folderItem = e.target.closest('.folder-item');
if (!folderItem) return; if (!folderItem) return;
// 如果点击已选中的文件夹，则取消选择
if (folderItem.classList.contains('selected')) {
folderItem.classList.remove('selected');
} else {
// 取消其他选中状态 // 取消其他选中状态
this.folderBrowser.querySelectorAll('.folder-item').forEach(item => { this.folderBrowser.querySelectorAll('.folder-item').forEach(item => {
item.classList.remove('selected'); item.classList.remove('selected');
}); });
// 设置当前选中状态 // 设置当前选中状态
folderItem.classList.add('selected'); folderItem.classList.add('selected');
}
this.updatePathPreview(); this.updatePathPreview();
}); });