Update app.py

app.py

@@ -27,6 +27,7 @@ for species, imgpath in species_to_imgpath.items():
 class Node():
     def __init__(self, name):
         self.name = name
+        self.parent = None
         self.children = []
 
     def add_child(child):
@@ -43,15 +44,24 @@ for species, imgpath in species_to_imgpath.items():
             child_names = [filename.split('.')[0].split('-')[0] for filename in os.listdir(folderpath)]
 
             if node_name in name_to_node:
-                node = name_to_node
+                node = name_to_node[node_name]
+            else:
+                node = Node(node_name)
+                
+            child_nodes = []
+            for child_name in child_names:
+                if child_name in name_to_node:
+                    child_node = name_to_node[child_name]
+                else:
+                    child_node = Node(child_name)
+                    name_to_node[child_name] = child_node
+                child_node.parent = node
+                child_nodes.append(child_node)
+
+            node.children = child_nodes
+            
             # To be finished
-            # for filename in os.listdir(folderpath):
-            #     if filename.endswith('png') or filename.endswith('jpg'):
-            #         filepath = os.path.join(folderpath, filename)
-            #         imgname_to_filepath[filename] = filepath
-            #         nodename = filename.split('.')[0].split('-')[0]
-            #         protoID = filename.split('.')[0].split('-')[1]
-            #         nodename_to_protoIDs[nodename].append(protoID)
+