Option to toggle ligand showing
Browse files
hexviz/🧬Attention_Visualization.py
CHANGED
@@ -24,6 +24,9 @@ chains = get_chains(structure)
|
|
24 |
selected_chains = st.sidebar.multiselect(label="Select Chain(s)", options=chains, default=st.session_state.get("selected_chains", None) or chains)
|
25 |
st.session_state.selected_chains = selected_chains
|
26 |
|
|
|
|
|
|
|
27 |
|
28 |
st.sidebar.markdown(
|
29 |
"""
|
@@ -75,8 +78,9 @@ def get_3dview(pdb):
|
|
75 |
stmol.add_hover(xyzview, backgroundColor="black", fontColor="white")
|
76 |
|
77 |
# Show all ligands as stick (heteroatoms)
|
78 |
-
|
79 |
-
|
|
|
80 |
|
81 |
hidden_chains = [x for x in chains if x not in selected_chains]
|
82 |
for chain in hidden_chains:
|
|
|
24 |
selected_chains = st.sidebar.multiselect(label="Select Chain(s)", options=chains, default=st.session_state.get("selected_chains", None) or chains)
|
25 |
st.session_state.selected_chains = selected_chains
|
26 |
|
27 |
+
show_ligands = st.sidebar.checkbox("Show ligands", value=st.session_state.get("show_ligands", True))
|
28 |
+
st.session_state.show_ligands = show_ligands
|
29 |
+
|
30 |
|
31 |
st.sidebar.markdown(
|
32 |
"""
|
|
|
78 |
stmol.add_hover(xyzview, backgroundColor="black", fontColor="white")
|
79 |
|
80 |
# Show all ligands as stick (heteroatoms)
|
81 |
+
if show_ligands:
|
82 |
+
xyzview.addStyle({"hetflag": True},
|
83 |
+
{"stick": {"radius": 0.2}})
|
84 |
|
85 |
hidden_chains = [x for x in chains if x not in selected_chains]
|
86 |
for chain in hidden_chains:
|