Filter Tree

[eladbej]

Hi !
There is a smart way to filter dynamically the tree from some nodes?

[dkapur17]

Hi @eladbej. Could you provide an example?

[eladbej]

I need to do something like that but to the same tree object.
Currently the code I do it with not working for the middle nodes:

from streamlit_flow import streamlit_flow
from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
from streamlit_flow.state import StreamlitFlowState
from streamlit_flow.layouts import TreeLayout
import streamlit as st

counter = 0
def update_streamlit_state(state: StreamlitFlowState, counter=None):
    updated_state = streamlit_flow(state=state,
                                   key=f'tree_layout_{counter}',
                                   fit_view=True,
                                   get_node_on_click=True,
                                   get_edge_on_click=True)

    return updated_state


def edges_wo_selected(edges, node):
    for edge in edges:
        if edge.source == node.id or edge.target == node.id:
            edges.remove(edge)
    return edges


nodes = [
    StreamlitFlowNode(id='1', pos=(100, 100), data={'content': 'Node 1'}, node_type='input', source_position='right',
                      draggable=False),
    StreamlitFlowNode('2', (350, 50), {'content': 'Node 2'}, 'default', 'right', 'left', draggable=False),
    StreamlitFlowNode('3', (350, 150), {'content': 'Node 3'}, 'default', 'right', 'left', draggable=False),
    StreamlitFlowNode('4', (600, 100), {'content': 'Node 4'}, 'output', target_position='left', draggable=False)]

edges = [StreamlitFlowEdge('1-2', '1', '2', animated=True),
         StreamlitFlowEdge('1-3', '1', '3', animated=True),
         StreamlitFlowEdge('2-4', '2', '4', animated=True),
         StreamlitFlowEdge('3-4', '3', '4', animated=True)]

state = StreamlitFlowState(nodes, edges)

update_streamlit_state(state,counter)

node_names_list = [node.data['content'] for node in nodes]
node_names_list.insert(0,'None')

select_box = st.selectbox('Select a node to filter:', node_names_list)
filter_button = st.button('Filter', key='filter_button')
nodes_tmp = nodes
edges_tmp = edges
try:
    node = [node for node in nodes if node.data['content'] == select_box][0]
    nodes_tmp = filter(lambda x: x.id != node.id, nodes)
    edges_tmp = edges_wo_selected(edges, node)
except Exception as e:
    st.write('No node selected')
    st.write(e)


if filter_button:
    counter = counter + 1
    update_streamlit_state(StreamlitFlowState(nodes_tmp, edges_tmp),counter)

Screen.Recording.2025-03-03.at.11.00.15.mov