From f1b9abb4e07253befa3c83a8570109cdd98151a2 Mon Sep 17 00:00:00 2001
From: Raven Scott <raven@ssh.surf>
Date: Wed, 2 Oct 2024 15:31:33 -0400
Subject: [PATCH] remove last message if 429

---
 public/js/chat.js | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/public/js/chat.js b/public/js/chat.js
index b6d590f..1a0eb59 100644
--- a/public/js/chat.js
+++ b/public/js/chat.js
@@ -157,10 +157,21 @@ function displayAlert(type, message) {
 
 // Handles error responses based on status code
 function handleErrorResponse(status) {
+    const messages = document.getElementById('messages');
+    const lastMessage = messages.lastElementChild;
+
     if (status === 429) {
         displayAlert('error', 'Sorry, I am currently too busy at the moment!');
+        // Remove the last user message if the status is 429
+        if (lastMessage && lastMessage.classList.contains('user')) {
+            messages.removeChild(lastMessage);
+        }
     } else {
         displayMessage('Error: ' + status, 'assistant');
+        // Remove the last user message for any other errors
+        if (lastMessage && lastMessage.classList.contains('user')) {
+            messages.removeChild(lastMessage);
+        }
     }
 }