Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -191,13 +191,11 @@ def pdb(input_set, mode, impute):
 
         for prot in protein:
             pdbs.append(get_pdb_ids(prot))
-        print('PDBs', pdbs)
         if len(pdbs) >= 1:
-            print('pdbs not empty')
             pdbs = [item for sublist in pdbs for item in sublist]
-            print('NEW', pdbs)
+            print('PDB List:', pdbs)
         else:
-            print('pdbs empty')
+            print('PDB List Empty')
             pdbs = []
         print('Processing PDB structures...\n')
         if pdbs == []:
@@ -218,14 +216,13 @@ def pdb(input_set, mode, impute):
         cnt = 0
         st.write('this is the pdbs', pdbs)
         def fetch_uniprot_ids(pdb_code):
-            try:
-                response = requests.get(f"https://www.ebi.ac.uk/pdbe/api/mappings/uniprot/{pdb_code}")
-                response.raise_for_status()  # Check for a successful response
-                data = response.json()
-                st.write(list(list(list(data.values())[0].values())[0].keys()))
-                return list(list(list(data.values())[0].values())[0].keys())
-            except :
-                return []
+            response = requests.get(f"https://www.ebi.ac.uk/pdbe/api/mappings/uniprot/{pdb_code}")
+            st.write('response', response)
+            response.raise_for_status()  # Check for a successful response
+            data = response.json()
+            st.write(list(list(list(data.values())[0].values())[0].keys()))
+            return list(list(list(data.values())[0].values())[0].keys())
+
         for search in pdbs:
             # Step 1: Fetch the PDB file
             pdb_url = f"https://files.rcsb.org/download/{search}.pdb"