MediaWiki:Gadget-LabelScan.js: Unterschied zwischen den Versionen
Admin (Diskussion | Beiträge) Keine Bearbeitungszusammenfassung |
Admin (Diskussion | Beiträge) Keine Bearbeitungszusammenfassung |
||
| Zeile 17: | Zeile 17: | ||
const warn=(...a)=>{ if(CFG.debug) console.warn('[LabelScan]',...a); }; | const warn=(...a)=>{ if(CFG.debug) console.warn('[LabelScan]',...a); }; | ||
const err=(...a)=>{ console.error('[LabelScan]',...a); }; | const err=(...a)=>{ console.error('[LabelScan]',...a); }; | ||
const qs=id=>document.getElementById(id); | const qs=id=>document.getElementById(id); | ||
const setStatus=t=>{ const el=qs('ados-scan-status'); if(el) el.textContent=t||''; }; | const setStatus=t=>{ const el=qs('ados-scan-status'); if(el) el.textContent=t||''; }; | ||
| Zeile 23: | Zeile 22: | ||
if(p==null){ bar.hidden=true; bar.value=0; } else { bar.hidden=false; bar.value=Math.max(0,Math.min(1,p)); } }; | if(p==null){ bar.hidden=true; bar.value=0; } else { bar.hidden=false; bar.value=Math.max(0,Math.min(1,p)); } }; | ||
function | function base64ToFloat32(b64){ const bin=atob(b64), len=bin.length; | ||
const buf=new ArrayBuffer(len), view=new Uint8Array(buf); for(let i=0;i<len;i++) view[i]=bin.charCodeAt(i); return new Float32Array(buf); } | |||
const | |||
// ---------------- Index ---------------- | // ---------------- Index ---------------- | ||
| Zeile 43: | Zeile 36: | ||
if(!Array.isArray(json)) throw new Error('Index ist keine Array-JSON'); | if(!Array.isArray(json)) throw new Error('Index ist keine Array-JSON'); | ||
INDEX = json; | INDEX = json; | ||
INDEX_EMB = INDEX.map | INDEX_EMB = INDEX.map(it => (typeof it.embed==='string' && it.embed.length) ? base64ToFloat32(it.embed) : null); | ||
log('Index geladen:', INDEX.length, 'Einträge'); | log('Index geladen:', INDEX.length, 'Einträge'); | ||
log('Embeddings vorhanden:', | log('Embeddings vorhanden:', INDEX_EMB.filter(v=>v&&v.length).length, '/', INDEX.length); | ||
if(ui) setProgress(0.06); | if(ui) setProgress(0.06); | ||
return INDEX; | return INDEX; | ||
} | } | ||
| Zeile 63: | Zeile 47: | ||
async function ensureClipVision(){ | async function ensureClipVision(){ | ||
if(_visionLoadPromise) return _visionLoadPromise; | if(_visionLoadPromise) return _visionLoadPromise; | ||
setStatus('Modell laden …'); setProgress(0.08); | setStatus('Modell laden …'); setProgress(0.08); | ||
| Zeile 69: | Zeile 52: | ||
const mod = await import(/* webpackIgnore: true */ CFG.transformersURL); | const mod = await import(/* webpackIgnore: true */ CFG.transformersURL); | ||
// | // nur LOKAL | ||
mod.env.allowLocalModels = true; | mod.env.allowLocalModels = true; | ||
mod.env.allowRemoteModels = false; | mod.env.allowRemoteModels = false; | ||
mod.env.localModelPath = CFG.localModelPath; | mod.env.localModelPath = CFG.localModelPath; | ||
// WASM von transformers-CDN | // WASM von transformers-CDN | ||
mod.env.backends = mod.env.backends || {}; | mod.env.backends = mod.env.backends || {}; | ||
mod.env.backends.onnx = mod.env.backends.onnx || {}; | mod.env.backends.onnx = mod.env.backends.onnx || {}; | ||
| Zeile 86: | Zeile 69: | ||
]); | ]); | ||
let backend='unknown'; | let backend='unknown'; try { backend = model?.session?.executionProvider || backend; } catch(_){} | ||
log('CLIP ready (vision, local):', model?.constructor?.name || 'unknown', '| Backend:', backend); | log('CLIP ready (vision, local):', model?.constructor?.name || 'unknown', '| Backend:', backend); | ||
return { mod, processor, model }; | return { mod, processor, model }; | ||
})(); | })(); | ||
| Zeile 96: | Zeile 77: | ||
} | } | ||
// Datei -> | // Datei -> HTMLImageElement -> Processor(image) -> Modell | ||
async function embedFileImage(file){ | async function embedFileImage(file){ | ||
function loadImage(f){ | function loadImage(f){ | ||
| Zeile 107: | Zeile 88: | ||
img.src=url; | img.src=url; | ||
}); | }); | ||
} | } | ||
| Zeile 122: | Zeile 94: | ||
setStatus('Bild vorbereiten …'); setProgress(0.20); | setStatus('Bild vorbereiten …'); setProgress(0.20); | ||
const img = await loadImage(file); | const img = await loadImage(file); | ||
setStatus('Bild analysieren …'); setProgress(0.38); | setStatus('Bild analysieren …'); setProgress(0.38); | ||
// WICHTIG: Processor | // 🔧 WICHTIG: Gib dem Processor direkt das HTMLImageElement | ||
const inputs = await processor( | const inputs = await processor(img, { return_tensors: 'pt' }); | ||
const out = await model.forward({ pixel_values: inputs.pixel_values }); | const out = await model.forward({ pixel_values: inputs.pixel_values }); | ||
| Zeile 135: | Zeile 106: | ||
} | } | ||
function normalize(v){ | function normalize(v){ let n=0; for(let i=0;i<v.length;i++) n+=v[i]*v[i]; n=Math.sqrt(n)||1; const o=new Float32Array(v.length); for(let i=0;i<v.length;i++) o[i]=v[i]/n; return o; } | ||
function cosine(a,b){ let s=0,L=Math.min(a.length,b.length); for(let i=0;i<L;i++) s+=a[i]*b[i]; return s; } | function cosine(a,b){ let s=0,L=Math.min(a.length,b.length); for(let i=0;i<L;i++) s+=a[i]*b[i]; return s; } | ||
function rankByCosine(q){ const s=[]; for(let i=0;i<INDEX.length;i++){ const v=INDEX_EMB[i]; if(!v) continue; s.push({i,score:cosine(q,v)}); } s.sort((a,b)=>b.score-a.score); return s.slice(0,CFG.topK); } | |||
function rankByCosine(q){ | |||
function renderResults(r){ | function renderResults(r){ | ||
const box=qs('ados-scan-results'); | const box=qs('ados-scan-results'); if(!box) return; box.innerHTML=''; | ||
if(!r.length){ box.innerHTML='<div class="empty">Keine klaren Treffer.</div>'; return; } | |||
r.forEach(({i,score})=>{ | |||
if(!r.length){ | |||
r.forEach(({ i, score })=>{ | |||
const it=INDEX[i]; | const it=INDEX[i]; | ||
const link = mw.util.getUrl((it.title||'').replace(/ /g,'_')); | const link = mw.util.getUrl((it.title||'').replace(/ /g,'_')); | ||
const thumb = it.thumb || ''; | const thumb=it.thumb||''; | ||
box.innerHTML += | box.innerHTML+= | ||
`<div class="ados-hit" style="display:grid;grid-template-columns:60px 1fr auto;gap:10px;align-items:center;padding:.35rem 0;"> | `<div class="ados-hit" style="display:grid;grid-template-columns:60px 1fr auto;gap:10px;align-items:center;padding:.35rem 0;"> | ||
${thumb ? `<img src="${thumb}" style="width:60px;border-radius:6px;">` : `<div></div>`} | ${thumb?`<img src="${thumb}" style="width:60px;border-radius:6px;">`:`<div></div>`} | ||
<div><b><a href="${link}">${mw.html.escape(it.title||'')}</a></b></div> | <div><b><a href="${link}">${mw.html.escape(it.title||'')}</a></b></div> | ||
<div style="color:#666;font-variant-numeric:tabular-nums">${score.toFixed(3)}</div> | <div style="color:#666;font-variant-numeric:tabular-nums">${score.toFixed(3)}</div> | ||
| Zeile 186: | Zeile 136: | ||
const btnCam = qs('ados-scan-btn-camera'); | const btnCam = qs('ados-scan-btn-camera'); | ||
const btnGal = qs('ados-scan-btn-gallery'); | const btnGal = qs('ados-scan-btn-gallery'); | ||
if(!btnRun || !inCam || !inGal) return; | if(!btnRun || !inCam || !inGal) return; | ||
| Zeile 194: | Zeile 142: | ||
btnGal && btnGal.addEventListener('click', ()=> inGal.click()); | btnGal && btnGal.addEventListener('click', ()=> inGal.click()); | ||
const pick = e => { const f=e.target.files?.[0]; if(f) | const pick = e => { const f=e.target.files?.[0]; if(f){ | ||
const url=URL.createObjectURL(f); | |||
const prev=qs('ados-scan-preview'); | |||
if(prev){ prev.innerHTML=`<img alt="Vorschau" style="max-width:100%;height:auto;border-radius:8px;" src="${url}">`; prev.setAttribute('aria-hidden','false'); } | |||
}}; | |||
inCam.addEventListener('change', pick); | inCam.addEventListener('change', pick); | ||
inGal.addEventListener('change', pick); | inGal.addEventListener('change', pick); | ||
btnRun.addEventListener('click', onRunClick); | btnRun.addEventListener('click', onRunClick); | ||
BOUND=true; | BOUND=true; log('UI gebunden.'); | ||
} | } | ||
async function onRunClick(){ | async function onRunClick(){ | ||
try{ | try{ | ||
const btnRun = qs('ados-scan-run'); // | const btnRun = qs('ados-scan-run'); // hier neu holen | ||
const inCam = qs('ados-scan-file-camera'); | const inCam = qs('ados-scan-file-camera'); | ||
const inGal = qs('ados-scan-file-gallery'); | const inGal = qs('ados-scan-file-gallery'); | ||
| Zeile 236: | Zeile 167: | ||
await loadIndex({ ui:true }); | await loadIndex({ ui:true }); | ||
await ensureClipVision(); // warmup | |||
await ensureClipVision(); | |||
const q = await embedFileImage(file); | const q = await embedFileImage(file); | ||
setProgress(0.70); setStatus('Abgleich …'); | |||
setProgress(0.70); | |||
renderResults(rankByCosine(q)); | renderResults(rankByCosine(q)); | ||
setStatus('Fertig.'); | setStatus('Fertig.'); setProgress(null); | ||
} catch(e){ | } catch(e){ | ||
err('Fehler', e); | err('Fehler', e); | ||
| Zeile 259: | Zeile 184: | ||
function init(){ | function init(){ | ||
if(document.readyState==='loading'){ | if(document.readyState==='loading'){ document.addEventListener('DOMContentLoaded', bindUI, { once: true }); } | ||
else { bindUI(); } | |||
loadIndex({ ui:false }).catch(err).finally(()=>{ setStatus('Bereit.'); setProgress(null); }); | loadIndex({ ui:false }).catch(err).finally(()=>{ setStatus('Bereit.'); setProgress(null); }); | ||
} | } | ||