From ec5049283a253edb5c654a907e8f8be880713d41 Mon Sep 17 00:00:00 2001
From: ilhooq <contact@sphilip.com>
Date: Tue, 10 Sep 2024 14:37:34 +0200
Subject: [PATCH] =?UTF-8?q?Gestion=20suppression=20et=20reg=C3=A9n=C3=A9ra?=
 =?UTF-8?q?tion=20de=20message?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 vite/src/ChatApp.svelte     | 23 ++++++++++++++++++++++-
 vite/src/ChatMessage.svelte | 14 +++++++++++---
 2 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/vite/src/ChatApp.svelte b/vite/src/ChatApp.svelte
index 166b29d..a959fee 100644
--- a/vite/src/ChatApp.svelte
+++ b/vite/src/ChatApp.svelte
@@ -391,6 +391,25 @@
         sendMessage(content);
     }
 
+    function handleRegenerateMessage(event) {
+        const index = event.detail;
+
+        messages[index].content = '';
+
+        messages = messages.slice(0, index + 1);
+
+        complete(messages, proxyBaseUrl, assistant, (message) => {
+            if (message.content)
+                messages[index].content = message.content;
+            }
+        );
+    }
+
+    function handleDeleteMessage(event) {
+        const index = event.detail;
+        messages = messages.slice(0, index);
+    }
+
 </script>
 
 <header>
@@ -433,7 +452,9 @@
     <div class="chat-container" bind:this="{chatContainer}" >
         {#each messages as message, index (index)}
         <ChatMessage message="{message}"
-        message_index={index} on:editMessage="{handleEditMessage}"
+        message_index={index} on:editMessage={handleEditMessage}
+        on:regenerate={handleRegenerateMessage}
+        on:delete={handleDeleteMessage}
         container="{chatContainer.parentElement}" markdown={md} />
         {/each}
     </div>
diff --git a/vite/src/ChatMessage.svelte b/vite/src/ChatMessage.svelte
index 7e1ea7a..4b38a99 100644
--- a/vite/src/ChatMessage.svelte
+++ b/vite/src/ChatMessage.svelte
@@ -60,7 +60,6 @@
         }
     }
 
-
 	function handleKeyDown(event) {
 		if (event.key === 'Enter' && event.shiftKey) {
 			// Allow the default behavior of adding a new line
@@ -86,6 +85,7 @@
 				break;
 		}
 	}
+
     function handleCopy(event) {
 
         const btn = event.currentTarget;
@@ -104,6 +104,14 @@
             })
             .catch(err => console.log(`Failed to copy clipboard: ${err.message}`));
     }
+
+    function handleRegenerate() {
+        dispatch('regenerate', message_index);
+    }
+
+    function handleDelete() {
+        dispatch('delete', message_index);
+    }
 </script>
 
 <style>
@@ -139,10 +147,10 @@
                 <button class="btn btn-primary btn-sm" title="Copier" on:click={handleCopy}>
                     <span class="icon-content_copy"></span>
                 </button>
-                <button class="btn btn-primary btn-sm" title="Régénérer">
+                <button class="btn btn-primary btn-sm" on:click={handleRegenerate} title="Régénérer">
                     <span class="icon-loop"></span>
                 </button>
-                <button class="btn btn-primary btn-sm" title="Supprimer le message">
+                <button class="btn btn-primary btn-sm" on:click={handleDelete} title="Supprimer le message">
                     <span class="icon-delete"></span>
                 </button>
             {/if}