Update code/add_3Dalignment.py

code/add_3Dalignment.py

@@ -226,19 +226,24 @@ def get_alignments_3D(identifier, model_num, pdb_path, pdbSequence, source, chai
             print(pdb_url)
             response = requests.get(pdb_url)
             response.raise_for_status()  # Check for a successful response
-            print(response)
             # Step 2: Parse the PDB file from memory
             atoms = [i for i in response.text.split('\n') if i.startswith('ATOM')]
+            print('ATOMS')
             print(atoms)
             atoms = [i.split() for i in atoms]
             atoms = [i for i in atoms if (i[2] == 'CA' and i[4]  == chain)]
+            print('ATOMS 2')
             print(atoms)
 
+            
             atomSequence = ''.join([three_to_one(i[3]) for i in atoms])
+            print('atomSequence')
             print(atomSequence)
             coords = [[i[6] ,i[7] ,i[8]] for i in atoms]
+            print('coords')
             print(coords)
             resnums_for_sasa = [i[5] for i in atoms]
+            print('resnums_for_sasa')
             print(resnums_for_sasa)
         elif source == 'SWISSMODEL':
             atomSequence = ''