From 6a2db62801ba963d6dfe2c9d7ba58c1c71c6b551 Mon Sep 17 00:00:00 2001
From: Russell Ballestrini <russell.ballestrini@gmail.com>
Date: Sun, 29 Oct 2023 21:09:08 -0400
Subject: [PATCH] clear buffer

---
 app.py              | 4 ++--
 templates/chat.html | 5 ++++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/app.py b/app.py
index 439f673..8f4b15a 100644
--- a/app.py
+++ b/app.py
@@ -218,7 +218,7 @@ def chat_claude(username, room, message):
             db.session.add(new_message)
             db.session.commit()
 
-    socketio.emit("delete_processing_message", "", room=room)
+    socketio.emit("delete_processing_message", msg_id, room=room)
 
 
 def chat_gpt(username, room, message):
@@ -291,7 +291,7 @@ def chat_gpt(username, room, message):
             db.session.add(new_message)
             db.session.commit()
 
-    socketio.emit("delete_processing_message", "", room=room)
+    socketio.emit("delete_processing_message", msg_id, room=room)
 
 
 if __name__ == "__main__":
diff --git a/templates/chat.html b/templates/chat.html
index 9012cad..1fa0c86 100644
--- a/templates/chat.html
+++ b/templates/chat.html
@@ -166,14 +166,17 @@ socket.on('previous_messages', (data) => {
 });
 
 
-socket.on('delete_processing_message', (data) => {
+socket.on('delete_processing_message', (msg_id) => {
     const tempMessages = document.querySelectorAll("#message-null");
     tempMessages.forEach((tempMessage) => {
         tempMessage.remove();
     });
+    // Clear the message buffer for the corresponding message ID
+    delete messageBuffers[msg_id];
 });
 
 
+
 // A dictionary to hold buffers for each message ID
 const messageBuffers = {};