Update static/script.js

static/script.js

@@ -54,6 +54,24 @@ function getMessages(roomId=myRoomId, url="https://jacobinathanialpeterson-chatb
 		})
 		.then(response => { if (!response.ok) { throw new Error('Network response was not ok.') } return response.json() })
 		.then(data => {
+            if (data) {
+                // Get the button element you want to preserve
+                const preserveButton = document.getElementById('createRoom');
+                
+                // Get the parent element containing all the elements you want to remove
+                const parentElement = document.body; // Change this selector according to your DOM structure
+                
+                // Get all children of the parent element
+                const children = parentElement.children;
+                
+                // Loop through the children and remove elements that are not the preserveButton
+                for (let i = 0; i < children.length; i++) {
+                    const element = children[i];
+                    if (element !== preserveButton) {
+                        element.remove(); // Remove the element from the DOM
+                    }
+                }
+            }
 			// Get the keys of the object
 			const keys = Object.keys(data);
 
@@ -105,24 +123,6 @@ function updateUI() {
           <div class="msg">${value.message}</div>
         </div>`;
     });
-  } else {
-    // Get the button element you want to preserve
-    const preserveButton = document.getElementById('createRoom');
-    
-    // Get the parent element containing all the elements you want to remove
-    const parentElement = document.body; // Change this selector according to your DOM structure
-    
-    // Get all children of the parent element
-    const children = parentElement.children;
-    
-    // Loop through the children and remove elements that are not the preserveButton
-    for (let i = 0; i < children.length; i++) {
-        const element = children[i];
-        if (element !== preserveButton) {
-            element.remove(); // Remove the element from the DOM
-        }
-    }
-
   }
   messagesContainer.innerHTML = elementValue;