Update code/add_3Dalignment.py

code/add_3Dalignment.py

@@ -228,23 +228,13 @@ def get_alignments_3D(identifier, model_num, pdb_path, pdbSequence, source, chai
             response.raise_for_status()  # Check for a successful 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)
 
-            
+            atoms = [i.split() for i in atoms]
+            atoms = [i for i in atoms if (i[2] == 'CA' and i[4]  == chain)]  
             atomSequence = ''.join([threeToOne(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 = ''
             coords = []
@@ -278,7 +268,10 @@ def get_alignments_3D(identifier, model_num, pdb_path, pdbSequence, source, chai
         aligner.extend_gap_score = -1
         alignments = aligner.align(pdbSequence, atomSequence)
         alignments = (list(alignments))
-
+        print('IN FUNC')
+        print(alignments)
+        print(coords)
+        print(resnums_for_sasa)
         return alignments, coords, resnums_for_sasa
     elif mode==2:
             atomSequence = ''