Add Token counter plugin

public/scripts/extensions/token-counter/index.js

@@ -0,0 +1,40 @@
+import { callPopup, main_api } from "../../../script.js";
+import { getContext } from "../../extensions.js";
+import { oai_settings } from "../../openai.js";
+
+async function doTokenCounter() {
+    const selectedTokenizer = main_api == 'openai'
+        ? `tiktoken (${oai_settings.openai_model})`
+        : $("#tokenizer").find(':selected').text();
+    const html = `
+    <div class="wide100p">
+        <h3>Token Counter</h3>
+        <div class="justifyLeft">
+            <h4>Type / paste in the box below to see the number of tokens in the text.</h4>
+            <p>Selected tokenizer: ${selectedTokenizer}</p>
+            <textarea id="token_counter_textarea" class="wide100p textarea_compact margin-bot-10px" rows="10"></textarea>
+            <div>Tokens: <span id="token_counter_result">0</span></div>
+        </div>
+    </div>`;
+
+    const dialog = $(html);
+    dialog.find('#token_counter_textarea').on('input', () => {
+        const text = $('#token_counter_textarea').val();
+        const context = getContext();
+        const count = context.getTokenCount(text);
+        $('#token_counter_result').text(count);
+    });
+
+    $('#dialogue_popup').addClass('wide_dialogue_popup');
+    callPopup(dialog, 'text');
+}
+
+jQuery(() => {
+    const buttonHtml = `
+        <div id="token_counter" class="list-group-item flex-container flexGap5">
+            <div class="fa-solid fa-1 extensionsMenuExtensionButton" /></div>
+            Token Counter
+        </div>`;
+    $('#extensionsMenu').prepend(buttonHtml);
+    $('#token_counter').on('click', doTokenCounter);
+});

public/scripts/extensions/token-counter/manifest.json

@@ -0,0 +1,11 @@
+{
+    "display_name": "Token Counter",
+    "loading_order": 15,
+    "requires": [],
+    "optional": [],
+    "js": "index.js",
+    "css": "style.css",
+    "author": "Cohee#1207",
+    "version": "1.0.0",
+    "homePage": "https://github.com/Cohee1207/SillyTavern"
+}

public/style.css

@@ -557,6 +557,7 @@ code {
     font-size: 20px;
     height: 20px;
     width: 20px;
+    text-align: center;
 }
 
 hr {
@@ -4466,4 +4467,4 @@ body.waifuMode #avatar_zoom_popup {
         overflow-y: auto;
         overflow-x: hidden;
     }
-}
+}