Change token probability when switching tokens

mikupad.html

@@ -4812,7 +4812,8 @@
 			...promptChunks.slice(0, i),
 			{
 				...promptChunks[i],
-				content: tok,
+				content: tok.tok_str,
+				prob: tok.prob
 			},
 		];
 		setPromptChunks(newPrompt);
@@ -4961,7 +4962,7 @@
 				${probs.map((prob, i) => {
 					const index = currentPromptChunk?.index;
 					const isCurrentToken = promptChunks[index]?.prob == prob.prob;
-					return html`<button key=${i} className=${isCurrentToken ? 'current' : ''} onClick=${() => switchCompletion(index, prob.tok_str)}>
+					return html`<button key=${i} className=${isCurrentToken ? 'current' : ''} onClick=${() => switchCompletion(index, prob)}>
 						<div className="tok">${replaceUnprintableBytes(prob.tok_str)}</div>
 						<div className="prob">${(prob.prob * 100).toFixed(2)}%</div>
 					</button>`;