ai: Bump script version.

README.md

@@ -56,4 +56,9 @@ models:
 - zai-org/GLM-4.5
 - zai-org/GLM-4.5-Air
 - zai-org/GLM-4.5V
+- deepseek-ai/DeepSeek-V3.1
+- deepseek-ai/DeepSeek-V3.1-Base
+# Used to promote this Hugging Face Space
+datasets:
+- fka/awesome-chatgpt-prompts
 ---

assets/css/chat/markdown.css

@@ -159,6 +159,5 @@
 }
 
 .md-content table td pre, .md-content table th pre {
-  max-width: 480px;
   overflow: auto;
 }

assets/plugins/loader.js

@@ -6,8 +6,10 @@
 // Prism.
 Prism.plugins.autoloader.languages_path = 'https://cdnjs.cloudflare.com/ajax/libs/prism/1.29.0/components/';
 
-// WebSocket.
-const socket = new WebSocket((window.location.protocol==="https:"?"wss":"ws")+"//"+window.location.host);
+// WebSocket URL helper.
+function createWebSocket() {
+  return new WebSocket((window.location.protocol === "https:" ? "wss" : "ws") + "//" + window.location.host);
+}
 
 // UI elements.
 const chatArea = document.getElementById('chatArea');
@@ -24,9 +26,12 @@ const homeBtn = document.getElementById('homeBtn');
 const clearBtn = document.getElementById('clearBtn');
 
 // Track state.
+let socket = null;
 let streamMsg = null;
 let conversationHistory = [];
 let currentAssistantText = "";
+let isRequestActive = false;
+let abortController = null;
 
 // Render markdown content.
 function renderMarkdown(el) {
@@ -84,23 +89,23 @@ function leaveChatView() {
 // Chat bubble.
 function addMsg(who, text) {
   const div = document.createElement('div');
-  div.className = 'bubble ' + (who==='user' ? 'bubble-user' : 'bubble-assist');
+  div.className = 'bubble ' + (who === 'user' ? 'bubble-user' : 'bubble-assist');
   div.dataset.text = text;
   renderMarkdown(div);
   chatBox.appendChild(div);
-  chatBox.style.display='flex';
+  chatBox.style.display = 'flex';
   chatBox.scrollTop = chatBox.scrollHeight;
   return div;
 }
 
 // Clear all chat.
 function clearAllMessages() {
-  try { socket.send(JSON.stringify({type:'stop'})); } catch {}
+  stopStream(true);
   conversationHistory = [];
   currentAssistantText = "";
-  if(streamMsg) {
+  if (streamMsg) {
     const loadingEl = streamMsg.querySelector('.loading');
-    if(loadingEl) loadingEl.remove();
+    if (loadingEl) loadingEl.remove();
     streamMsg = null;
   }
   chatBox.innerHTML = "";
@@ -111,57 +116,164 @@ function clearAllMessages() {
   enterChatView();
 }
 
-// Wait for socket ready.
-function sendWhenReady(msgFn) {
-  if (socket.readyState === WebSocket.OPEN) {
-    msgFn();
-  } else {
-    socket.addEventListener('open', function handler() {
-      msgFn();
-      socket.removeEventListener('open', handler);
-    });
+// Reconnect WebSocket.
+let reconnectAttempts = 0;
+function setupWebSocket() {
+  if (socket) {
+    socket.onopen = null;
+    socket.onclose = null;
+    socket.onmessage = null;
+    socket.onerror = null;
+    socket.close();
+    socket = null;
   }
+
+  socket = createWebSocket();
+
+  socket.onopen = () => {
+    reconnectAttempts = 0;
+  };
+
+  socket.onclose = () => {
+    reconnectAttempts++;
+    // Try reconnecting.
+    const delay = Math.min(30000, 1000 * Math.pow(2, reconnectAttempts));  // 30 seconds.
+    setTimeout(setupWebSocket, delay);
+  };
+
+  socket.onmessage = handleSocketMessage;
+
+  socket.onerror = () => {
+    socket.close();
+  };
 }
 
-// Prompts.
-promptItems.forEach(p => {
-  p.addEventListener('click', () => {
-    input.value = p.dataset.prompt;
-    sendWhenReady(submitMessage);
-  });
-});
+// Handle incoming socket messages.
+function handleSocketMessage(e) {
+  const data = JSON.parse(e.data);
+  if (data.type === 'chunk') {
+    if (streamMsg) {
+      const loadingEl = streamMsg.querySelector('.loading');
+      if (loadingEl) loadingEl.remove();
+      streamMsg.dataset.text += data.chunk;
+      currentAssistantText = streamMsg.dataset.text || "";
+      renderMarkdown(streamMsg);
+      chatBox.scrollTop = chatBox.scrollHeight;
+    }
+  } else if (data.type === 'end' || data.type === 'error') {
+    if (streamMsg) {
+      const loadingEl = streamMsg.querySelector('.loading');
+      if (loadingEl) loadingEl.remove();
+      streamMsg.dataset.done = '1';
+      if (data.type === 'error') {
+        streamMsg.dataset.text = data.error || 'An error occurred during the request.';
+        renderMarkdown(streamMsg);
+      } else {
+        conversationHistory.push({ role: 'assistant', content: streamMsg.dataset.text });
+      }
+      streamMsg = null;
+      isRequestActive = false;
+      abortController = null;
+    }
+    btn.style.display = 'inline-flex';
+    stopBtn.style.display = 'none';
+    stopBtn.style.pointerEvents = 'auto';
+  }
+}
 
 // Send user message.
 async function submitMessage() {
   const message = input.value.trim();
-  if(!message) return;
+  if (!message || isRequestActive) return;
   enterChatView();
   addMsg('user', message);
-  conversationHistory.push({role: 'user', content: message});
+  conversationHistory.push({ role: 'user', content: message });
   streamMsg = addMsg('assistant', '');
   const loadingEl = document.createElement('span');
   loadingEl.className = 'loading';
   streamMsg.appendChild(loadingEl);
   stopBtn.style.display = 'inline-flex';
   btn.style.display = 'none';
-  input.value='';
+  input.value = '';
   btn.disabled = true;
+
+  isRequestActive = true;
+
+  // Stopping request.
+  abortController = new AbortController();
   try {
-    socket.send(JSON.stringify({type:'ask', message, history: conversationHistory}));
+    socket.send(JSON.stringify({
+      type: 'ask',
+      message,
+      history: conversationHistory,
+      abortSignal: true
+    }));
   } catch (error) {
-    if(streamMsg){ 
+    if (streamMsg) {
       const loadingEl = streamMsg.querySelector('.loading');
-      if(loadingEl) loadingEl.remove();
+      if (loadingEl) loadingEl.remove();
       streamMsg.dataset.text = error.message || 'An error occurred during the request.';
       renderMarkdown(streamMsg);
-      streamMsg.dataset.done='1'; 
-      streamMsg=null; 
+      streamMsg.dataset.done = '1';
+      streamMsg = null;
+      isRequestActive = false;
+      abortController = null;
     }
     btn.style.display = 'inline-flex';
     stopBtn.style.display = 'none';
   }
 }
 
+// Stop streaming and cancel the ongoing request.
+function stopStream(forceCancel = false) {
+  if (!isRequestActive) return;
+
+  isRequestActive = false;
+
+  if (abortController) {
+    abortController.abort();
+    abortController = null;
+  }
+
+  // Notify server to stop sending streams / processing.
+  try {
+    socket.send(JSON.stringify({ type: 'stop' }));
+  } catch {}
+
+  if (streamMsg && !forceCancel) {
+    const loadingEl = streamMsg.querySelector('.loading');
+    if (loadingEl) loadingEl.remove();
+    streamMsg.dataset.text += '';
+    renderMarkdown(streamMsg);
+    streamMsg.dataset.done = '1';
+    streamMsg = null;
+  }
+
+  stopBtn.style.display = 'none';
+  btn.style.display = 'inline-flex';
+  stopBtn.style.pointerEvents = 'auto';
+}
+
+// Wait for socket ready.
+function sendWhenReady(msgFn) {
+  if (socket.readyState === WebSocket.OPEN) {
+    msgFn();
+  } else {
+    socket.addEventListener('open', function handler() {
+      msgFn();
+      socket.removeEventListener('open', handler);
+    });
+  }
+}
+
+// Prompts.
+promptItems.forEach(p => {
+  p.addEventListener('click', () => {
+    input.value = p.dataset.prompt;
+    sendWhenReady(submitMessage);
+  });
+});
+
 // Submit.
 form.addEventListener('submit', e => {
   e.preventDefault();
@@ -171,7 +283,7 @@ form.addEventListener('submit', e => {
 // Stop.
 stopBtn.addEventListener('click', () => {
   stopBtn.style.pointerEvents = 'none';
-  try { socket.send(JSON.stringify({type:'stop'})); } catch {}
+  stopStream();
 });
 
 // Home.
@@ -184,49 +296,20 @@ clearBtn.addEventListener('click', () => {
   clearAllMessages();
 });
 
-// Socket messages.
-socket.onmessage = (e) => {
-  const data = JSON.parse(e.data);
-  if(data.type==='chunk') {
-    if(streamMsg) {
-      const loadingEl = streamMsg.querySelector('.loading');
-      if(loadingEl) loadingEl.remove();
-      streamMsg.dataset.text += data.chunk;
-      currentAssistantText = streamMsg.dataset.text ||"";
-      renderMarkdown(streamMsg);
-      chatBox.scrollTop = chatBox.scrollHeight;
-    }
-  } else if(data.type==='end' || data.type==='error') {
-    if(streamMsg){
-      const text = streamMsg.dataset.text || "";
-      const loadingEl = streamMsg.querySelector('.loading');
-      if(loadingEl) loadingEl.remove();
-      streamMsg.dataset.done='1';
-      if(data.type==='error') {
-        streamMsg.dataset.text = data.error || 'An error occurred during the request.';
-        renderMarkdown(streamMsg);
-      } else {
-        conversationHistory.push({role: 'assistant', content: text});
-      }
-      streamMsg = null;
-    }
-    btn.style.display = 'inline-flex';
-    stopBtn.style.display = 'none';
-    stopBtn.style.pointerEvents = 'auto';
-  }
-};
-
 // Enable send button only if input has text.
 input.addEventListener('input', () => {
-  btn.disabled= input.value.trim() === '';
+  btn.disabled = input.value.trim() === '';
 });
 
 // Animations.
-document.addEventListener('DOMContentLoaded', function() {
+document.addEventListener('DOMContentLoaded', function () {
   AOS.init({
     duration: 800,
     easing: 'ease-out-cubic',
     once: true,
     offset: 50
   });
-});
+});
+
+// Initialize WebSocket connection.
+setupWebSocket();

package.json

@@ -1,6 +1,6 @@
 {
   "name": "UltimaX Intelligence",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "type": "module",
   "main": "mapping.js",
   "scripts": {