Add redo button to new web ui

Mozilla-Ocho · Nov 24, 2024 · d18ddf1 · d18ddf1
1 parent bc82424
commit d18ddf1
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 3 deletions.
diff --git a/llamafile/server/www/chatbot.css b/llamafile/server/www/chatbot.css
@@ -263,6 +263,20 @@ ul li:first-child {
   display: block;
 }
 
+.redo-button {
+  padding: 8px;
+  background: transparent;
+  border: none;
+  cursor: pointer;
+  font-size: 20px;
+  color: #666;
+  transition: color 0.2s;
+}
+
+.redo-button:hover {
+  color: #000;
+}
+
 .settings-button {
   padding: 0.75rem 1rem;
   background: #6c757d;

diff --git a/llamafile/server/www/chatbot.js b/llamafile/server/www/chatbot.js
@@ -38,6 +38,7 @@ const stopButton = document.getElementById("stop-button");
 const settingsButton = document.getElementById("settings-button");
 const settingsModal = document.getElementById("settings-modal");
 const closeSettings = document.getElementById("close-settings");
+const redoButton = document.getElementById('redo-button');
 
 let abortController = null;
 let disableAutoScroll = false;
@@ -71,13 +72,16 @@ function onChatInput() {
 }
 
 function cleanupAfterMessage() {
-  disableAutoScroll = false;
   chatMessages.scrollTop = chatMessages.scrollHeight;
   chatInput.disabled = false;
   sendButton.style.display = "inline-block";
   stopButton.style.display = "none";
   abortController = null;
-  chatInput.focus();
+  if (!disableAutoScroll) {
+    scrollToBottom();
+    chatInput.focus();
+  }
+  disableAutoScroll = false;
 }
 
 function onWheel(e) {
@@ -146,7 +150,6 @@ function fixUploads(str) {
   str = uploadedFiles.reduce(
     (text, [from, to]) => text.replaceAll(from, to),
     str);
-  uploadedFiles.length = 0;
   return str;
 }
 
@@ -452,13 +455,34 @@ function setupSettings() {
   });
 }
 
+function removeLastDirectChild(element) {
+  if (element.lastElementChild) {
+    element.removeChild(element.lastElementChild);
+  }
+}
+
+function onRedo() {
+  if (!chatHistory.length)
+    return;
+  removeLastDirectChild(chatMessages);
+  let msg = chatHistory.pop();
+  if (msg.role === "assistant") {
+    removeLastDirectChild(chatMessages);
+    msg = chatHistory.pop();
+  }
+  chatInput.value = msg.content;
+  chatInput.focus();
+  chatInput.dispatchEvent(new Event("input")); // adjust textarea height
+}
+
 async function chatbot() {
   flagz = await fetchFlagz();
   updateModelInfo();
   setupSettings();
   startChat([{ role: "system", content: getSystemPrompt() }]);
   sendButton.addEventListener("click", sendMessage);
   stopButton.addEventListener("click", stopMessage);
+  redoButton.addEventListener("click", onRedo);
   chatInput.addEventListener("input", onChatInput);
   chatInput.addEventListener("keydown", onKeyDown);
   document.addEventListener("wheel", onWheel);

diff --git a/llamafile/server/www/index.html b/llamafile/server/www/index.html
@@ -23,6 +23,7 @@ <h1>
     <button class="send-button" id="send-button">Send</button>
     <button class="stop-button" id="stop-button" style="display:none">Stop</button>
     <button class="settings-button" id="settings-button" title="Settings">⚙️</button>
+    <button class="redo-button" id="redo-button" title="Redo last message">↺</button>
   </div>
 </div>