Update plot_pdb.py

plot_pdb.py

@@ -77,19 +77,7 @@ def get_pdb(sequence):
 def plot_struc(sequence):
 
     pdb_string, pdb_filename = get_pdb(sequence) 
-
-    '''
-    name = sequence[:3] + sequence[-3:]  #combine the firt and last 3 AAs of sequence as a filename.
-
-    outpath = (
-        Path.cwd() / f"PDB-{name}.pdb")
-    
-    with open(outpath.name, "w") as f:
-        f.write(pdb_string)
-
-    outpath_str = str(outpath)
-    '''
     
-    html_view = display_pdb_by_pdb(pdb_string, pdb_filename)
+    html_view = display_pdb_by_pdb(pdb_string, pdb_filename. color_map = "spectrum")
 
     return pdb_filename, html_view