Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -37,13 +37,15 @@ def getModelInfo(uniprotID, varPos, wt, models_we_need, path_to_output_files ):
             st.write('take1')
             pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
         except u.EntryNotFoundError:
-            st.write('take2')
-            pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures3", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
-        except u.EntryNotFoundError:
-            st.write('take3')
-            pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures2", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
-        except  u.EntryNotFoundError:
-            st.write('Requested file not found.')
+            try:
+                st.write('take2')
+                pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures2", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+            except u.EntryNotFoundError:
+                try:
+                    st.write('take3')
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures3", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                except  u.EntryNotFoundError:
+                    st.write('Requested file not found.')
 
             
         sasa_val = freesasa(pdb_path, path_to_output_files, uniprotID, mod, varPos, wt)