Refine cluster name extraction in K-Means implementation to handle special tokens and improve naming accuracy

main.js

@@ -166,7 +166,11 @@ document.getElementById("kmeans-btn").onclick = async () => {
             do_sample: false,
             streamer,
         });
-        let name = tokenizer.decode(outputTokens[0], { skip_special_tokens: false }).trim();
+        let rawName = tokenizer.decode(outputTokens[0], { skip_special_tokens: false }).trim();
+        // Extract cluster name after last '</think>' if present, and remove '<|im_end|>' if present
+        let idx = rawName.lastIndexOf('</think>');
+        let name = idx !== -1 ? rawName.slice(idx + 8).trim() : rawName;
+        if (name.endsWith('<|im_end|>')) name = name.slice(0, -11).trim();
         clusterNames.push(name.length > 0 ? name : `Cluster ${c + 1}`);
     }
     // After all names are generated, update the trace names and render once