add seq

app.py

@@ -14,7 +14,7 @@ def read_mol(molpath):
     return mol
 
 
-def molecule(input_pdb):
+def molecule(input_pdb, h_seq, l_seq):
 
     mol = read_mol(input_pdb)
 
@@ -58,8 +58,11 @@ def molecule(input_pdb):
                 let viewer = $3Dmol.createViewer(element, config);
                 viewer.addModel(pdb, "pdb");
                 viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
+                viewer.setStyle({chain: 'H'}, {color: 'red'});
+                viewer.setStyle({chain: 'L'}, {color: 'blue'});
                 viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.4, color: "lightblue"});
-                viewer.addResLabels(proteinSequence);
+                viewer.addResLabels("""+h_seq+""", {chain: 'H'});
+                viewer.addResLabels("""+l_seq+""", {chain: 'L'});
                 viewer.zoomTo();
                 viewer.render();
                 viewer.zoom(0.8, 2000);
@@ -85,7 +88,6 @@ def molecule(input_pdb):
     </div>
     """
 
-
 def validate(seq):
     alphabet = set('ACDEFGHIKLMNPQRSTVWY')
     leftover = set(seq.upper()) - alphabet
@@ -95,6 +97,7 @@ def pred_seq(h_seq, l_seq):
     h_seq = h_seq.upper().replace(' ', '')
     l_seq = l_seq.upper().replace(' ', '')
     print(h_seq)
+    print(l_seq)
     h_is_valid = validate(h_seq)
     l_is_valid = validate(l_seq)
     if h_is_valid and l_is_valid:    
@@ -115,7 +118,7 @@ def pred_seq(h_seq, l_seq):
             do_renum=False
         )
 
-        html = molecule(pred_pdb)
+        html = molecule(pred_pdb, h_seq, l_seq)
 
     else:
         html = "<p>ERROR! Not valid sequence</p>"