Update app.py

app.py

@@ -28,17 +28,17 @@ import numpy as np
 from data.scripts.data_utils import modify_bfactor_biotite
 
 def process_pdb_file(pdb_file, backbones, sequences, names):
-        _name = pdb_file[:-4]
-        _chain = ""
-        parsed_pdb = parse_PDB(pdb_file, name=_name, input_chain_list=[_chain])[0]
-        backbone, sequence = parsed_pdb['coords_chain_{}'.format(_chain)], parsed_pdb['seq_chain_{}'.format(_chain)]
-        if len(sequence) > 1023:
-            print("Sequence length is greater than 1023, skipping {}".format(_name))
-        else:
-            backbones.append(backbone)
-            sequences.append(sequence)
-            names.append(_name)
-        return backbones, sequences, names
+    _name = pdb_file[:-4]
+    _chain = ""
+    parsed_pdb = parse_PDB(pdb_file, name=_name, input_chain_list=[_chain])[0]
+    backbone, sequence = parsed_pdb['coords_chain_{}'.format(_chain)], parsed_pdb['seq_chain_{}'.format(_chain)]
+    if len(sequence) > 1023:
+        print("Sequence length is greater than 1023, skipping {}".format(_name))
+    else:
+        backbones.append(backbone)
+        sequences.append(sequence)
+        names.append(_name)
+    return backbones, sequences, names
 
 def flex_seq(input_seq, input_file):
     if not input_seq:
@@ -399,6 +399,9 @@ def rescale_bfactors(pdb_file):
 
 def clear_files():
     folder = 'prediction_results/'
+    if not os.path.isdir(folder):
+        os.makedirs(folder)
+    
     for filename in os.listdir(folder):
         file_path = os.path.join(folder, filename)
         os.remove(file_path)