i_vorobyev commited on
Commit
ea99d62
·
1 Parent(s): fdaee7c

add legend

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -46,8 +46,15 @@ def molecule(input_pdb):
46
  <body>
47
  <div id="container" class="mol-container"></div>
48
 
49
- <script>
50
- let pdb = `"""
 
 
 
 
 
 
 
51
  + mol
52
  + """`
53
  $(document).ready(function () {
@@ -57,6 +64,7 @@ def molecule(input_pdb):
57
  viewer.addModel(pdb, "pdb");
58
  viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
59
  viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.7, color: "lightblue"});
 
60
  viewer.zoomTo();
61
  viewer.render();
62
  viewer.zoom(0.8, 2000);
 
46
  <body>
47
  <div id="container" class="mol-container"></div>
48
 
49
+ <script>
50
+
51
+ var legend = '<div style="position: absolute; top: 10px; right: 10px; background-color: white; padding: 10px; border: 1px solid black;">' +
52
+ '<div style="background-color: red; width: 20px; height: 20px; display: inline-block;"></div> Hydrophobic<br>' +
53
+ '<div style="background-color: blue; width: 20px; height: 20px; display: inline-block;"></div> Hydrophilic<br>' +
54
+ '</div>';
55
+
56
+
57
+ let pdb = `"""
58
  + mol
59
  + """`
60
  $(document).ready(function () {
 
64
  viewer.addModel(pdb, "pdb");
65
  viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
66
  viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.7, color: "lightblue"});
67
+ viewer.addCustomElement(legend);
68
  viewer.zoomTo();
69
  viewer.render();
70
  viewer.zoom(0.8, 2000);