add legend

app.py

@@ -47,12 +47,6 @@ def molecule(input_pdb):
     <div id="container" class="mol-container"></div>
   
         <script>
-
-        var legend = '<div style="position: absolute; top: 10px; right: 10px; background-color: white; padding: 10px; border: 1px solid black;">' +
-             '<div style="background-color: red; width: 20px; height: 20px; display: inline-block;"></div> Hydrophobic<br>' +
-             '<div style="background-color: blue; width: 20px; height: 20px; display: inline-block;"></div> Hydrophilic<br>' +
-             '</div>';
-
         
         let pdb = `"""
         + mol
@@ -64,7 +58,6 @@ def molecule(input_pdb):
                 viewer.addModel(pdb, "pdb");
                 viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
                 viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.7, color: "lightblue"});
-                viewer.addCustomElement(legend);
                 viewer.zoomTo();
                 viewer.render();
                 viewer.zoom(0.8, 2000);