Update worker.js

worker.js

@@ -17,25 +17,39 @@ let extractor = null;
 let generator = null;
 let vectorStore = []; // Lưu trữ chunks và vectors: { text: string, vector: number[] }
 
-// Khởi tạo Models
 async function initModels() {
     try {
-        console.log("Đang tải Embedding Model...");
+        self.postMessage({ type: 'status', payload: "Đang tải Embedding Model (có thể lâu lần đầu)..." });
+        
+        // Thêm cờ { use_cache: true } dù mặc định nó đã có, để đảm bảo
         extractor = await pipeline('feature-extraction', EMBEDDING_MODEL_ID, {
             device: 'webgpu',
-            dtype: 'fp32', // Embedding thường cần độ chính xác
+            dtype: 'fp32',
+            use_cache: true 
         });
 
-        console.log("Đang tải LLM Granite 4.0...");
+        self.postMessage({ type: 'status', payload: "Đang tải LLM Granite 4.0 (Model nặng, vui lòng chờ)..." });
+        
         generator = await pipeline('text-generation', LLM_MODEL_ID, {
             device: 'webgpu',
-            dtype: 'q4', // Quantization 4-bit để chạy mượt trên browser
-            use_external_data_format: true 
+            dtype: 'q4',
+            use_external_data_format: true,
+            // Thêm progress_callback để UI không tưởng là bị treo
+            progress_callback: (data) => {
+                if (data.status === 'progress') {
+                    // Gửi tiến độ về main thread để hiện loading bar nếu cần
+                    self.postMessage({ 
+                        type: 'download_progress', 
+                        payload: { file: data.file, progress: data.progress } 
+                    });
+                }
+            }
         });
 
         self.postMessage({ type: 'init_complete' });
     } catch (e) {
-        self.postMessage({ type: 'error', payload: "Lỗi tải model: " + e.message });
+        console.error(e); // Log chi tiết ra console
+        self.postMessage({ type: 'error', payload: "Lỗi tải model (Kiểm tra Console F12): " + e.message });
     }
 }