From 1f2ffce85bdeda256016307bcd5e337debd6ddb4 Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Tue, 16 Apr 2024 17:36:10 +0530
Subject: [PATCH] Copy chat message with it's markdown formatting in Web,
 Desktop apps

---
 src/interface/desktop/chat.html  | 15 ++++++++++-----
 src/khoj/interface/web/chat.html | 15 ++++++++++-----
 2 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/src/interface/desktop/chat.html b/src/interface/desktop/chat.html
index fc7ecc2e..d796d930 100644
--- a/src/interface/desktop/chat.html
+++ b/src/interface/desktop/chat.html
@@ -13,9 +13,14 @@
 
     <script>
         let chatOptions = [];
-        function copyParentText(event) {
+        function createCopyParentText(message) {
+            return function(event) {
+                copyParentText(event, message);
+            }
+        }
+        function copyParentText(event, message=null) {
             const button = event.currentTarget;
-            const textContent = button.parentNode.textContent.trim();
+            const textContent = message ?? button.parentNode.textContent.trim();
             navigator.clipboard.writeText(textContent).then(() => {
                 button.firstChild.src = "./assets/icons/copy-button-success.svg";
                 setTimeout(() => {
@@ -306,9 +311,9 @@
             return renderMessage(message, by, dt, references, false, "return");
         }
 
-        function formatHTMLMessage(htmlMessage, raw=false, willReplace=true) {
+        function formatHTMLMessage(message, raw=false, willReplace=true) {
             var md = window.markdownit();
-            let newHTML = htmlMessage;
+            let newHTML = message;
 
             // Remove any text between <s>[INST] and </s> tags. These are spurious instructions for the AI chat model.
             newHTML = newHTML.replace(/<s>\[INST\].+(<\/s>)?/g, '');
@@ -340,7 +345,7 @@
                 copyIcon.src = "./assets/icons/copy-button.svg";
                 copyIcon.classList.add("copy-icon");
                 copyButton.appendChild(copyIcon);
-                copyButton.addEventListener('click', copyParentText);
+                copyButton.addEventListener('click', createCopyParentText(message));
                 element.append(copyButton);
             }
 
diff --git a/src/khoj/interface/web/chat.html b/src/khoj/interface/web/chat.html
index c96fb773..4ede4080 100644
--- a/src/khoj/interface/web/chat.html
+++ b/src/khoj/interface/web/chat.html
@@ -30,9 +30,14 @@ To get started, just start typing below. You can also type / to see a list of co
         const allowedExtensions = ['text/org', 'text/markdown', 'text/plain', 'text/html', 'application/pdf'];
         const allowedFileEndings = ['org', 'md', 'txt', 'html', 'pdf'];
         let chatOptions = [];
-        function copyParentText(event) {
+        function createCopyParentText(message) {
+            return function(event) {
+                copyParentText(event, message);
+            }
+        }
+        function copyParentText(event, message=null) {
             const button = event.currentTarget;
-            const textContent = button.parentNode.textContent.trim();
+            const textContent = message ?? button.parentNode.textContent.trim();
             navigator.clipboard.writeText(textContent).then(() => {
                 button.firstChild.src = "/static/assets/icons/copy-button-success.svg";
                 setTimeout(() => {
@@ -332,9 +337,9 @@ To get started, just start typing below. You can also type / to see a list of co
             return renderMessage(message, by, dt, references, false, "return");
         }
 
-        function formatHTMLMessage(htmlMessage, raw=false, willReplace=true) {
+        function formatHTMLMessage(message, raw=false, willReplace=true) {
             var md = window.markdownit();
-            let newHTML = htmlMessage;
+            let newHTML = message;
 
             // Remove any text between <s>[INST] and </s> tags. These are spurious instructions for the AI chat model.
             newHTML = newHTML.replace(/<s>\[INST\].+(<\/s>)?/g, '');
@@ -366,7 +371,7 @@ To get started, just start typing below. You can also type / to see a list of co
                 copyIcon.src = "/static/assets/icons/copy-button.svg";
                 copyIcon.classList.add("copy-icon");
                 copyButton.appendChild(copyIcon);
-                copyButton.addEventListener('click', copyParentText);
+                copyButton.addEventListener('click', createCopyParentText(message));
                 element.append(copyButton);
             }