Add cache management feature: implement clear cache API and modal confirmation

py/routes/misc_routes.py

@@ -4,6 +4,7 @@ import asyncio
 import json
 import time
 import aiohttp
+import shutil
 from server import PromptServer # type: ignore
 from aiohttp import web
 from ..services.settings_manager import settings
@@ -39,6 +40,9 @@ class MiscRoutes:
     def setup_routes(app):
         """Register miscellaneous routes"""
         app.router.add_post('/api/settings', MiscRoutes.update_settings)
+        
+        # Add new route for clearing cache
+        app.router.add_post('/api/clear-cache', MiscRoutes.clear_cache)
 
         # Usage stats routes
         app.router.add_post('/api/update-usage-stats', MiscRoutes.update_usage_stats)
@@ -53,6 +57,55 @@ class MiscRoutes:
         # Lora code update endpoint
         app.router.add_post('/api/update-lora-code', MiscRoutes.update_lora_code)
     
+    @staticmethod
+    async def clear_cache(request):
+        """Clear all cache files from the cache folder"""
+        try:
+            # Get the cache folder path (relative to project directory)
+            project_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+            cache_folder = os.path.join(project_dir, 'cache')
+            
+            # Check if cache folder exists
+            if not os.path.exists(cache_folder):
+                logger.info("Cache folder does not exist, nothing to clear")
+                return web.json_response({'success': True, 'message': 'No cache folder found'})
+            
+            # Get list of cache files before deleting for reporting
+            cache_files = [f for f in os.listdir(cache_folder) if os.path.isfile(os.path.join(cache_folder, f))]
+            deleted_files = []
+            
+            # Delete each .msgpack file in the cache folder
+            for filename in cache_files:
+                if filename.endswith('.msgpack'):
+                    file_path = os.path.join(cache_folder, filename)
+                    try:
+                        os.remove(file_path)
+                        deleted_files.append(filename)
+                        logger.info(f"Deleted cache file: {filename}")
+                    except Exception as e:
+                        logger.error(f"Failed to delete {filename}: {e}")
+                        return web.json_response({
+                            'success': False,
+                            'error': f"Failed to delete {filename}: {str(e)}"
+                        }, status=500)
+            
+            # If we want to completely remove the cache folder too (optional, 
+            # but we'll keep the folder structure in place here)
+            # shutil.rmtree(cache_folder)
+            
+            return web.json_response({
+                'success': True,
+                'message': f"Successfully cleared {len(deleted_files)} cache files",
+                'deleted_files': deleted_files
+            })
+            
+        except Exception as e:
+            logger.error(f"Error clearing cache files: {e}", exc_info=True)
+            return web.json_response({
+                'success': False,
+                'error': str(e)
+            }, status=500)
+
     @staticmethod
     async def update_settings(request):
         """Update application settings"""

py/services/model_scanner.py

@@ -103,7 +103,7 @@ class ModelScanner:
             else:
                 os.rename(temp_path, cache_path)
                 
-            logger.info(f"Saved {self.model_type} cache with {len(self._cache.raw_data)} models to {cache_path}")
+            logger.debug(f"Saved {self.model_type} cache with {len(self._cache.raw_data)} models to {cache_path}")
             return True
         except Exception as e:
             logger.error(f"Error saving {self.model_type} cache to disk: {e}")

static/js/managers/ModalManager.js

@@ -170,6 +170,18 @@ export class ModalManager {
             });
         }
         
+        // Add clearCacheModal registration
+        const clearCacheModal = document.getElementById('clearCacheModal');
+        if (clearCacheModal) {
+            this.registerModal('clearCacheModal', {
+                element: clearCacheModal,
+                onClose: () => {
+                    this.getModal('clearCacheModal').element.classList.remove('show');
+                    document.body.classList.remove('modal-open');
+                }
+            });
+        }
+        
         // Set up event listeners for modal toggles
         const supportToggle = document.getElementById('supportToggleBtn');
         if (supportToggle) {
@@ -233,7 +245,7 @@ export class ModalManager {
         // Store current scroll position before showing modal
         this.scrollPosition = window.scrollY;
 
-        if (id === 'deleteModal' || id === 'excludeModal' || id === 'duplicateDeleteModal') {
+        if (id === 'deleteModal' || id === 'excludeModal' || id === 'duplicateDeleteModal' || id === 'clearCacheModal') {
             modal.element.classList.add('show');
         } else {
             modal.element.style.display = 'block';

static/js/managers/SettingsManager.js

@@ -341,6 +341,37 @@ export class SettingsManager {
         }
     }
 
+    confirmClearCache() {
+        // Show confirmation modal
+        modalManager.showModal('clearCacheModal');
+    }
+
+    async executeClearCache() {
+        try {
+            // Call the API endpoint to clear cache files
+            const response = await fetch('/api/clear-cache', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                }
+            });
+
+            const result = await response.json();
+            
+            if (result.success) {
+                showToast('Cache files have been cleared successfully. Cache will rebuild on next action.', 'success');
+            } else {
+                showToast(`Failed to clear cache: ${result.error}`, 'error');
+            }
+            
+            // Close the confirmation modal
+            modalManager.closeModal('clearCacheModal');
+        } catch (error) {
+            showToast(`Error clearing cache: ${error.message}`, 'error');
+            modalManager.closeModal('clearCacheModal');
+        }
+    }
+
     async reloadContent() {
         if (this.currentPage === 'loras') {
             // Reload the loras without updating folders

templates/components/modals.html

@@ -39,6 +39,21 @@
     </div>
 </div>
 
+<!-- Cache Clear Confirmation Modal -->
+<div id="clearCacheModal" class="modal delete-modal">
+    <div class="modal-content delete-modal-content">
+        <h2>Clear Cache Files</h2>
+        <p class="delete-message">Are you sure you want to clear all cache files?</p>
+        <div class="delete-model-info">
+            <p>This will remove all cached model data. The system will need to rebuild the cache on next startup, which may take some time depending on your model collection size.</p>
+        </div>
+        <div class="modal-actions">
+            <button class="cancel-btn" onclick="modalManager.closeModal('clearCacheModal')">Cancel</button>
+            <button class="delete-btn" onclick="settingsManager.executeClearCache()">Clear Cache</button>
+        </div>
+    </div>
+</div>
+
 <!-- Settings Modal -->
 <div id="settingsModal" class="modal">
     <div class="modal-content settings-modal">
@@ -183,6 +198,29 @@
                 </div>
             </div>
             
+            <!-- Add Cache Management Section -->
+            <div class="settings-section">
+                <h3>Cache Management</h3>
+                
+                <div class="setting-item">
+                    <div class="setting-row">
+                        <div class="setting-info">
+                            <label>Clear Cache Files</label>
+                        </div>
+                        <div class="setting-control">
+                            <button id="clearCacheBtn" class="secondary-btn" onclick="settingsManager.confirmClearCache()">
+                                <i class="fas fa-trash"></i>
+                                <span>Clear Cache</span>
+                            </button>
+                        </div>
+                    </div>
+                    <div class="input-help">
+                        Removes cached model data. System will rebuild cache on next startup or when triggered manually.
+                        <span class="warning-text">May cause temporary performance impact during rebuild.</span>
+                    </div>
+                </div>
+            </div>
+            
             <!-- Add Example Images Settings Section -->
             <div class="settings-section">
                 <h3>Example Images</h3>