Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -291,6 +291,7 @@ def get_coords(annot, alignments, coords, resnums_for_sasa, mode):
 
 
 from huggingface_hub import hf_hub_download
+from huggingface_hub import utils as u
 
 
 # FUNCTIONS
@@ -620,13 +621,17 @@ def alphafold(input_set, mode, impute):
                         except:
                             KeyError
                     info_per_model[mod][annot] = annotation_pos_on_pdb_
-                st.write('so far so good')
-                pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
-                st.write('pdb_path', pdb_path)
-
+                try:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                    st.write('1')
+                except u.EntryNotFoundError:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures2", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                    st.write('2')
+                except u.EntryNotFoundError:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures3", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                    st.write('3')
 
                 pdbSequence = convert_non_standard_amino_acids(pdbSequence)
-                st.write('pdbSequence', pdbSequence)
     
                 atomSequence = ''
                 coords = []