svg fix

fp2lig.py

@@ -43,14 +43,14 @@ def draw(smiles):
     drawer.DrawMolecule(mc)
     drawer.FinishDrawing()
     svg = drawer.GetDrawingText()
-    svg = svg.replace('svg:','')
+    svg = svg.replace(':svg','')
     savepath = os.path.join("static", "mol_drawings", f"{smiles_hash}.svg")
     # Draw.MolToFile( mc, savepath )
+    # for line in svg_file:
+        # if "xmlns:svg" in line:
+            # line = line.replace(":svg", "")
     with open(savepath, "w") as svg_file:
-        for line in svg_file:
-            if "xmlns:svg" in line:
-                line = line.replace(":svg", "")
-            svg_file.write(line)
+        svg_file.write(svg)
     # drawing = svg2rlg(f"../Data/{name}.svg")
     # renderPDF.drawToFile(svg, f"../Data/ligand_images/{name}.pdf")
 def screen(fp_pred, library, n_hits=30):