Update app.py

app.py

@@ -125,10 +125,14 @@ def get_tree(imgpath):
 #     fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
 #     return fig
 
-def display_tree(root):
+ROOT = None
+
+def display_tree():
     nodes = []
     edges = []
     positions = {}
+
+    root = ROOT
     
     def traverse(node, depth=0, index=0):
         # This assigns a unique index to each node, which we can use for positioning
@@ -201,10 +205,10 @@ def get_image(nodename, protoID):
 with gr.Blocks() as demo:
 
     imgpath = species_to_imgpath['bird']
-    root = get_tree(imgpath)
+    ROOT = get_tree(imgpath)
     gr.Markdown("## Interactive Tree and Image Display")
     with gr.Row():
-        tree_output = gr.Plot(display_tree, root=root)  # Connect the function directly
+        tree_output = gr.Plot(display_tree)  # Connect the function directly
     
     with gr.Row():
         with gr.Column():