Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -240,7 +240,6 @@ def pdb(input_set, mode, impute):
             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)
@@ -249,6 +248,7 @@ def pdb(input_set, mode, impute):
                 st.write(model)
                 for pp in ppb.build_peptides(model):
                     sequence = pp.get_sequence()
+                    st.write(sequence)
                 for chain in model:
                     chain_id = chain.get_id()
                     # Extract UniProt ID if available in the chain's annotations