Add download cancellation option for models

Author: one-data-cookie

assets/js/chatmkAI.js

@@ -8,6 +8,7 @@ class ChatMKAI {
     this.wllama = null;
     this.isLoading = false;
     this.isLoaded = false;
+    this.downloadController = null; // For cancelling downloads
     this.modelPath =
       "https://huggingface.co/QuantFactory/SmolLM2-360M-Instruct-GGUF/resolve/main/SmolLM2-360M-Instruct.Q8_0.gguf";
   }
@@ -21,6 +22,7 @@ class ChatMKAI {
     }
 
     this.isLoading = true;
+    this.downloadController = new AbortController(); // Create abort controller
     console.log("ChatMK AI: Starting model initialization...");
 
     try {
@@ -61,6 +63,11 @@ class ChatMKAI {
         n_ctx: 2048, // Set context size during model loading
         n_ctx_per_seq: 2048, // Set per-sequence context during model loading
         progressCallback: ({ loaded, total }) => {
+          // Check if cancelled
+          if (this.downloadController?.signal.aborted) {
+            throw new Error('Download cancelled by user');
+          }
+
           const percent = Math.round((loaded / total) * 100);
           const mbLoaded = Math.round(loaded / 1024 / 1024);
           const mbTotal = Math.round(total / 1024 / 1024);
@@ -79,22 +86,36 @@ class ChatMKAI {
 
       this.isLoaded = true;
       this.isLoading = false;
+      this.downloadController = null;
       console.log("ChatMK AI: Model loaded successfully!");
 
-      // Update UI to show AI is ready
-      this.updateLoadingStatus("Language model ready!");
-
       return true;
     } catch (error) {
-      console.error("ChatMK AI: Failed to initialize model:", error);
       this.isLoading = false;
+      this.downloadController = null;
 
-      // For now, disable AI if it fails to load
-      this.updateLoadingStatus("Language model unavailable - using search only");
+      if (error.message.includes('cancelled')) {
+        console.log("ChatMK AI: Download cancelled by user");
+        this.showProgressInModal(0, 0, 150, 'Language model skipped → using search only');
+        return false;
+      }
+
+      console.error("ChatMK AI: Failed to initialize model:", error);
+      this.showProgressInModal(0, 0, 150, 'Language model failed → using search only');
       return false;
     }
   }
 
+  /**
+   * Cancel the download
+   */
+  cancelDownload() {
+    if (this.downloadController) {
+      this.downloadController.abort();
+      console.log("ChatMK AI: Download cancelled");
+    }
+  }
+
   /**
    * Generate AI response (without search results - those are shown separately)
    */
@@ -194,26 +215,20 @@ ${userMessage}
     return cleaned.trim();
   }
 
-  /**
-   * Update loading status in the UI
-   */
-  updateLoadingStatus(message) {
-    // This will be called from the modal to update status
-    if (typeof window.updateAIStatus === "function") {
-      window.updateAIStatus(message);
-    }
-  }
+
 
   /**
    * Show progress in ChatMK modal if it's open
    */
-  showProgressInModal(percent, mbLoaded, mbTotal) {
+  showProgressInModal(percent, mbLoaded, mbTotal, customMessage = null) {
     const modal = document.getElementById("chatmk-modal");
     if (modal && modal.classList.contains("is-active")) {
       // Add a system message about AI loading progress
-      const progressMsg = percent === 100 
+      const progressMsg = customMessage 
+        ? customMessage
+        : percent === 100 
         ? `Language model loaded: SmolLM2-360M-Q8`
-        : `Language model loading: ${percent}% (${mbLoaded}/${mbTotal} MB)`;
+        : `Language model loading: ${percent}% (${mbLoaded}/${mbTotal} MB) <span onclick="window.chatMKAI.cancelDownload()" style="margin-left: 10px; font-size: 11px; color: var(--text-sub); cursor: pointer; text-decoration: underline;">skip magic</span>`;
 
       // Check if there's already a progress message and update it
       const messages = document.getElementById("chatmk-messages");
@@ -252,6 +267,19 @@ ${userMessage}
     }
   }
 
+  /**
+   * Hide AI progress message
+   */
+  hideProgressMessage() {
+    const messagesContainer = document.getElementById("chatmk-messages");
+    if (messagesContainer) {
+      const progressMessage = messagesContainer.querySelector(".ai-progress");
+      if (progressMessage) {
+        progressMessage.remove();
+      }
+    }
+  }
+
   /**
    * Check if AI is ready
    */

assets/js/chatmkSearch.js

@@ -9,6 +9,7 @@ class ChatMKSearch {
     this.isLoaded = false;
     this.model = null;
     this.transformersLoaded = false;
+    this.embeddingController = null; // For cancelling embedding downloads
   }
 
   /**
@@ -72,7 +73,7 @@ class ChatMKSearch {
   /**
    * Show embedding model loading progress
    */
-  showEmbeddingProgress(percent, mbLoaded, mbTotal) {
+  showEmbeddingProgress(percent, mbLoaded, mbTotal, customMessage = null) {
     const messagesContainer = document.getElementById('chatmk-messages');
     if (messagesContainer) {
       // Check if there's already a progress message and update it
@@ -82,19 +83,23 @@ class ChatMKSearch {
         // Update existing progress message
         const content = existingProgress.querySelector('.message-content');
         if (content) {
-          if (percent === 100) {
+          if (customMessage) {
+            content.innerHTML = `<p>${customMessage}</p>`;
+          } else if (percent === 100) {
             content.innerHTML = `<p>Embedding model loaded: MiniLM-L6-v2</p>`;
           } else {
-            content.innerHTML = `<p>Embedding model loading: ${percent}% (${mbLoaded}/${mbTotal} MB)</p>`;
+            content.innerHTML = `<p>Embedding model loading: ${percent}% (${mbLoaded}/${mbTotal} MB) <span onclick="window.chatMKSearch.cancelEmbeddingDownload()" style="margin-left: 10px; font-size: 11px; color: var(--text-sub); cursor: pointer; text-decoration: underline;">skip magic</span></p>`;
           }
         }
       } else {
         // Create new progress message
         const messageDiv = document.createElement('div');
         messageDiv.className = 'chatmk-message system embedding-loading';
-        const messageContent = percent === 100 
+        const messageContent = customMessage 
+          ? `<p>${customMessage}</p>`
+          : percent === 100 
           ? `<p>Embedding model loaded: MiniLM-L6-v2</p>`
-          : `<p>Embedding model loading: ${percent}% (${mbLoaded}/${mbTotal} MB)</p>`;
+          : `<p>Embedding model loading: ${percent}% (${mbLoaded}/${mbTotal} MB) <span onclick="window.chatMKSearch.cancelEmbeddingDownload()" style="margin-left: 10px; font-size: 11px; color: var(--text-sub); cursor: pointer; text-decoration: underline;">skip magic</span></p>`;
 
         messageDiv.innerHTML = `
           <div class="message-content">
@@ -108,6 +113,16 @@ class ChatMKSearch {
     }
   }
 
+  /**
+   * Cancel the embedding download
+   */
+  cancelEmbeddingDownload() {
+    if (this.embeddingController) {
+      this.embeddingController.abort();
+      console.log("Embedding download cancelled");
+    }
+  }
+
   /**
    * Simulate embedding model loading progress
    */
@@ -116,6 +131,11 @@ class ChatMKSearch {
 
     // Simulate progress from 0% to 100%
     for (let percent = 0; percent <= 100; percent += 20) {
+      // Check if cancelled
+      if (this.embeddingController?.signal.aborted) {
+        throw new Error('Embedding download cancelled by user');
+      }
+      
       const mbLoaded = Math.round((percent / 100) * totalMB);
       this.showEmbeddingProgress(percent, mbLoaded, totalMB);
 
@@ -134,6 +154,7 @@ class ChatMKSearch {
       return;
     }
 
+    this.embeddingController = new AbortController(); // Create abort controller
     console.log('ChatMK: Pre-loading embedding model...');
 
     try {
@@ -160,9 +181,19 @@ class ChatMKSearch {
       // Keep showing 100% when done (like AI model)
       this.showEmbeddingProgress(100, 25, 25);
 
+      this.embeddingController = null;
+      
     } catch (error) {
+      this.embeddingController = null;
+      
+      if (error.message.includes('cancelled')) {
+        console.log('ChatMK: Embedding download cancelled by user');
+        this.showEmbeddingProgress(0, 0, 25, 'Embedding model skipped → this won\'t work');
+        return false;
+      }
+      
       console.error('ChatMK: Failed to pre-load embedding model:', error);
-      this.hideModelLoadingMessage();
+      this.showEmbeddingProgress(0, 0, 25, 'Embedding model failed → this won\'t work');
       // Don't throw - allow fallback to keyword search
     }
   }
@@ -378,3 +409,6 @@ class ChatMKSearch {
 
 // Export for use in other modules
 window.ChatMKSearch = ChatMKSearch;
+
+// Global instance
+window.chatMKSearch = new ChatMKSearch();