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 = {};