add legend

app.py

@@ -47,6 +47,7 @@ def molecule(input_pdb):
     <div id="container" class="mol-container"></div>
   
         <script>
+
         
         let pdb = `"""
         + mol
@@ -56,10 +57,7 @@ def molecule(input_pdb):
                 let config = { backgroundColor: "white" };
                 let viewer = $3Dmol.createViewer(element, config);
                 viewer.addModel(pdb, "pdb");
-                viewer.getModel(0).setStyle({}, { sphere: {} })
-                viewer.setClickable({}, true, function(atom, _viewer, _event, _container) {
-                    viewer.addLabel(atom.resn + ":" + atom.atom, { position: atom, backgroundColor: 'darkgreen', backgroundOpacity: 0.8 })
-                })
+                viewer.getModel(0).setStyle({}, {sphere:{colorscheme:"greenCarbon"}});
                 viewer.zoomTo();
                 viewer.render();
                 viewer.zoom(0.8, 2000);