add legend

app.py

@@ -56,7 +56,7 @@ def molecule(input_pdb):
                 let config = { backgroundColor: "white" };
                 let viewer = $3Dmol.createViewer(element, config);
                 viewer.addModel(pdb, "pdb");
-                viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
+                viewer.getModel(0).setStyle({}, { sphere: {} })
                 viewer.addSurface($3Dmol.SurfaceType.VDW, {opacity: 0.7, color: "lightblue"});
                 viewer.zoomTo();
                 viewer.render();