Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -367,10 +367,14 @@ def alphafold(input_set, mode, impute):
                             KeyError
                     info_per_model[mod][annot] = annotation_pos_on_pdb_
                 #pdb_path = Path(f'{alphafold_path}/AF-{uniprotID}-F{mod}-{file_str}.pdb.gz')
-
+                st.write('HERE1')
                 st.write(f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz")
-                pdb_path= hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures",repo_type = 'dataset', filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz")
-                print('HERE2')
+                pdb_path= hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{name}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                with gzip.open(pdb_path, mode="rt") as f:
+                    file_content = f.read()
+                    st.write(file_content)
+
+                st.write('HERE2')
                 st.write(pdb_path)
                 if get_alignments_3D(uniprotID, mod, pdb_path, pdbSequence, 'nan', 'nan', 'nan', mode, Path(path_to_output_files / '3D_alignment'),
                                      'gzip') != None: