From 75090c4fa4297a611c17b835014801b178e85817 Mon Sep 17 00:00:00 2001
From: RossAscends <124905043+RossAscends@users.noreply.github.com>
Date: Sun, 18 Jun 2023 08:27:24 +0900
Subject: [PATCH] WorldInfo can now be routed into Author's Note

---
 public/index.html            |  6 ++++++
 public/scripts/world-info.js | 11 +++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/public/index.html b/public/index.html
index 9ce46b9e0..555fcc24a 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2902,6 +2902,7 @@
 
                             </div>
                             <div class="world_entry_form_control world_entry_form_radios  wi-enter-footer-text ">
+                                <small>Placement</small>
                                 <div>
                                     <label><input type="radio" name="position" value="0">
                                         <span data-i18n="Before Char">Before Char</span>
@@ -2912,6 +2913,11 @@
                                         <span data-i18n="After Char">After Char</span>
                                     </label>
                                 </div>
+                                <div>
+                                    <label><input type="radio" name="position" value="2">
+                                        <span data-i18n="Author's Note">Author's Note</span>
+                                    </label>
+                                </div>
 
                             </div>
                             <div class="world_entry_form_control wi-enter-footer-text  flex-container flexNoGap ">
diff --git a/public/scripts/world-info.js b/public/scripts/world-info.js
index d71da7612..22ecb8311 100644
--- a/public/scripts/world-info.js
+++ b/public/scripts/world-info.js
@@ -1,5 +1,6 @@
 import { saveSettings, callPopup, substituteParams, getTokenCount, getRequestHeaders } from "../script.js";
 import { download, debounce, delay, initScrollHeight, resetScrollHeight } from "./utils.js";
+import { getContext } from "./extensions.js";
 
 export {
     world_info,
@@ -34,6 +35,7 @@ const saveSettingsDebounced = debounce(() => saveSettings(), 1000);
 const world_info_position = {
     before: 0,
     after: 1,
+    authorsnote: 2,
 };
 
 function getWorldInfoPrompt(chat2) {
@@ -216,7 +218,7 @@ function appendWorldEntry(entry) {
     commentToggle.on("input", function () {
         const uid = $(this).data("uid");
         const value = $(this).prop("checked");
-        console.log(value)
+        //console.log(value)
         const commentContainer = $(this)
             .closest(".world_entry")
             .find(".commentContainer");
@@ -533,6 +535,7 @@ function checkWorldInfo(chat) {
         return "";
     }
 
+    const context = getContext();
     const messagesToLookBack = world_info_depth * 2;
     let textToScan = transformString(chat.slice(0, messagesToLookBack).join(""));
     let worldInfoBefore = "";
@@ -595,10 +598,14 @@ function checkWorldInfo(chat) {
                 worldInfoAfter = `${substituteParams(
                     entry.content
                 )}\n${worldInfoAfter}`;
-            } else {
+            } else if (entry.position === world_info_position.before) {
                 worldInfoBefore = `${substituteParams(
                     entry.content
                 )}\n${worldInfoBefore}`;
+            } else {
+                let originalAN = context.extensionPrompts['2_floating_prompt'].value;
+                let ANWithWI = originalAN + "\n" + entry.content;
+                context.extensionPrompts['2_floating_prompt'].value = ANWithWI;
             }
 
             if (