add legend

app.py

@@ -57,7 +57,9 @@ def molecule(input_pdb):
                 let viewer = $3Dmol.createViewer(element, config);
                 viewer.addModel(pdb, "pdb");
                 viewer.getModel(0).setStyle({}, { sphere: {} })
-                viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.7, color: "lightblue"});
+                viewer.setClickable({}, true, function(atom, _viewer, _event, _container) {
+                    viewer.addLabel(atom.resn + ":" + atom.atom, { position: atom, backgroundColor: 'darkgreen', backgroundOpacity: 0.8 })
+                })
                 viewer.zoomTo();
                 viewer.render();
                 viewer.zoom(0.8, 2000);