adding support for more chains in py3dmol plot

Author: sokrypton

AlphaFold2.ipynb

@@ -393,6 +393,7 @@
         "import glob\n",
         "import matplotlib.pyplot as plt\n",
         "from colabfold.colabfold import plot_plddt_legend\n",
+        "from colabfold.colabfold import pymol_color_list, alphabet_list\n",
         "rank_num = 1 #@param [\"1\", \"2\", \"3\", \"4\", \"5\"] {type:\"raw\"}\n",
         "color = \"lDDT\" #@param [\"chain\", \"lDDT\", \"rainbow\"]\n",
         "show_sidechains = False #@param {type:\"boolean\"}\n",
@@ -414,9 +415,9 @@
         "    view.setStyle({'cartoon': {'color':'spectrum'}})\n",
         "  elif color == \"chain\":\n",
         "    chains = len(queries[0][1]) + 1 if is_complex else 1\n",
-        "    for n,chain,color in zip(range(chains),list(\"ABCDEFGH\"),\n",
-        "                     [\"lime\",\"cyan\",\"magenta\",\"yellow\",\"salmon\",\"white\",\"blue\",\"orange\"]):\n",
-        "      view.setStyle({'chain':chain},{'cartoon': {'color':color}})\n",
+        "    for n,chain,color in zip(range(chains),alphabet_list,pymol_color_list):\n",
+        "       view.setStyle({'chain':chain},{'cartoon': {'color':color}})\n",
+        "\n",
         "  if show_sidechains:\n",
         "    BB = ['C','O','N']\n",
         "    view.addStyle({'and':[{'resn':[\"GLY\",\"PRO\"],'invert':True},{'atom':BB,'invert':True}]},\n",