more tests working!!

molecularnodes/entities/molecule/base.py

@@ -257,6 +257,7 @@ def add_style(
 
         if assembly:
             bl.nodes.assembly_initialise(self.object)
+            bl.nodes.assembly_insert(self.object)
 
         return self
 
@@ -310,11 +311,7 @@ def _setup_frames_collection(self):
     def _store_object_custom_properties(obj, reader: ReaderBase):
         obj["entity_ids"] = reader.entity_ids()
         obj["chain_ids"] = reader.chain_ids()
-
-        try:
-            obj.mn.biological_assemblies = json.dumps(reader.assemblies())
-        except InvalidFileError:
-            obj.mn.biological_assemblies = ""
+        obj.mn.biological_assemblies = reader.assemblies(as_json_string=True)
 
     @classmethod
     def _create_object(

molecularnodes/entities/molecule/reader.py

@@ -66,11 +66,13 @@ def entity_ids(self) -> list[str] | None:
     def chain_ids(self) -> list[str] | None:
         return np.unique(self.array.chain_id).tolist()
 
-    def assemblies(self):
-        return json.dumps(self._assemblies())
+    def assemblies(self, as_json_string: bool = False) -> dict | str:
+        if as_json_string:
+            return json.dumps(self._assemblies())
+        return self._assemblies()
 
     def _assemblies(self):
-        return ""
+        return {}
 
     @staticmethod
     def set_extra_annotations(

molecularnodes/ui/ops.py

@@ -99,6 +99,9 @@ def poll(self, context):
 
     def execute(self, context):
         obj = context.active_object
+        if not isinstance(obj, bpy.types.Object):
+            self.report({"ERROR"}, "No active object")
+            return {"CANCELLED"}
         with databpy.nodes.DuplicatePrevention():
             try:
                 if self.inset_node: