Add prompt cluster logic for summarizing multiple sentences into concise phrases

main.js

@@ -1,6 +1,6 @@
 import { pipeline, TextStreamer, AutoTokenizer, AutoModelForCausalLM } from 'https://cdn.jsdelivr.net/npm/@huggingface/transformers@3.6.0';
 import { UMAP } from "https://cdn.jsdelivr.net/npm/umap-js@1.4.0/+esm";
-
+import { prompt_cluster } from "./prompt_cluster.js"
 const embed = await pipeline(
     "feature-extraction",
     "onnx-community/Qwen3-Embedding-0.6B-ONNX",
@@ -121,8 +121,10 @@ document.getElementById("kmeans-btn").onclick = async () => {
         progressBarInner.style.width = `${Math.round(((c) / k) * 100)}%`;
         const joined = clustered[c].join("\n");
         const messages = [
-            { role: "system", content: "Given the following texts, provide a short 1-3 word summary in plaintext" },
-            { role: "user", content: `${joined}` }
+            {
+                role: "system", content: prompt_cluster
+            },
+            { role: "user", content: `Input:\n${joined}\nOutput:` }
         ];
 
         const inputs = tokenizer.apply_chat_template(messages, {

prompt_cluster.js

@@ -0,0 +1,32 @@
+export const prompt_cluster = `Given the following texts, Summarize multiple sentences into a single 1-3 word phrase plain text.
+
+Example 1
+Input:
+The new smartphone features a foldable display and 5G connectivity.
+Virtual reality headsets are becoming more affordable and accessible.
+Output: Next Generation Devices
+
+Example 2
+Input:
+Athletes are breaking records in international competitions.
+Sports streaming is reaching global audiences in real time.
+Output: Global Competitive Sports
+
+Example 3
+Input:
+In the world of AI, transformers have revolutionized natural language processing.
+Quantum computing promises to solve problems beyond classical computers' reach.
+Output: Intelligent Computational Systems
+
+Example 4
+Input:
+Blockchain technology is being explored for secure voting systems.
+Quantum computing promises to solve problems beyond classical computers' reach.
+In the world of AI, transformers have revolutionized natural language processing.
+Output: Emerging Digital Technologies
+
+Example 5
+Input:
+Homemade meals are becoming more popular due to health consciousness.
+Cooking tutorials are widely available on social media platforms.
+Output: Modern Home Cooking`