Update app.py

app.py

@@ -1,6 +1,7 @@
 
 import os
 import json
+import uuid
 
 import gradio as gr
 from gradio_moleculeview import moleculeview
@@ -327,23 +328,22 @@ def predict(input_mol, style, contour_level, view_str, chains):
     chain_str = ""
     chain_dict = json.loads(chains)
 
+    outputfile = str(uuid.uuid4())+".svg"
     # sort keys in dict and add colors to chain_str
     for chain in sorted(chain_dict.keys()):
         chain_str += f" '{chain_dict[chain]}'"
     if style == "Goodsell3D":
-        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline residue --color_by chain --depth_shading --depth_lines --colors {chain_str}  --depth flat --back_outline --view view_matrix --save outline_all.svg")
+        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline residue --color_by chain --depth_shading --depth_lines --colors {chain_str}  --depth flat --back_outline --view view_matrix --save {outputfile}")
     elif style == "Contour":
-        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --color_by chain  --depth_contour_interval {contour_level} --colors {chain_str}  --depth contours --back_outline --view view_matrix --save outline_all.svg")
+        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --color_by chain  --depth_contour_interval {contour_level} --colors {chain_str}  --depth contours --back_outline --view view_matrix --save {outputfile}")
     else:
-        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --colors {chain_str}  --depth flat --back_outline --view view_matrix --save outline_all.svg")
+        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --colors {chain_str}  --depth flat --back_outline --view view_matrix --save {outputfile}")
     
     #read content of file
-    print(os.stat("outline_all.svg").st_size / (1024 * 1024))
-    os.system("inkscape outline_all.svg --actions='select-all;path-simplify;export-plain-svg' --export-filename pdb_opt.svg")
-    print(os.stat("outline_all.svg").st_size / (1024 * 1024))
+    os.system(f"inkscape {outputfile} --actions='select-all;path-simplify;export-plain-svg' --export-filename {outputfile.replace('.svg', '_opt.svg')}")
 
     
-    return html_output("outline_all.svg")
+    return html_output(outputfile.replace('.svg', '_opt.svg'))