Update app.py

app.py

@@ -189,35 +189,36 @@ def rgb_to_hex(rgb):
 
 
 def mol_to_pharm3d(mol, mode='html'):
-    if mol is None:
-        return
-    # AllChem.Compute2DCoords(mol)
-    mol = Chem.AddHs(mol)
-    params = AllChem.ETKDGv3()
-    params.randomSeed = 0xf00d  # for reproducibility
-    AllChem.EmbedMolecule(mol, params)
-
-    feats = FEAT_FACTORY.GetFeaturesForMol(mol)
-
-    view = View3DmolCell(width=320, height=200)
-    for feat in feats:
-        pos = feat.GetPos()
-        color = _featColors.get(feat.GetFamily(), (.5, .5, .5))
-        view.addSphere({
-            'center': {'x': pos.x, 'y': pos.y, 'z': pos.z},
-            'radius': 0.5,
-            'color': rgb_to_hex(color)
-        })
-
-    mol_block = Chem.MolToMolBlock(mol)
-    view.addModel(mol_block, 'sdf')
-    view.setStyle({'stick': {}})
-    view.zoomTo()
-
-    if mode == 'html':
-        return view.write_html()
-        # case 'png':
-        #     return view.png()
+    try:
+        # AllChem.Compute2DCoords(mol)
+        mol = Chem.AddHs(mol)
+        params = AllChem.ETKDGv3()
+        params.randomSeed = 0xf00d  # for reproducibility
+        AllChem.EmbedMolecule(mol, params)
+    
+        feats = FEAT_FACTORY.GetFeaturesForMol(mol)
+    
+        view = View3DmolCell(width=320, height=200)
+        for feat in feats:
+            pos = feat.GetPos()
+            color = _featColors.get(feat.GetFamily(), (.5, .5, .5))
+            view.addSphere({
+                'center': {'x': pos.x, 'y': pos.y, 'z': pos.z},
+                'radius': 0.5,
+                'color': rgb_to_hex(color)
+            })
+    
+        mol_block = Chem.MolToMolBlock(mol)
+        view.addModel(mol_block, 'sdf')
+        view.setStyle({'stick': {}})
+        view.zoomTo()
+    
+        if mode == 'html':
+            return view.write_html()
+            # case 'png':
+            #     return view.png()
+    except Exception:
+        return None
 
 
 class HelpTip: