From 13f13eb095086e981c824ea56b795a0657732466 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Sat, 13 Sep 2025 09:20:55 +0800
Subject: [PATCH] fix: update preview versions keys for consistency in state
 management, fixes #406

---
 static/js/components/shared/ModelCard.js | 8 --------
 static/js/state/index.js                 | 6 +++---
 2 files changed, 3 insertions(+), 11 deletions(-)

diff --git a/static/js/components/shared/ModelCard.js b/static/js/components/shared/ModelCard.js
index 16bdf45d..aa3208a0 100644
--- a/static/js/components/shared/ModelCard.js
+++ b/static/js/components/shared/ModelCard.js
@@ -216,13 +216,6 @@ function handleCardClick(card, modelType) {
 }
 
 async function showModelModalFromCard(card, modelType) {
-    // Get the appropriate preview versions map
-    const previewVersionsKey = modelType;
-    const previewVersions = state.pages[previewVersionsKey]?.previewVersions || new Map();
-    const version = previewVersions.get(card.dataset.filepath);
-    const previewUrl = card.dataset.preview_url || '/loras_static/images/no-preview.png';
-    const versionedPreviewUrl = version ? `${previewUrl}?t=${version}` : previewUrl;
-    
     // Create model metadata object
     const modelMeta = {
         sha256: card.dataset.sha256,
@@ -235,7 +228,6 @@ async function showModelModalFromCard(card, modelType) {
         from_civitai: card.dataset.from_civitai === 'true',
         base_model: card.dataset.base_model,
         notes: card.dataset.notes || '',
-        preview_url: versionedPreviewUrl,
         favorite: card.dataset.favorite === 'true',
         // Parse civitai metadata from the card's dataset
         civitai: JSON.parse(card.dataset.meta || '{}'),
diff --git a/static/js/state/index.js b/static/js/state/index.js
index 65d5619f..71f6b859 100644
--- a/static/js/state/index.js
+++ b/static/js/state/index.js
@@ -10,9 +10,9 @@ const savedSettings = getStorageItem('settings', {
 });
 
 // Load preview versions from localStorage for each model type
-const loraPreviewVersions = getMapFromStorage('lora_preview_versions');
-const checkpointPreviewVersions = getMapFromStorage('checkpoint_preview_versions');
-const embeddingPreviewVersions = getMapFromStorage('embedding_preview_versions');
+const loraPreviewVersions = getMapFromStorage('loras_preview_versions');
+const checkpointPreviewVersions = getMapFromStorage('checkpoints_preview_versions');
+const embeddingPreviewVersions = getMapFromStorage('embeddings_preview_versions');
 
 export const state = {
     // Global state