Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -245,7 +245,7 @@ def pdb(input_set, mode, impute):
                 existing_pdb = list(glob.glob(f"{path_pdb}/*"))
                 st.write('after download 2:', existing_pdb)
 
-                st.write('NEW MWTHID')
+                st.write('NEW METHOD')
     
                 # Define the URL to retrieve the PDB file
                 url = f"https://files.rcsb.org/download/{search}.pdb"
@@ -275,16 +275,26 @@ def pdb(input_set, mode, impute):
                         repo_id="HUBioDataLab/ASCARIS",
                         repo_type="space")
                     st.write(f"PDB file {search}.pdb downloaded successfully.")
+
+
+                    # Aug 23
+                    content = response.content.decode("utf-8")  # Decode the content if it's not already a string
+                    parsed_records = list(SeqIO.parse(content, "fasta"))
+                    for rec in parsed_records:
+                        st.write(rec)
+                        st.write(rec.id)
+                        st.write(rec.pdbSequence)
                 else:
                     st.write('Here2')
                     st.write(f"Failed to retrieve PDB file for {search}.")
+                    
                 
                 st.write('what')
                 existing_pdb = list(Path(path_to_output_files / 'pdb_structures').glob("*"))
                 st.write('existing_pdb3', existing_pdb)
 
-            
-    
+                
+        
                 resolution_method = parser.get_structure(search, file)
                 for record in SeqIO.parse(file, "pdb-seqres"):
                     if record.dbxrefs[0].split(':')[0] == 'UNP':