Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -623,31 +623,11 @@ def alphafold(input_set, mode, impute):
                             KeyError
                     info_per_model[mod][annot] = annotation_pos_on_pdb_
 
-                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:
-                    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:
-                    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')
-
-                #with gzip.open(pdb_path, mode="rt") as f:
-                #    file_content = f.read()
-                #    st.write(file_content)
-
-                
-                
-                #st.write(get_alignments_3D(uniprotID, mod, pdb_path, pdbSequence, 'nan', 'nan', 'nan', mode, Path(path_to_output_files / '3D_alignment'),
-                #                     'gzip'))
-
-
-
+                pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
 
 
                 pdbSequence = convert_non_standard_amino_acids(pdbSequence)
-           
+                st.write('pdbSequence', pdbSequence)
     
                 atomSequence = ''
                 coords = []