Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -82,7 +82,6 @@ def alphafold(input_set, mode, impute):
     path_to_input_files, path_to_output_files, path_to_domains, fisher_path, path_to_interfaces, alphafold_path, alphafold_summary= manage_files(mode)
     #out_path = path_to_output_files / 'log.txt'
     #sys.stdout = open(out_path, 'w')
-    alphafold_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures",repo_type = 'dataset', filename="AF-A0A075B7D0-F1-model_v4.pdb.gz")
 
     #file_base = str(Path(alphafold_path / '*'))
     #file_str = glob.glob(file_base)[0].split('-')[-1].split('.')[0]
@@ -210,9 +209,13 @@ def alphafold(input_set, mode, impute):
     ## With new updates, can be updated separately.
 
     pdb_info = pd.read_csv(alphafold_summary, sep='\t')
+    st.write('PDB INFO')
     st.write(pdb_info)
+    alphafold_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures",repo_type = 'dataset')
+
     ## Keeping how many models each AlphaFold structure has.
     model_count = modelCount(alphafold_path)
+    st.write(model_count)
     for k, v in model_count.items():
         model_count[k] = int(v / 2)  # two types of files for each file.
     uniprot_matched = uniprot_matched.astype(str)