Update app.py

app.py

@@ -331,9 +331,8 @@ def preprocess_mol(pdb_code="", filepath=""):
         mol = Molecule(f"{pdb_code}.pdb")
 
     print("print molecule loaded")
-    tf_original = tempfile.NamedTemporaryFile(delete=False, suffix=".pdb") 
-    print(tf_original)
-    mol.write(tf_original.name)
+    random_dir = tempfile.TemporaryDirectory() 
+    mol.write(f"{random_dir.name}/original.pdb")
     # clean messy files and only include protein itself
     mol.filter("protein")
     # renumber using moleculekit 0...len(protein)
@@ -346,13 +345,10 @@ def preprocess_mol(pdb_code="", filepath=""):
             indexes.append(j)
             j += 1
     df["proteinMPNN_index"] = indexes
-    tf_cleaned = tempfile.NamedTemporaryFile(delete=False, suffix=".pdb") 
-    print("cleaned", tf_cleaned)
-    mol.write(tf_cleaned.name)
-    print(os.listdir())
-    print("tf_cleaned", tf_cleaned.name)
-    print("tf_original", tf_original.name)
-    return tf_cleaned.name, df, tf_original.name
+
+    mol.write(f"{random_dir.name}/original.pdb")
+
+    return f"{random_dir.name}/cleaned.pdb", df, f"{random_dir.name}/original.pdb"
 
 
 def assign_sasa(mol):