StructureMoleculeComponent use "material_id" or "id" in image filenam…

…e if found in Structure.properties
materialsproject · Nov 11, 2023 · 47e4305 · 47e4305
1 parent 4b8d4d5
commit 47e4305
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/crystal_toolkit/components/structure.py b/crystal_toolkit/components/structure.py
@@ -460,7 +460,12 @@ def download_image(image_data_timestamp, image_data, data):
             spg_symbol = getattr(struct_or_mol, "get_space_group_info", lambda: [""])()[
                 0
             ]
-            request_filename = f"{formula}-{spg_symbol}-crystal-toolkit.png"
+            request_filename = f"{formula}-{spg_symbol}.png"
+            material_id = struct_or_mol.properties.get(
+                "material_id", struct_or_mol.properties.get("id")
+            )
+            if material_id:
+                request_filename = f"{material_id}-{request_filename}"
 
             return {
                 "content": image_data[len("data:image/png;base64,") :],