Commit 0f070528 authored by Ryan Morelock's avatar Ryan Morelock
Browse files

Added 'look' argument to viewer.py

parent 4040ab27

vdW_structures/viewer.py

+1 −1
Original line number Diff line number Diff line 
import py3Dmol

from pymatgen.core.periodic_table import Element 



def view_structure(structure, color_map):

def view_structure(structure, color_map, look='b'):

    view = py3Dmol.view(width=800, height=800)

    cif_str = structure.to(fmt="cif")

    view.addModel(cif_str, "cif")