TokenVisualizer / index.html
PeterPinetree's picture
Update index.html
c994801 verified
raw
history blame
9.81 kB
<!doctype html>
<html lang="en">
<head>
<meta charset="utf-8" />
<meta
name="viewport"
content="width=device-width, initial-scale=1"
/>
<title>TokenVisualizer — Two-Pane</title>
<link rel="preconnect" href="https://fonts.googleapis.com">
<link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;600&family=JetBrains+Mono:wght@400;600&display=swap" rel="stylesheet">
<style>
:root{
--bg:#0b0f14; /* black-leaning bg */
--text:#ffffff; /* white */
--muted:#9aa4b2;
--accent:#38bdf8; /* sky blue */
--card:#0e1624;
--border:#1f2a3a;
--chip:#111827;
--chip-border:#263246;
--chip-hover:#1a2434;
--mono:'JetBrains Mono', ui-monospace, SFMono-Regular, Menlo, Consolas, monospace;
--sans:Inter, system-ui, -apple-system, Segoe UI, Roboto, Ubuntu, Cantarell, Noto Sans, 'Helvetica Neue', Arial;
}
*{box-sizing:border-box}
body{margin:0;background:radial-gradient(900px 500px at 10% -10%, #07314a, transparent),var(--bg);color:var(--text);font-family:var(--sans)}
.container{max-width:1100px;margin:0 auto;padding:1.25rem}
header{padding-top:1.5rem}
h1{margin:.2rem 0 .4rem;font-size:1.9rem}
.sub{color:var(--muted);margin:.25rem 0 1rem}
.card{background:linear-gradient(180deg,#0c1624,#0a1220);border:1px solid var(--border);border-radius:14px;padding:1rem;box-shadow:0 10px 40px rgba(0,0,0,.35)}
label span{color:var(--muted);font-size:.9rem}
select,textarea{width:100%;border-radius:10px;border:1px solid var(--border);background:#0a1220;color:var(--text);padding:.7rem .85rem;outline:none}
select:focus,textarea:focus{border-color:var(--accent)}
.controls{display:grid;gap:.8rem;margin-bottom:1rem}
.row{display:flex;gap:.75rem;align-items:center}
.btn{background:var(--accent);color:#07222d;border:0;border-radius:10px;padding:.55rem .95rem;font-weight:600;cursor:pointer}
.btn.secondary{background:#152236;color:var(--text);border:1px solid var(--border)}
.status{color:var(--muted)}
.grid{display:grid;gap:1rem;grid-template-columns:1fr}
@media (min-width:900px){.grid{grid-template-columns:1fr 1fr}}
.head{display:flex;align-items:center;justify-content:space-between;margin-bottom:.5rem}
.actions .link{background:none;border:none;color:var(--accent);cursor:pointer;margin-left:.5rem}
.tokens{display:flex;flex-wrap:wrap;gap:.5rem;max-height:360px;overflow:auto;padding:.25rem}
.chip{border:1px solid var(--chip-border);background:var(--chip);padding:.35rem .5rem;border-radius:10px;font-family:var(--mono);font-size:.9rem;transition:background .12s,border-color .12s}
.chip:hover{background:var(--chip-hover);border-color:var(--accent)}
.chip.active{outline:2px solid var(--accent)}
pre.ids{font-family:var(--mono);background:#0a1220;border:1px solid var(--border);border-radius:10px;padding:.75rem;max-height:360px;overflow:auto;white-space:pre-wrap}
footer{color:var(--muted);text-align:center;padding:1.25rem 0 2rem}
a{color:var(--accent)}
</style>
</head>
<body>
<header class="container">
<h1>TokenVisualizer</h1>
<p class="sub">See tokens and token IDs at the same time. (Runs entirely in your browser.)</p>
</header>
<main class="container">
<section class="card controls">
<label>
<span>Model</span>
<select id="model">
<option value="Xenova/gpt2">GPT-2 (BPE)</option>
<option value="Xenova/llama2-tokenizer">Llama-2 (SentencePiece/BPE)</option>
<option value="Xenova/mistral-tokenizer">Mistral (SentencePiece/BPE)</option>
<option value="Xenova/gemma-tokenizer">Gemma (SentencePiece/BPE)</option>
<option value="Xenova/bert-base-uncased">BERT Base Uncased (WordPiece)</option>
</select>
</label>
<label>
<span>Text</span>
<textarea id="input" rows="3">Hello world! This is a tokenizer demo.</textarea>
</label>
<div class="row">
<button id="tokenize" class="btn">Tokenize</button>
<span id="status" class="status">Loading tokenizer…</span>
</div>
</section>
<section class="grid">
<article class="card">
<div class="head">
<h3>Tokens</h3>
<div class="actions">
<button id="copyTokens" class="link">Copy</button>
<button id="exportTokens" class="link">Export JSON</button>
</div>
</div>
<div id="tokens" class="tokens"></div>
</article>
<article class="card">
<div class="head">
<h3>Token IDs</h3>
<div class="actions">
<button id="copyIds" class="link">Copy</button>
<button id="exportIds" class="link">Export JSON</button>
</div>
</div>
<pre id="ids" class="ids"></pre>
</article>
</section>
<section class="container" style="text-align:right;margin:1rem 0">
<button id="exportCSV" class="btn secondary">Download CSV (index, token, id)</button>
</section>
</main>
<footer class="container">
<small>Built by Peter Adams • Powered by <a href="https://github.com/xenova/transformers.js" target="_blank" rel="noreferrer">Transformers.js</a></small>
</footer>
<!-- Transformers.js (ESM) -->
<script type="module">
// Load transformers.js from CDN
const tf = await import('https://cdn.jsdelivr.net/npm/@xenova/[email protected]/dist/transformers.min.mjs');
tf.env.useBrowserCache = true;
tf.env.allowLocalModels = false;
const $ = sel => document.querySelector(sel);
const el = {
model: $('#model'),
input: $('#input'),
btn: $('#tokenize'),
status: $('#status'),
tokens: $('#tokens'),
ids: $('#ids'),
copyTokens: $('#copyTokens'),
exportTokens: $('#exportTokens'),
copyIds: $('#copyIds'),
exportIds: $('#exportIds'),
exportCSV: $('#exportCSV'),
};
let tokenizer = null;
let last = { tokens: [], ids: [] };
let runId = 0;
function status(msg){ el.status.textContent = msg; }
function debounce(fn, ms=250){ let t; return (...a)=>{ clearTimeout(t); t=setTimeout(()=>fn(...a), ms); }; }
async function loadTokenizer(modelId){
status('Loading tokenizer…');
tokenizer = await tf.AutoTokenizer.from_pretrained(modelId);
status('Tokenizer ready.');
}
async function tokenize(){
const myRun = ++runId;
if (!tokenizer) await loadTokenizer(el.model.value);
const text = el.input.value ?? '';
if (!text.trim()){
last = { tokens: [], ids: [] };
render();
status('Type to tokenize…');
return;
}
status('Tokenizing…');
try{
const enc = await tokenizer.encode(text);
if (myRun !== runId) return; // stale run
last.tokens = enc.tokens;
last.ids = enc.ids;
render();
status(`Done. ${last.tokens.length} tokens.`);
}catch(err){
console.error(err);
status('Error tokenizing. See console.');
}
}
function render(){
// Tokens pane
el.tokens.innerHTML = '';
last.tokens.forEach((tok, i)=>{
const span = document.createElement('span');
span.className = 'chip';
span.dataset.idx = i;
span.textContent = tok;
span.addEventListener('mouseenter', ()=>highlightID(i,true));
span.addEventListener('mouseleave', ()=>highlightID(i,false));
el.tokens.appendChild(span);
});
// IDs pane
el.ids.textContent = last.ids.join(' ');
if (!last.tokens.length) status('Type to tokenize…');
}
function highlightID(i, on){
if (!last.ids.length) return;
const parts = last.ids.map((id, idx)=> (idx===i && on ? `[${id}]` : String(id)));
el.ids.textContent = parts.join(' ');
const chip = el.tokens.querySelector(`[data-idx="${i}"]`);
if (chip) chip.classList.toggle('active', on);
}
// Copy / Export
el.copyTokens.addEventListener('click', async ()=>{
if (!last.tokens.length) return;
await navigator.clipboard.writeText(last.tokens.join(' '));
status('Tokens copied.');
});
el.exportTokens.addEventListener('click', ()=>{
download('tokens.json', JSON.stringify(last.tokens, null, 2), 'application/json');
});
el.copyIds.addEventListener('click', async ()=>{
if (!last.ids.length) return;
await navigator.clipboard.writeText(last.ids.join(' '));
status('IDs copied.');
});
el.exportIds.addEventListener('click', ()=>{
download('ids.json', JSON.stringify(last.ids, null, 2), 'application/json');
});
el.exportCSV.addEventListener('click', ()=>{
if (!last.tokens.length) return;
const rows = last.tokens.map((t,i)=>[i,t,last.ids[i]]);
const csv = [['index','token','id'], ...rows].map(r=>r.map(csvCell).join(',')).join('\n');
download('tokens_and_ids.csv', csv, 'text/csv');
});
function csvCell(v){
const s = String(v);
return /[",\n]/.test(s) ? `"${s.replace(/"/g,'""')}"` : s;
}
function download(name, data, type){
const blob = new Blob([data], {type});
const url = URL.createObjectURL(blob);
const a = Object.assign(document.createElement('a'), {href:url, download:name});
a.click(); URL.revokeObjectURL(url);
}
// Events
el.btn.addEventListener('click', tokenize);
el.input.addEventListener('input', debounce(tokenize, 250));
el.model.addEventListener('change', async ()=>{
tokenizer = null;
await loadTokenizer(el.model.value);
tokenize();
});
// Initial load
await loadTokenizer(el.model.value);
tokenize();
</script>
</body>
</html>