From dd961386c7922d94a00a7817cdc259bcae7eef0e Mon Sep 17 00:00:00 2001 From: huchukato Date: Sat, 21 Feb 2026 03:07:14 +0100 Subject: [PATCH] Fix placeholder paths and model name parsing 1. Changed placeholder paths to relative './' instead of absolute 2. Enhanced model name parsing to use model_info['name'] when available 3. Fallback to model_id parsing for better display names 4. Should show 'wan2.2-Remix' instead of 'Unknown Model' when possible --- server/routes/DownloadModel.py | 7 +++++-- web/js/huggingfaceDownloader.js | 2 +- web/js/ui/previewRenderer.js | 2 +- web/js/ui/searchRenderer.js | 2 +- web/js/ui/statusRenderer.js | 2 +- 5 files changed, 9 insertions(+), 6 deletions(-) diff --git a/server/routes/DownloadModel.py b/server/routes/DownloadModel.py index 7b81640..a1a484b 100644 --- a/server/routes/DownloadModel.py +++ b/server/routes/DownloadModel.py @@ -129,8 +129,11 @@ async def route_download_model(request): download_id = download_manager.add_to_queue(download_info) - # Extract model name from model_id for better display - model_display_name = target_model_id.split('/')[-1] if target_model_id else "Unknown Model" + # Extract model name from model_info if available, otherwise from model_id + if model_info and isinstance(model_info, dict) and model_info.get('name'): + model_display_name = model_info['name'] + else: + model_display_name = target_model_id.split('/')[-1] if target_model_id else "Unknown Model" response_data = { "download_id": download_id, diff --git a/web/js/huggingfaceDownloader.js b/web/js/huggingfaceDownloader.js index 4717120..7ca38ab 100644 --- a/web/js/huggingfaceDownloader.js +++ b/web/js/huggingfaceDownloader.js @@ -7,7 +7,7 @@ console.log("Loading HuggingFace UI..."); // --- Configuration --- const EXTENSION_NAME = "HuggingFace"; const CSS_URL = `../huggingfaceDownloader.css`; -const PLACEHOLDER_IMAGE_URL = `/custom_nodes/ComfyUI-HuggingFace/images/placeholder.jpg`; +const PLACEHOLDER_IMAGE_URL = `./images/placeholder.jpg`; // Add Menu Button to ComfyUI function addMenuButton() { diff --git a/web/js/ui/previewRenderer.js b/web/js/ui/previewRenderer.js index f88d0c7..adc223d 100644 --- a/web/js/ui/previewRenderer.js +++ b/web/js/ui/previewRenderer.js @@ -1,6 +1,6 @@ // Renders the download preview panel -const PLACEHOLDER_IMAGE_URL = `/custom_nodes/ComfyUI-HuggingFace/web/images/placeholder.jpg`; +const PLACEHOLDER_IMAGE_URL = `./web/images/placeholder.jpg`; export function renderDownloadPreview(ui, data) { if (!ui.downloadPreviewArea) return; diff --git a/web/js/ui/searchRenderer.js b/web/js/ui/searchRenderer.js index c98c415..2a5b59c 100644 --- a/web/js/ui/searchRenderer.js +++ b/web/js/ui/searchRenderer.js @@ -1,7 +1,7 @@ // Rendering of search results list // Usage: renderSearchResults(uiInstance, itemsArray) -const PLACEHOLDER_IMAGE_URL = `/custom_nodes/ComfyUI-HuggingFace/images/placeholder.jpg`; +const PLACEHOLDER_IMAGE_URL = `./images/placeholder.jpg`; export function renderSearchResults(ui, items) { ui.feedback?.ensureFontAwesome(); diff --git a/web/js/ui/statusRenderer.js b/web/js/ui/statusRenderer.js index d3aec4a..ba6faa5 100644 --- a/web/js/ui/statusRenderer.js +++ b/web/js/ui/statusRenderer.js @@ -1,6 +1,6 @@ // Renders active/queued/history download lists -const PLACEHOLDER_IMAGE_URL = `/custom_nodes/ComfyUI-HuggingFace/images/placeholder.jpg`; +const PLACEHOLDER_IMAGE_URL = `./images/placeholder.jpg`; export function renderDownloadList(ui, items, container, emptyMessage) { if (!items || items.length === 0) {