Merge pull request #3036 from procule/update-token-probabilities Enhancement: Token Probabilities viewer with dynamic prefix adjustment via Ctrl-click

cfd1545b8a0f25c070bccfbaf2946660512029b6

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

Parents: 237381ac3d0db93bbc6d0c8a963813546afa2f73 abf7b901352a2af87b08de6b641577a297f433a0

public/scripts/logprobs.js+50 -2 Viewed

         prefixSpan.text(prefix);
         prefixSpan.html(prefixSpan.html().replace(/\n/g, '<br>'));
         prefixSpan.addClass('logprobs_output_prefix');
-        prefixSpan.attr('title', 'Select to reroll the last \'Continue\' generation');
+        prefixSpan.attr('title', 'Select to reroll the last \'Continue\' generation.\nHold the CTRL key when clicking to reroll from before that word.');
         prefixSpan.click(onPrefixClicked);
         addKeyboardProps(prefixSpan);
         tokenSpans.push(...withVirtualWhitespace(prefix, prefixSpan));

+}
 /**
+ * getTextBeforeClickedWord retrieves the portion of text within a span
+ * that appears before the word clicked by the user. Using the x and y
+ * coordinates from a PointerEvent, this function identifies the exact
+ * word clicked and returns the text preceding it within the span.
+ *
+ * If the clicked position does not resolve to a valid word or text node,
+ * the entire span text is returned as a fallback.
+ *
+ * @param {PointerEvent} event - The click event containing the x and y coordinates.
+ * @param {string} spanText - The full text content of the span element.
+ * @returns {string} The text before the clicked word, or the entire span text as fallback.
+ */
+function getTextBeforeClickedWord(event, spanText) {
+    const x = event.clientX;
+    const y = event.clientY;
+    const range = document.caretRangeFromPoint(x, y);
+    if (range && range.startContainer.nodeType === Node.TEXT_NODE) {
+        const textNode = range.startContainer;
+        const offset = range.startOffset;
+        // Get the full text content of the text node
+        const text = textNode.nodeValue;
+        // Find the boundaries of the clicked word
+        const start = text.lastIndexOf(" ", offset - 1) + 1;
+        // Return the text before the clicked word
+        return text.slice(0, start);
+    }
+    // If we can't determine the exact word, return the full span text as a fallback
+    return spanText;
+}
+/**
  * onPrefixClicked is called when the user clicks on the carried-over prefix
  * in the token output view. It allows them to reroll the last 'continue'
  * completion with none of the output generated from it, in case they don't
  * like the results.
+ *
+ * If the user holds the Ctrl key while clicking, only the portion of text
+ * before the clicked word is retained as the prefix for rerolling
  */
 function onPrefixClicked() {
     if (!checkGenerateReady()) {

     const { continueFrom } = getActiveMessageLogprobData();
     const messageId = chat.length - 1;
-    const prefix = continueFrom || '';
+    // Check if Ctrl key is pressed during the click
+    let prefix = continueFrom || '';
+    if (event.ctrlKey) {
+        // Ctrl is pressed - use the text before the clicked word
+        prefix = getTextBeforeClickedWord(event, continueFrom);
+    }
+    // Use the determined `prefix`
     createSwipe(messageId, prefix);
     $('.swipe_right:last').click();
     Generate('continue').then(_ => void _);