Upload modrag_molecule_functions.py

modrag_molecule_functions.py

@@ -174,8 +174,12 @@ def structure_node(smiles_list: list[str]) -> (list[str], str, list):
   img = Draw.MolsToGridImage(all_mols, molsPerRow=3, subImgSize=(250, 250))
 
   #save the image as current_image.png
-  pic = img.data
-  with open('current_image.png', 'wb') as f:
-    f.write(pic)
+  try:
+    img.save('current_image.png')
+  except:
+    pic = img.data
+    with open('current_image.png', 'wb') as f:
+      f.write(pic)
   img = Image.open('current_image.png')
+  
   return all_structures, output_string, img