checkpoint

2026-03-24 22:52:12 -03:00 · 2025-03-17 15:10:11 +08:00
parent 1034282161
commit a4ee82b51f
9 changed files with 808 additions and 16 deletions
--- a/static/js/managers/LoraSearchManager.js
+++ b/static/js/managers/LoraSearchManager.js
@@ -0,0 +1,135 @@
+/**
+ * LoraSearchManager - Specialized search manager for the LoRAs page
+ * Extends the base SearchManager with LoRA-specific functionality
+ */
+import { SearchManager } from './SearchManager.js';
+import { appendLoraCards } from '../api/loraApi.js';
+import { resetAndReload } from '../api/loraApi.js';
+import { state } from '../state/index.js';
+import { showToast } from '../utils/uiHelpers.js';
+
+export class LoraSearchManager extends SearchManager {
+  constructor(options = {}) {
+    console.log("initializing lora search manager");
+    super({
+      page: 'loras',
+      ...options
+    });
+    
+    this.currentSearchTerm = '';
+    
+    // Store this instance in the state
+    if (state) {
+      state.searchManager = this;
+    }
+  }
+  
+  async performSearch() {
+    const searchTerm = this.searchInput.value.trim().toLowerCase();
+    
+    // Log the search attempt for debugging
+    console.log('LoraSearchManager performSearch called with:', searchTerm);
+    
+    if (searchTerm === this.currentSearchTerm && !this.isSearching) {
+      return; // Avoid duplicate searches
+    }
+    
+    this.currentSearchTerm = searchTerm;
+    
+    const grid = document.getElementById('loraGrid');
+    if (!grid) {
+      console.error('Error: Could not find loraGrid element');
+      return;
+    }
+    
+    if (!searchTerm) {
+      if (state) {
+        state.currentPage = 1;
+      }
+      await resetAndReload();
+      return;
+    }
+
+    try {
+      this.isSearching = true;
+      if (state && state.loadingManager) {
+        state.loadingManager.showSimpleLoading('Searching...');
+      }
+
+      // Store current scroll position
+      const scrollPosition = window.pageYOffset || document.documentElement.scrollTop;
+      
+      if (state) {
+        state.currentPage = 1;
+        state.hasMore = true;
+      }
+
+      const url = new URL('/api/loras', window.location.origin);
+      url.searchParams.set('page', '1');
+      url.searchParams.set('page_size', '20');
+      url.searchParams.set('sort_by', state ? state.sortBy : 'name');
+      url.searchParams.set('search', searchTerm);
+      url.searchParams.set('fuzzy', 'true');
+      
+      // Add search options
+      const searchOptions = this.getActiveSearchOptions();
+      console.log('Active search options:', searchOptions);
+      
+      // Make sure we're sending boolean values as strings
+      url.searchParams.set('search_filename', searchOptions.filename ? 'true' : 'false');
+      url.searchParams.set('search_modelname', searchOptions.modelname ? 'true' : 'false');
+      url.searchParams.set('search_tags', searchOptions.tags ? 'true' : 'false');
+
+      // Always send folder parameter if there is an active folder
+      if (state && state.activeFolder) {
+        url.searchParams.set('folder', state.activeFolder);
+        // Add recursive parameter when recursive search is enabled
+        const recursive = this.recursiveSearchToggle ? this.recursiveSearchToggle.checked : false;
+        url.searchParams.set('recursive', recursive.toString());
+      }
+
+      console.log('Search URL:', url.toString());
+      const response = await fetch(url);
+      
+      if (!response.ok) {
+        throw new Error(`Search failed with status: ${response.status}`);
+      }
+
+      const data = await response.json();
+      console.log('Search results:', data);
+      
+      if (searchTerm === this.currentSearchTerm) {
+        grid.innerHTML = '';
+        
+        if (data.items.length === 0) {
+          grid.innerHTML = '<div class="no-results">No matching loras found</div>';
+          if (state) {
+            state.hasMore = false;
+          }
+        } else {
+          appendLoraCards(data.items);
+          if (state) {
+            state.hasMore = state.currentPage < data.total_pages;
+            state.currentPage++;
+          }
+        }
+        
+        // Restore scroll position after content is loaded
+        setTimeout(() => {
+          window.scrollTo({
+            top: scrollPosition,
+            behavior: 'instant' // Use 'instant' to prevent animation
+          });
+        }, 10);
+      }
+    } catch (error) {
+      console.error('Search error:', error);
+      showToast('Search failed', 'error');
+    } finally {
+      this.isSearching = false;
+      if (state && state.loadingManager) {
+        state.loadingManager.hide();
+      }
+    }
+  }
+}