From 10148167ba6ae7785928ebb068781d872a1423a9 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Wed, 30 Aug 2023 16:31:53 +0300
Subject: [PATCH] Option to disable group impersonation ban. Hide CFG for
 simple UI

---
 public/index.html                      |  7 +++++++
 public/script.js                       | 10 ++++++++++
 public/scripts/extensions/cfg/index.js |  2 +-
 public/scripts/power-user.js           |  7 +++++++
 4 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/public/index.html b/public/index.html
index 18376a045..e0685fc52 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2926,6 +2926,10 @@
                                 <input id="encode_tags" type="checkbox" />
                                 <span data-i18n="Show tags in responses">Show &lt;tags&gt; in responses</span>
                             </label>
+                            <label data-newbie-hidden class="checkbox_label" for="disable_group_trimming" title="Allow AI messages in groups to contain lines spoken by other group members.">
+                                <input id="disable_group_trimming" type="checkbox" />
+                                <span data-i18n="Show impersonated replies in groups">Show impersonated replies in groups</span>
+                            </label>
                             <label data-newbie-hidden class="checkbox_label" for="console_log_prompts">
                                 <input id="console_log_prompts" type="checkbox" />
                                 <span data-i18n="Log prompts to console">Log prompts to console</span>
@@ -4322,6 +4326,9 @@
                 <i class="fa-lg fa-solid fa-note-sticky"></i>
                 <span data-i18n="Author's Note">Author's Note</span>
             </a>
+            <div data-newbie-hidden id="options_advanced">
+
+            </div>
             <a id="option_back_to_main">
                 <i class="fa-lg fa-solid fa-left-long"></i>
                 <span data-i18n="Back to parent chat">Back to parent chat</span>
diff --git a/public/script.js b/public/script.js
index c39fc9621..5ef41fc2a 100644
--- a/public/script.js
+++ b/public/script.js
@@ -1895,7 +1895,17 @@ export function extractMessageBias(message) {
     }
 }
 
+/**
+ * Removes impersonated group member lines from the group member messages.
+ * Doesn't do anything if group reply trimming is disabled.
+ * @param {string} getMessage Group message
+ * @returns Cleaned-up group message
+ */
 function cleanGroupMessage(getMessage) {
+    if (power_user.disable_group_trimming) {
+        return getMessage;
+    }
+
     const group = groups.find((x) => x.id == selected_group);
 
     if (group && Array.isArray(group.members) && group.members) {
diff --git a/public/scripts/extensions/cfg/index.js b/public/scripts/extensions/cfg/index.js
index f572d2c63..238c6c5d6 100644
--- a/public/scripts/extensions/cfg/index.js
+++ b/public/scripts/extensions/cfg/index.js
@@ -385,7 +385,7 @@ jQuery(async () => {
 
     const buttonHtml = $(await $.get(`${extensionFolderPath}/menuButton.html`));
     buttonHtml.on('click', onCfgMenuItemClick)
-    buttonHtml.insertAfter("#option_toggle_AN");
+    buttonHtml.appendTo("#options_advanced");
 
     // Hook events
     eventSource.on(event_types.CHAT_CHANGED, async () => {
diff --git a/public/scripts/power-user.js b/public/scripts/power-user.js
index 5bbf62a24..a3f36e7bf 100644
--- a/public/scripts/power-user.js
+++ b/public/scripts/power-user.js
@@ -165,6 +165,7 @@ let power_user = {
     continue_on_send: false,
     trim_spaces: true,
     relaxed_api_urls: false,
+    disable_group_trimming: false,
 
     default_instruct: '',
     instruct: {
@@ -789,6 +790,7 @@ function loadPowerUserSettings(settings, data) {
     $("#trim_sentences_checkbox").prop("checked", power_user.trim_sentences);
     $("#include_newline_checkbox").prop("checked", power_user.include_newline);
     $('#render_formulas').prop("checked", power_user.render_formulas);
+    $('#disable_group_trimming').prop("checked", power_user.disable_group_trimming);
     $("#markdown_escape_strings").val(power_user.markdown_escape_strings);
     $("#fast_ui_mode").prop("checked", power_user.fast_ui_mode);
     $("#waifuMode").prop("checked", power_user.waifuMode);
@@ -2160,6 +2162,11 @@ $(document).ready(() => {
         saveSettingsDebounced();
     });
 
+    $('#disable_group_trimming').on('input', function () {
+        power_user.disable_group_trimming = !!$(this).prop('checked');
+        saveSettingsDebounced();
+    });
+
     $('#debug_menu').on('click', function () {
         showDebugMenu();
     });