Add support for external TAI API calls

config.conf

@@ -3,7 +3,8 @@ const port = 8000;
 const whitelist = ['127.0.0.1']; //Example for add several IP in whitelist: ['127.0.0.1', '192.168.0.10']
 const whitelistMode = false; //Disabling enabling the ip whitelist mode. true/false
 const autorun = true; //Autorun in the browser. true/false
+const enableExtensions = true; //Enables support for TavernAI-extras project
 
 module.exports = {
-  port, whitelist, whitelistMode, autorun
+  port, whitelist, whitelistMode, autorun, enableExtensions,
 };

public/index.html

@@ -33,6 +33,8 @@
 
 
     <script type=module>
+        // API OBJECT FOR EXTERNAL WIRING
+        window['TavernAI'] = {};
 
         import {encode, decode} from "../scripts/gpt-2-3-tokenizer/mod.js";
         $(document).ready(function(){
@@ -765,6 +767,14 @@
                 const newMessage = $(`#chat [mesid="${count_view_mes}"]`);
                 newMessage.data('isSystem', isSystem);
 
+                if (mes.extra?.image) {
+                    const image = document.createElement('img');
+                    image.src = mes.extra?.image;
+                    image.classList.add('img_extra');
+                    newMessage.find('.mes_text').css('font-size', '0px');
+                    newMessage.find('.mes_text').append(image);
+                }
+
                 if (isSystem) {
                     newMessage.find('.mes_edit').hide();
                 }
@@ -3371,6 +3381,16 @@
                                 }
                             });
                             // end world info settings
+
+                            if (data.enable_extensions) {
+                                const src = 'scripts/extensions.js';
+                                if ($(`script[src="${src}"]`).length === 0) {
+                                    const script = document.createElement('script');
+                                    script.type = 'text/javascript';
+                                    script.src = src;
+                                    $('body').append(script);
+                                }
+                            }
                         }
                         if(!is_checked_colab) isColab();
                         
@@ -4326,6 +4346,19 @@
                 }
                 return array;
             }
+
+            window['TavernAI'].getContext = function() {
+                return {
+                    chat: chat,
+                    characters: characters,
+                    name1: name1,
+                    name2: name2,
+                    characterId: this_chid,
+                    onlineStatus: online_status,
+                    addOneMessage: addOneMessage,
+                    generate: Generate,
+                };
+            };
         });
     </script>
 <title>Tavern.AI</title>
@@ -4921,6 +4954,5 @@
             </form>
         </div>
     </div>
-<script src="scripts/extensions.js" type="module"></script>
 </body>
 </html>

public/scripts/extensions.js

@@ -1,4 +1,9 @@
 
+const extensions_urlKey = 'extensions_url';
+const extensions_autoConnectKey = 'extensions_autoconnect';
+let extensions = [];
+
+(function () {
     const settings_html = `
     <div class="extensions_block">
         <hr>
@@ -6,6 +11,7 @@ const settings_html = `
         <input id="extensions_url" type="text" class="text_pole" />
         <div class="extensions_url_block">
             <input id="extensions_connect" type="submit" value="Connect" />
+            <span class="expander"></span>
             <input id="extensions_autoconnect" type="checkbox"/><h4>Auto-connect</h4>
         </div>
         <div id="extensions_status">Not connected</div>
@@ -59,6 +65,7 @@ const settings_style = `
         display: flex;
         align-items: center;
         margin: 10px;
+        width: 90%
     }
     
     .extensions_url_block- h4 {
@@ -77,24 +84,25 @@ const settings_style = `
     .failure {
         color: red;
     }
+    
+    .expander {
+        flex-grow: 1;
+    }
     </style>
     `;
 
-const urlKey = 'extensions_url';
-const autoConnectKey = 'extensions_autoconnect';
     const defaultUrl = "http://localhost:5100";
     let connectedToApi = false;
-let extensions = [];
 
     async function connectClickHandler() {
         const baseUrl = $("#extensions_url").val();
-    localStorage.setItem(urlKey, baseUrl);
+        localStorage.setItem(extensions_urlKey, baseUrl);
         await connectToApi(baseUrl);
     }
 
     function autoConnectInputHandler() {
         const value = $(this).prop('checked');
-    localStorage.setItem(autoConnectKey, value.toString());
+        localStorage.setItem(extensions_autoConnectKey, value.toString());
 
         if (value && !connectedToApi) {
             $("#extensions_connect").trigger('click');
@@ -150,19 +158,11 @@ function applyExtensions(baseUrl) {
         }
 
         for (let extension of extensions) {
-        if (extension.metadata.js) {
+            addExtensionStyle(extension);
-            url.pathname = `/api/script/${extension.name}`;
+            addExtensionScript(extension);
-            const src = url.toString();
-
-            if ($(`script[src="${src}"]`).length === 0) {
-                const script = document.createElement('script');
-                script.type = 'module';
-                script.src = src;
-                $('body').append(script);
-            }
-
         }
 
+        function addExtensionStyle(extension) {
             if (extension.metadata.css) {
                 url.pathname = `/api/style/${extension.name}`;
                 const href = url.toString();
@@ -176,11 +176,25 @@ function applyExtensions(baseUrl) {
                 }
             }
         }
+
+        function addExtensionScript(extension) {
+            if (extension.metadata.js) {
+                url.pathname = `/api/script/${extension.name}`;
+                const src = url.toString();
+
+                if ($(`script[src="${src}"]`).length === 0) {
+                    const script = document.createElement('script');
+                    script.type = 'module';
+                    script.src = src;
+                    $('body').append(script);
+                }
+            }
+        }
     }
 
     $(document).ready(async function () {
-    const url = localStorage.getItem(urlKey) ?? defaultUrl;
+        const url = localStorage.getItem(extensions_urlKey) ?? defaultUrl;
-    const autoConnect = Boolean(localStorage.getItem(autoConnectKey)) ?? false;
+        const autoConnect = Boolean(localStorage.getItem(extensions_autoConnectKey)) ?? false;
         $('#rm_api_block').append(settings_html);
         $('head').append(settings_style);
         $("#extensions_url").val(url);
@@ -188,3 +202,4 @@ $(document).ready(async function () {
         $("#extensions_autoconnect").on('input', autoConnectInputHandler);
         $("#extensions_autoconnect").prop('checked', autoConnect).trigger('input');
     });
+})();

server.js

@@ -25,7 +25,7 @@ const server_port = config.port;
 const whitelist = config.whitelist;
 const whitelistMode = config.whitelistMode;
 const autorun = config.autorun;
-
+const enableExtensions = config.enableExtensions;
 
 
 var Client = require('node-rest-client').Client;
@@ -804,7 +804,8 @@ app.post('/getsettings', jsonParser, (request, response) => { //Wintermute's cod
         koboldai_setting_names,
         world_names,
         novelai_settings,
-        novelai_setting_names
+        novelai_setting_names,
+        enable_extensions: enableExtensions,
     });
 });