Update code/add_3Dalignment.py

code/add_3Dalignment.py

@@ -229,12 +229,17 @@ def get_alignments_3D(identifier, model_num, pdb_path, pdbSequence, source, chai
             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)
             atoms = [i.split() for i in atoms]
             atoms = [i for i in atoms if (i[2] == 'CA' and i[4]  == chain)]
+            print(atoms)
+
             atomSequence = ''.join([three_to_one(i[3]) for i in atoms])
+            print(atomSequence)
             coords = [[i[6] ,i[7] ,i[8]] for i in atoms]
+            print(coords)
             resnums_for_sasa = [i[5] for i in atoms]
-            print(atoms)
+            print(resnums_for_sasa)
         elif source == 'SWISSMODEL':
             atomSequence = ''
             coords = []