Update app.py

app.py

@@ -28,7 +28,7 @@ class Node():
     def __init__(self, name):
         self.name = name
         self.parent = None
-        self.children = []
+        self.children = [] # list of type Node
 
     def add_child(child):
             self.children.append(child)
@@ -36,7 +36,16 @@ class Node():
 
 name_to_node = {}
 
-for species, imgpath in species_to_imgpath.items():
+def get_root(node):
+    root = node
+    while node:
+        root = node
+        node = node.parent
+
+    return root
+
+def get_tree(imgpath):
+    
     for foldername in os.listdir(imgpath):
         if os.path.isdir(os.path.join(imgpath, foldername)):
             folderpath = os.path.join(imgpath, foldername)
@@ -60,7 +69,8 @@ for species, imgpath in species_to_imgpath.items():
 
             node.children = child_nodes
             
-            # To be finished
+        # To be finished
+    return get_root(node)
             
 
         
@@ -68,7 +78,6 @@ for species, imgpath in species_to_imgpath.items():
 
 def display_tree():
     # This function should create and return a Plotly figure of the tree
-    # Currently returns a simple string, but should be replaced with actual graph
 
     # Define the nodes and edges for the graph
     nodes = ['Node 1', 'Node 2', 'Node 3', 'Node 4']