Console log 2

output.py

@@ -352,15 +352,20 @@ STARTUP_JS = """
     window.selected_elements = {}
     
     function handleMessage(event) {
-        console.log("New message: ", event.data)
-        let atom = event.data.data["atom"];
-        let add = event.data.data["add"];
-        console.log("add: ", add, " atom: ", atom);
-        window.selected_elements[atom] = add;
+        if ("atom" in event.data.data) {
+            if ("add" in event.data.data) {
+                console.log("New message: ", event.data)
+                let atom = event.data.data["atom"];
+                let add = event.data.data["add"];
+                console.log("add: ", add, " atom: ", atom);
+                window.selected_elements[atom] = add;
+            }
+        }
     }
     
     window.addEventListener("message", handleMessage);
     console.log("Listener Added"); 
+    console.log(window.selected_elements);
 }
 """
 
@@ -378,6 +383,7 @@ RETURN_SELECTION_JS = """
       }
     }
     console.log("Finished parsing");
+    console.log(window.selected_elements);
     return [input_file, input_protein_file, n_steps, n_atoms, n_samples, selected.join(",")];
 }
 """