Add replace preview

nodes.py

@@ -34,7 +34,8 @@ class LorasEndpoint:
             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)
+            web.post('/api/fetch-civitai', instance.fetch_civitai),
+            web.post('/api/replace_preview', instance.replace_preview),
         ])
     
     def send_progress(self, current, total, status="Scanning"):
@@ -106,12 +107,6 @@ class LorasEndpoint:
             formatted_loras = [self.format_lora(l) for l in data]
             folders = sorted(list(set(l['folder'] for l in data)))
 
-            # Debug logging
-            if formatted_loras:
-                print(f"Sample lora data: {formatted_loras[0]}")
-            else:
-                print("Warning: No loras found")
-                
             context = {
                 "loras": formatted_loras,
                 "folders": folders,
@@ -263,10 +258,8 @@ class LorasEndpoint:
                 print(f"Error downloading preview image: {str(e)}")
 
     async def fetch_civitai(self, request):
-        print("Received fetch-civitai request")  # Debug log
         try:
             data = await request.json()
-            print(f"Request data: {data}")  # Debug log
             client = CivitaiClient()
             
             try:
@@ -338,5 +331,67 @@ class LorasEndpoint:
                 "error": f"Request processing error: {str(e)}"
             }, status=400)
 
+    async def replace_preview(self, request):
+        try:
+            reader = await request.multipart()
+            
+            # Get the 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()
+            
+            # Get the file name 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()
+            
+            # 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
+            
+            # Determine file extension based on content type
+            if content_type.startswith('video/'):
+                extension = '.preview.mp4'
+            else:
+                extension = '.preview.png'
+            
+            # Construct the preview file path
+            base_name = os.path.splitext(file_name)[0]  # Remove original extension
+            preview_name = base_name + extension
+            preview_path = os.path.join(self.loras_root, folder, preview_name)
+            
+            # Save the preview file
+            with open(preview_path, 'wb') as f:
+                f.write(file_data)
+            
+            # Update metadata if it exists
+            metadata_path = os.path.join(self.loras_root, 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, '/')
+                    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)
+            
+        except Exception as e:
+            print(f"Error replacing preview: {str(e)}")
+            return web.Response(text=str(e), status=500)
+
 # 注册路由
 LorasEndpoint.add_routes()

static/js/script.js

@@ -369,4 +369,76 @@ async function fetchCivitai() {
     }
 }
 
+async function replacePreview(fileName, folder) {
+    // Get loading elements first
+    const loadingOverlay = document.getElementById('loading-overlay');
+    const loadingStatus = document.querySelector('.loading-status');
+    
+    // Create a file input element
+    const input = document.createElement('input');
+    input.type = 'file';
+    input.accept = 'image/*,video/mp4';  // Accept images and MP4 videos
+    
+    // Handle file selection
+    input.onchange = async function() {
+        if (!input.files || !input.files[0]) return;
+        
+        const file = input.files[0];
+        const formData = new FormData();
+        formData.append('file', file);
+        formData.append('file_name', fileName);
+        formData.append('folder', folder);
+        
+        try {
+            // Show loading overlay
+            loadingOverlay.style.display = 'flex';
+            loadingStatus.textContent = 'Uploading preview...';
+            
+            const response = await fetch('/api/replace_preview', {
+                method: 'POST',
+                body: formData
+            });
+            
+            if (!response.ok) {
+                throw new Error('Upload failed');
+            }
+            
+            // Update the preview image in the card
+            const card = document.querySelector(`.lora-card[data-file_name="${fileName}"]`);
+            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');
+                video.controls = true;
+                video.autoplay = true;
+                video.muted = true;
+                video.loop = true;
+                video.src = newPreviewPath;
+                oldPreview.replaceWith(video);
+            } else {
+                const img = document.createElement('img');
+                img.src = newPreviewPath;
+                oldPreview.replaceWith(img);
+            }
+            
+        } catch (error) {
+            console.error('Error uploading preview:', error);
+            alert('Failed to upload preview image');
+        } finally {
+            loadingOverlay.style.display = 'none';
+        }
+    };
+    
+    // Trigger file selection
+    input.click();
+}
+
 initTheme();

templates/loras.html

@@ -88,7 +88,7 @@
                         <div class="card-actions">
                             <i class="fas fa-image" 
                                title="Replace Preview Image"
-                               onclick="event.stopPropagation(); replacePreview('{{ lora.file_name }}')"></i>
+                               onclick="event.stopPropagation(); replacePreview('{{ lora.file_name }}', '{{ lora.folder }}')"></i>
                         </div>
                     </div>
                 </div>