Update progress_callback.js

progress_callback.js

@@ -1,49 +1,49 @@
-import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@huggingface/transformers@3.0.0';
-
-env.allowRemoteModels = true;
-env.allowLocalModels = false;
-env.remotePathTemplate = 'https://huggingface.co/{model}/resolve/main/';
-env.backends.onnx.wasm.numThreads = 1;
-env.backends.onnx.wasm.proxy = false;
-
-const status = document.getElementById('status');
-const btn = document.getElementById('main-btn');
-const input = document.getElementById('chat-input');
-let generator = null;
-
-async function init() {
-    try {
-        btn.disabled = true;
-        status.textContent = "Loading Tiny Brain (Single-Thread)...";
-
-        generator = await pipeline(
-            'text-generation',
-            'Xenova/phi-1_5-tiny-onnx',   // replace with exact repo name
-            {
-                device: 'wasm',
-                dtype: 'q4',
-                progress_callback: (d) => {
-                    if (d.status === 'progress') {
-                        status.textContent = `Downloading: ${Math.round(d.progress)}%`;
-                    }
-                }
-            }
-        );
-
         status.textContent = "Ready! RAM usage minimized.";
         btn.textContent = "Send";
         input.disabled = false;
-
-        // Change button behavior: now it sends chat
         btn.disabled = false;
+
         btn.onclick = async () => {
-            const userText = input.value;
-            if (!userText) return;
+            const userText = input.value.trim();
+            if (!userText || btn.disabled) return;
+
+            // UI Updates
+            input.value = '';
+            btn.disabled = true;
+            status.textContent = "Thinking...";
+            
             addMessage('user', userText);
-            const output = await generator(userText, { max_new_tokens: 50 });
-            addMessage('ai', output[0].generated_text);
+
+            try {
+                const output = await generator(userText, { 
+                    max_new_tokens: 50,
+                    temperature: 0.7,
+                    do_sample: true 
+                });
+                
+                addMessage('ai', output[0].generated_text);
+            } catch (err) {
+                console.error("Generation Error:", err);
+                status.textContent = "Error during generation.";
+            } finally {
+                status.textContent = "Ready!";
+                btn.disabled = false;
+            }
         };
     } catch (e) {
-        console.error(e);
-        status.innerHTML = `
-           
+        console.error("Initialization Error:", e);
+        status.innerHTML = `<span style="color: red;">Failed to load model: ${e.message}</span>`;
+    }
+}
+
+// Helper function to append messages to your chat container
+function addMessage(sender, text) {
+    const container = document.getElementById('chat-container'); // Ensure this ID exists in your HTML
+    const msgDiv = document.createElement('div');
+    msgDiv.className = `message ${sender}-message`;
+    msgDiv.textContent = text;
+    container.appendChild(msgDiv);
+    container.scrollTop = container.scrollHeight;
+}
+
+init();