Update app.py

app.py

@@ -22,6 +22,38 @@ for species, imgpath in species_to_imgpath.items():
                     nodename = filename.split('.')[0].split('-')[0]
                     protoID = filename.split('.')[0].split('-')[1]
                     nodename_to_protoIDs[nodename].append(protoID)
+
+
+class Node():
+    def __init__(self, name):
+        self.name = name
+        self.children = []
+
+    def add_child(child):
+            self.children.append(child)
+
+
+name_to_node = {}
+
+for species, imgpath in species_to_imgpath.items():
+    for foldername in os.listdir(imgpath):
+        if os.path.isdir(os.path.join(imgpath, foldername)):
+            folderpath = os.path.join(imgpath, foldername)
+            node_name = foldername
+            child_names = [filename.split('.')[0].split('-')[0] for filename in os.listdir(folderpath)]
+
+            if node_name in name_to_node:
+                node = name_to_node
+            # 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)
+
+        
                     
 
 def display_tree():