Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -230,19 +230,23 @@ def pdb(input_set, mode, impute):
         for search in pdbs:
             # Step 1: Fetch the PDB file
             pdb_url = f"https://files.rcsb.org/download/{search}.pdb"
+            st.write(pdb_url)
             try:
                 response = requests.get(pdb_url)
+                st.write('response', response)
                 response.raise_for_status()  # Check for a successful response
             except :
                 continue  # Skip to the next PDB code if fetching fails
-                
+            st.write('response2', response)
             # Step 2: Parse the PDB file from memory
             pdb_data = response.text
+            st.write('pdb_data', pdb_data)
             pdb_parser = PDBParser(QUIET=True)  # QUIET=True suppresses warnings
             pdb_file_content = StringIO(pdb_data)
             structure = pdb_parser.get_structure(pdb_code, pdb_file_content)
             ppb = PPBuilder()
             for model in structure:
+                st.write(model)
                 for pp in ppb.build_peptides(model):
                     sequence = pp.get_sequence()
                 for chain in model: