Add delete confirmation modal and update related styles and logic

nodes.py

@@ -10,6 +10,7 @@ from .utils.file_utils import get_file_info, save_metadata, load_metadata, updat
 from .utils.lora_metadata import extract_lora_metadata
 from typing import Dict, Optional
 from .services.civitai_client import CivitaiClient
+import folder_paths
 
 class LorasEndpoint:
     def __init__(self):
@@ -19,9 +20,12 @@ class LorasEndpoint:
             ),
             autoescape=True
         )
-        # 配置Loras根目录（根据实际安装位置调整）
-        self.loras_root = os.path.join(Path(__file__).parents[2], "models", "loras")
-        # 添加 server 属性
+        # Configure Loras root directories (from ComfyUI folder paths settings)
+        self.loras_roots = [path.replace(os.sep, "/") for path in folder_paths.get_folder_paths("loras") if os.path.exists(path)]
+        if not self.loras_roots:
+            raise ValueError("No valid loras folders found")
+        print(f"Loras roots: {self.loras_roots}") # debug log
+        
         self.server = PromptServer.instance
 
     @classmethod
@@ -29,9 +33,16 @@ class LorasEndpoint:
         instance = cls()
         app = PromptServer.instance.app
         static_path = os.path.join(os.path.dirname(__file__), 'static')
+
+        # Generate multiple static paths based on the number of folders in instance.loras_roots
+        for idx, root in enumerate(instance.loras_roots, start=1):
+            # Create different static paths for each folder, like /loras_static/root1/preview
+            preview_path = f'/loras_static/root{idx}/preview'
+            app.add_routes([web.static(preview_path, root)])
+
         app.add_routes([
             web.get('/loras', instance.handle_loras_request),
-            web.static('/loras_static/previews', instance.loras_root),
+            # web.static('/loras_static/previews', instance.loras_root),
             web.static('/loras_static', static_path),
             web.post('/api/delete_model', instance.delete_model),
             web.post('/api/fetch-civitai', instance.fetch_civitai),
@@ -52,34 +63,35 @@ class LorasEndpoint:
 
     async def scan_loras(self):
         loras = []
-        for root, _, files in os.walk(self.loras_root):
-            safetensors_files = [f for f in files if f.endswith('.safetensors')]
-            total_files = len(safetensors_files)
-            
-            for idx, filename in enumerate(safetensors_files, 1):
-                self.send_progress(idx, total_files, f"Scanning: {filename}")
+        for loras_root in self.loras_roots:
+            for root, _, files in os.walk(loras_root):
+                safetensors_files = [f for f in files if f.endswith('.safetensors')]
+                total_files = len(safetensors_files)
                 
-                file_path = os.path.join(root, filename)
-                
-                # Try to load existing metadata first
-                metadata = await load_metadata(file_path, self.loras_root)
-                
-                if metadata is None:
-                    # Only get file info and extract metadata if no existing metadata
-                    metadata = await get_file_info(file_path, self.loras_root)
-                    base_model_info = await extract_lora_metadata(file_path)
-                    metadata.base_model = base_model_info['base_model']
-                    await save_metadata(file_path, metadata)
-                
-                # Convert to dict for API response
-                lora_data = metadata.to_dict()
-                # Get relative path and remove filename to get just the folder structure
-                rel_path = os.path.relpath(file_path, self.loras_root)
-                folder = os.path.dirname(rel_path)
-                # Ensure forward slashes for consistency across platforms
-                lora_data['folder'] = folder.replace(os.path.sep, '/')
-                
-                loras.append(lora_data)
+                for idx, filename in enumerate(safetensors_files, 1):
+                    self.send_progress(idx, total_files, f"Scanning: {filename}")
+                    
+                    file_path = os.path.join(root, filename).replace(os.sep, "/")
+                    
+                    # Try to load existing metadata first
+                    metadata = await load_metadata(file_path)
+                    
+                    if metadata is None:
+                        # Only get file info and extract metadata if no existing metadata
+                        metadata = await get_file_info(file_path)
+                        base_model_info = await extract_lora_metadata(file_path)
+                        metadata.base_model = base_model_info['base_model']
+                        await save_metadata(file_path, metadata)
+                    
+                    # Convert to dict for API response
+                    lora_data = metadata.to_dict()
+                    # Get relative path and remove filename to get just the folder structure
+                    rel_path = os.path.relpath(file_path, loras_root)
+                    folder = os.path.dirname(rel_path)
+                    # Ensure forward slashes for consistency across platforms
+                    lora_data['folder'] = folder.replace(os.path.sep, '/')
+                    
+                    loras.append(lora_data)
             
         self.send_progress(total_files, total_files, "Scan completed")
         return loras
@@ -89,13 +101,6 @@ class LorasEndpoint:
         """清理HTML格式的描述"""
         return desc.replace("<p>", "").replace("</p>", "\n").strip()
 
-    async def get_preview_url(self, preview_path, root_dir):
-        """生成预览图URL"""
-        if os.path.exists(preview_path):
-            rel_path = os.path.relpath(preview_path, self.loras_root)
-            return f"/loras_static/previews/{rel_path.replace(os.sep, '/')}"
-        return "/loras_static/images/no-preview.png"
-
     async def handle_loras_request(self, request):
         """处理Loras请求并渲染模板"""
         try:
@@ -145,29 +150,43 @@ class LorasEndpoint:
         return {
             "model_name": lora["model_name"],
             "file_name": lora["file_name"],   
-            "preview_url": lora["preview_url"],
+            "preview_url": self.get_static_url_for_preview(lora["preview_url"]),
             "base_model": lora["base_model"],
             "folder": lora["folder"],
             "sha256": lora["sha256"],
-            "file_path": lora["file_path"],
+            "file_path": lora["file_path"].replace(os.sep, "/"),
             "modified": lora["modified"],
             "from_civitai": lora.get("from_civitai", True),
             "civitai": self.filter_civitai_data(lora.get("civitai", {}))
         }
+    
+    def get_static_url_for_preview(self, preview_url):
+        """
+        Determines which loras_root the preview_url belongs to and
+        returns the corresponding static URL.
+        """
+        for idx, root in enumerate(self.loras_roots, start=1):
+            # Check if preview_url belongs to current root
+            if preview_url.startswith(root):
+                # Get relative path and generate static URL 
+                relative_path = os.path.relpath(preview_url, root)
+                static_url = f'/loras_static/root{idx}/preview/{relative_path.replace(os.sep, "/")}'
+                return static_url
+        
+        # If no matching root found, return empty string
+        return ""
 
 
     async def delete_model(self, request):
         try:
             data = await request.json()
-            file_name = data.get('file_name')
-            folder = data.get('folder')  # 从请求中获取folder信息
-            if not file_name:
-                return web.Response(text='Model name is required', status=400)
+            file_path = data.get('file_path')  # 从请求中获取file_path信息
+            if not file_path:
+                return web.Response(text='Model full path is required', status=400)
 
             # 构建完整的目录路径
-            target_dir = self.loras_root
-            if folder and folder != "root":
-                target_dir = os.path.join(self.loras_root, folder)
+            target_dir = os.path.dirname(file_path)
+            file_name = os.path.splitext(os.path.basename(file_path))[0]
 
             # List of file patterns to delete
             required_file = f"{file_name}.safetensors"  # 主文件必须存在
@@ -176,7 +195,13 @@ class LorasEndpoint:
                 f"{file_name}.preview.png",
                 f"{file_name}.preview.jpg",
                 f"{file_name}.preview.jpeg",
-                f"{file_name}.preview.webp"
+                f"{file_name}.preview.webp",
+                f"{file_name}.preview.mp4",
+                f"{file_name}.png",
+                f"{file_name}.jpg",
+                f"{file_name}.jpeg",
+                f"{file_name}.webp",
+                f"{file_name}.mp4"
             ]
             
             deleted_files = []
@@ -201,7 +226,7 @@ class LorasEndpoint:
                         except Exception as e:
                             print(f"Error deleting optional file {file_path}: {str(e)}")
             else:
-                return web.Response(text=f"Model file {required_file} not found in {folder}", status=404)
+                return web.Response(text=f"Model file {required_file} not found in {target_dir}", status=404)
                 
             return web.json_response({
                 'success': True,
@@ -268,7 +293,7 @@ class LorasEndpoint:
                 
                 # 4. 下载预览图
                 # Check if existing preview is valid and the file exists
-                if not local_metadata.get('preview_url') or not os.path.exists(os.path.join(self.loras_root, local_metadata['preview_url'].replace('/', os.sep))):
+                if not local_metadata.get('preview_url') or not os.path.exists(local_metadata['preview_url']):
                     first_preview = next((img for img in civitai_metadata.get('images', [])), None)
                     if first_preview:
                         
@@ -277,7 +302,7 @@ class LorasEndpoint:
                         preview_path = os.path.join(os.path.dirname(data['file_path']), preview_filename)
                         await client.download_preview_image(first_preview['url'], preview_path)
                         # 存储相对路径，使用正斜杠格式
-                        local_metadata['preview_url'] = os.path.relpath(preview_path, self.loras_root).replace(os.sep, '/')
+                        local_metadata['preview_url'] = preview_path.replace(os.sep, '/')
 
                 # 5. 保存更新后的元数据
                 with open(metadata_path, 'w', encoding='utf-8') as f:
@@ -307,28 +332,22 @@ class LorasEndpoint:
         try:
             reader = await request.multipart()
             
-            # Get the file field first
+            # Get the preview_file field first
             file_field = await reader.next()
-            if file_field.name != 'file':
-                raise ValueError("Expected 'file' field first")
-            file_data = await file_field.read()
+            if file_field.name != 'preview_file':
+                raise ValueError("Expected 'preview_file' field first")
+            preview_data = await file_field.read()
             
-            # Get the file name field
+            # Get the file model_path field
             name_field = await reader.next()
-            if name_field.name != 'file_name':
-                raise ValueError("Expected 'file_name' field second")
-            file_name = (await name_field.read()).decode()
-
-            # Get the folder field
-            folder_field = await reader.next()
-            if folder_field.name != 'folder':
-                raise ValueError("Expected 'folder' field third")
-            folder = (await folder_field.read()).decode()
+            if name_field.name != 'model_path':
+                raise ValueError("Expected 'model_path' field second")
+            model_path = (await name_field.read()).decode()
             
             # Get the content type from the file field headers
             content_type = file_field.headers.get('Content-Type', '')
 
-            print(f"Received preview file: {file_name} ({content_type})")  # Debug log
+            print(f"Received preview file: {model_path} ({content_type})")  # Debug log
             
             # Determine file extension based on content type
             if content_type.startswith('video/'):
@@ -337,29 +356,34 @@ class LorasEndpoint:
                 extension = '.preview.png'
             
             # Construct the preview file path
-            base_name = os.path.splitext(file_name)[0]  # Remove original extension
+            base_name = os.path.splitext(os.path.basename(model_path))[0]  # Remove original extension
             preview_name = base_name + extension
-            preview_path = os.path.join(self.loras_root, folder, preview_name)
+            # Get the folder path from the model_path
+            folder = os.path.dirname(model_path)
+            preview_path = os.path.join(folder, preview_name).replace(os.sep, '/')
             
             # Save the preview file
             with open(preview_path, 'wb') as f:
-                f.write(file_data)
+                f.write(preview_data)
             
             # Update metadata if it exists
-            metadata_path = os.path.join(self.loras_root, folder, base_name + '.metadata.json')
+            metadata_path = os.path.join(folder, base_name + '.metadata.json')
             if os.path.exists(metadata_path):
                 try:
                     with open(metadata_path, 'r', encoding='utf-8') as f:
                         metadata = json.load(f)
                     # Update the preview_url to match the new file name
-                    metadata['preview_url'] = os.path.join(folder, preview_name).replace(os.sep, '/')
+                    metadata['preview_url'] = preview_path
                     with open(metadata_path, 'w', encoding='utf-8') as f:
                         json.dump(metadata, f, indent=2, ensure_ascii=False)
                 except Exception as e:
                     print(f"Error updating metadata: {str(e)}")
                     # Continue even if metadata update fails
             
-            return web.Response(status=200)
+            return web.json_response({
+                "success": True,
+                "preview_url": self.get_static_url_for_preview(preview_path)
+            })
             
         except Exception as e:
             print(f"Error replacing preview: {str(e)}")

static/css/style.css

@@ -302,6 +302,114 @@ body.modal-open {
     box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1);
 }
 
+/* Delete Modal specific styles */
+.delete-modal-content {
+    max-width: 500px;
+    text-align: center;
+}
+
+.delete-message {
+    color: var(--text-color);
+    margin: var(--space-2) 0;
+}
+
+.delete-model-info {
+    background: var(--lora-surface);
+    border: 1px solid var(--lora-border);
+    border-radius: 8px;
+    padding: var(--space-2);
+    margin: var(--space-2) 0;
+    color: var(--text-color);
+    word-break: break-all;
+}
+
+/* Update delete modal styles */
+.delete-modal {
+    display: none;  /* Set initial display to none */
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background: rgba(0, 0, 0, 0.8);
+    z-index: var(--z-overlay);
+}
+
+/* Add new style for when modal is shown */
+.delete-modal.show {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+}
+
+.delete-modal-content {
+    max-width: 500px;
+    width: 90%;
+    text-align: center;
+    margin: 0 auto;
+    position: relative;
+    animation: modalFadeIn 0.2s ease-out;
+}
+
+.delete-model-info {
+    /* Update info display styling */
+    background: var(--lora-surface);
+    border: 1px solid var(--lora-border);
+    border-radius: 8px;
+    padding: var(--space-2);
+    margin: var(--space-2) 0;
+    color: var(--text-color);
+    word-break: break-all;
+    text-align: left;
+    line-height: 1.5;
+}
+
+@keyframes modalFadeIn {
+    from {
+        opacity: 0;
+        transform: translateY(-20px);
+    }
+    to {
+        opacity: 1;
+        transform: translateY(0);
+    }
+}
+
+.modal-actions {
+    display: flex;
+    gap: var(--space-2);
+    justify-content: center;
+    margin-top: var(--space-3);
+}
+
+.cancel-btn, .delete-btn {
+    padding: 8px var(--space-2);
+    border-radius: 6px;
+    border: none;
+    cursor: pointer;
+    font-weight: 500;
+    min-width: 100px;
+}
+
+.cancel-btn {
+    background: var(--lora-surface);
+    border: 1px solid var(--lora-border);
+    color: var(--text-color);
+}
+
+.delete-btn {
+    background: var(--lora-error);
+    color: white;
+}
+
+.cancel-btn:hover {
+    background: var(--lora-border);
+}
+
+.delete-btn:hover {
+    opacity: 0.9;
+}
+
 .carousel {
     display: grid;
     grid-auto-flow: column;

static/js/script.js

@@ -96,45 +96,78 @@ function openCivitai(modelName) {
     }
 }
 
-async function deleteModel(fileName) {
-    // Prevent event bubbling
-    event.stopPropagation();
-    
-    // Get the folder from the card's data attributes
-    const card = document.querySelector(`.lora-card[data-file_name="${fileName}"]`);
-    const folder = card ? card.dataset.folder : null;
-    
-    // Show confirmation dialog
-    const confirmed = confirm(`Are you sure you want to delete "${fileName}" and all associated files?`);
-    
-    if (confirmed) {
-        try {
-            const response = await fetch('/api/delete_model', {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({ 
-                    file_name: fileName,
-                    folder: folder
-                })
-            });
+let pendingDeletePath = null;
 
-            if (response.ok) {
-                // Remove the card from UI
-                if (card) {
-                    card.remove();
-                }
-            } else {
-                const error = await response.text();
-                alert(`Failed to delete model: ${error}`);
-            }
-        } catch (error) {
-            alert(`Error deleting model: ${error}`);
+function showDeleteModal(filePath) {
+    event.stopPropagation();
+    pendingDeletePath = filePath;
+    
+    const card = document.querySelector(`.lora-card[data-filepath="${filePath}"]`);
+    const modelName = card.dataset.name;
+    const modal = document.getElementById('deleteModal');
+    const modelInfo = modal.querySelector('.delete-model-info');
+    
+    // Format the info with better structure
+    modelInfo.innerHTML = `
+        <strong>Model:</strong> ${modelName}
+        <br>
+        <strong>File:</strong> ${filePath}
+    `;
+    
+    modal.classList.add('show');  // Use class instead of style.display
+    document.body.classList.add('modal-open');
+
+    // Add click outside to close
+    modal.onclick = function(event) {
+        if (event.target === modal) {
+            closeDeleteModal();
         }
+    };
+}
+
+function closeDeleteModal() {
+    const modal = document.getElementById('deleteModal');
+    modal.classList.remove('show');  // Use class instead of style.display
+    document.body.classList.remove('modal-open');
+    pendingDeletePath = null;
+}
+
+async function confirmDelete() {
+    if (!pendingDeletePath) return;
+    
+    const modal = document.getElementById('deleteModal');
+    const card = document.querySelector(`.lora-card[data-filepath="${pendingDeletePath}"]`);
+    
+    try {
+        const response = await fetch('/api/delete_model', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({ 
+                file_path: pendingDeletePath
+            })
+        });
+
+        if (response.ok) {
+            if (card) {
+                card.remove();
+            }
+            closeDeleteModal();
+        } else {
+            const error = await response.text();
+            alert(`Failed to delete model: ${error}`);
+        }
+    } catch (error) {
+        alert(`Error deleting model: ${error}`);
     }
 }
 
+// Replace the existing deleteModel function with this one
+async function deleteModel(filePath) {
+    showDeleteModal(filePath);
+}
+
 // 初始化排序
 document.getElementById('sortSelect')?.addEventListener('change', (e) => {
     sortCards(e.target.value);
@@ -381,7 +414,7 @@ async function fetchCivitai() {
     }
 }
 
-async function replacePreview(fileName, folder) {
+async function replacePreview(filePath) {
     // Get loading elements first
     const loadingOverlay = document.getElementById('loading-overlay');
     const loadingStatus = document.querySelector('.loading-status');
@@ -397,9 +430,8 @@ async function replacePreview(fileName, folder) {
         
         const file = input.files[0];
         const formData = new FormData();
-        formData.append('file', file);
-        formData.append('file_name', fileName);
-        formData.append('folder', folder);
+        formData.append('preview_file', file);
+        formData.append('model_path', filePath);
         
         try {
             // Show loading overlay
@@ -414,18 +446,15 @@ async function replacePreview(fileName, folder) {
             if (!response.ok) {
                 throw new Error('Upload failed');
             }
+
+            const data = await response.json();
+            const newPreviewPath = `${data.preview_url}?t=${new Date().getTime()}`;
             
             // Update the preview image in the card
-            const card = document.querySelector(`.lora-card[data-file_name="${fileName}"]`);
+            const card = document.querySelector(`.lora-card[data-filepath="${filePath}"]`);
             const previewContainer = card.querySelector('.card-preview');
             const oldPreview = previewContainer.querySelector('img, video');
             
-            // Force reload the preview by adding a timestamp
-            const timestamp = new Date().getTime();
-            const baseName = fileName.split('.')[0];
-            const extension = file.type.startsWith('video/') ? '.preview.mp4' : '.preview.png';
-            const newPreviewPath = `/loras_static/previews/${folder}/${baseName}${extension}?t=${timestamp}`;
-            
             // Create new preview element based on file type
             if (file.type.startsWith('video/')) {
                 const video = document.createElement('video');

templates/loras.html

@@ -27,6 +27,19 @@
         </div>
     </div>
 
+    <!-- Delete Confirmation Modal -->
+    <div id="deleteModal" class="modal delete-modal">
+        <div class="modal-content delete-modal-content">
+            <h2>Delete Model</h2>
+            <p class="delete-message">Are you sure you want to delete this model and all associated files?</p>
+            <div class="delete-model-info"></div>
+            <div class="modal-actions">
+                <button class="cancel-btn" onclick="closeDeleteModal()">Cancel</button>
+                <button class="delete-btn" onclick="confirmDelete()">Delete</button>
+            </div>
+        </div>
+    </div>
+
     <div class="container">
         <!-- 控制栏 -->
         <div class="controls">
@@ -61,11 +74,11 @@
                 <div class="card-preview">
                     {% if lora.preview_url.endswith('.mp4') or lora.preview_url.endswith('.webm') %}
                         <video controls autoplay muted loop>
-                            <source src="{{ '/loras_static/previews/' + lora.preview_url }}" type="video/mp4">
+                            <source src="{{ lora.preview_url }}" type="video/mp4">
                             Your browser does not support the video tag.
                         </video>
                     {% else %}
-                        <img src="{{ ('/loras_static/previews/' + lora.preview_url) if lora.preview_url else '/loras_static/images/no-preview.png' }}" alt="{{ lora.name }}">
+                        <img src="{{ lora.preview_url if lora.preview_url else '/loras_static/images/no-preview.png' }}" alt="{{ lora.name }}">
                     {% endif %}
                     <div class="card-header">
                         <span class="base-model-label" title="{{ lora.base_model }}">
@@ -81,7 +94,7 @@
                                onclick="event.stopPropagation(); navigator.clipboard.writeText(this.closest('.lora-card').dataset.file_name)"></i>
                             <i class="fas fa-trash" 
                                title="Delete Model"
-                               onclick="event.stopPropagation(); deleteModel('{{ lora.file_name }}')"></i>
+                               onclick="event.stopPropagation(); deleteModel('{{ lora.file_path }}')"></i>
                         </div>
                     </div>
                     <div class="card-footer">
@@ -91,7 +104,7 @@
                         <div class="card-actions">
                             <i class="fas fa-image" 
                                title="Replace Preview Image"
-                               onclick="event.stopPropagation(); replacePreview('{{ lora.file_name }}', '{{ lora.folder }}')"></i>
+                               onclick="event.stopPropagation(); replacePreview('{{ lora.file_path }}')"></i>
                         </div>
                     </div>
                 </div>

utils/file_utils.py

@@ -12,7 +12,7 @@ async def calculate_sha256(file_path: str) -> str:
             sha256_hash.update(byte_block)
     return sha256_hash.hexdigest()
 
-def _find_preview_file(base_name: str, dir_path: str, lora_root: str) -> str:
+def _find_preview_file(base_name: str, dir_path: str) -> str:
     """Find preview file for given base name in directory"""
     preview_patterns = [
         f"{base_name}.preview.png",
@@ -28,15 +28,15 @@ def _find_preview_file(base_name: str, dir_path: str, lora_root: str) -> str:
     for pattern in preview_patterns:
         full_pattern = os.path.join(dir_path, pattern)
         if os.path.exists(full_pattern):
-            return os.path.relpath(full_pattern, lora_root).replace("\\", "/")
+            return full_pattern.replace("\\", "/")
     return ""
 
-async def get_file_info(file_path: str, loras_root: str) -> LoraMetadata:
+async def get_file_info(file_path: str) -> LoraMetadata:
     """Get basic file information as LoraMetadata object"""
     base_name = os.path.splitext(os.path.basename(file_path))[0]
     dir_path = os.path.dirname(file_path)
     
-    preview_path = _find_preview_file(base_name, dir_path, loras_root)
+    preview_url = _find_preview_file(base_name, dir_path)
     
     return LoraMetadata(
         file_name=base_name,
@@ -47,7 +47,7 @@ async def get_file_info(file_path: str, loras_root: str) -> LoraMetadata:
         sha256=await calculate_sha256(file_path),
         base_model="Unknown",  # Will be updated later
         from_civitai=True,
-        preview_url=preview_path,
+        preview_url=preview_url,
     )
 
 async def save_metadata(file_path: str, metadata: LoraMetadata) -> None:
@@ -59,7 +59,7 @@ async def save_metadata(file_path: str, metadata: LoraMetadata) -> None:
     except Exception as e:
         print(f"Error saving metadata to {metadata_path}: {str(e)}")
 
-async def load_metadata(file_path: str, loras_root: str) -> Optional[LoraMetadata]:
+async def load_metadata(file_path: str) -> Optional[LoraMetadata]:
     """Load metadata from .metadata.json file"""
     metadata_path = f"{os.path.splitext(file_path)[0]}.metadata.json"
     try:
@@ -67,11 +67,11 @@ async def load_metadata(file_path: str, loras_root: str) -> Optional[LoraMetadat
             with open(metadata_path, 'r', encoding='utf-8') as f:
                 data = json.load(f)
                 preview_url = data.get('preview_url', '')
-                preview_path = os.path.join(loras_root, preview_url) if preview_url else ''
-                if not preview_url or not os.path.exists(preview_path):
+                # preview_path = os.path.join(loras_root, preview_url) if preview_url else ''
+                if not preview_url or not os.path.exists(preview_url):
                     base_name = os.path.splitext(os.path.basename(file_path))[0]
                     dir_path = os.path.dirname(file_path)
-                    data['preview_url'] = _find_preview_file(base_name, dir_path, loras_root)
+                    data['preview_url'] = _find_preview_file(base_name, dir_path)
                     if data['preview_url']:
                         with open(metadata_path, 'w', encoding='utf-8') as f:
                             json.dump(data, f, indent=2, ensure_ascii=False)