Update app.py

app.py

@@ -16,17 +16,17 @@ species_to_imgpath = {'bird': './descendent_specific_topk_heatmap_withbb_ep=last
 imgname_to_filepath = {} # this ignores the extension such as .png
 nodename_to_protoIDs = defaultdict(list)
 
-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)
-            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)
+# 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)
+#             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)
 
 
 class Node():
@@ -291,6 +291,19 @@ def get_protoIDs(nodename):
     return gr.Dropdown(choices=nodename_to_protoIDs[nodename], interactive=True)
 
 def get_nodenames(species_name):
+    global nodename_to_protoIDs
+    imgpath = species_to_imgpath[species_name]
+    for foldername in os.listdir(imgpath):
+        if os.path.isdir(os.path.join(imgpath, foldername)):
+            folderpath = os.path.join(imgpath, foldername)
+            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)
+                    
     return gr.Dropdown(choices=list(nodename_to_protoIDs.keys()), interactive=True)
 
 def get_image(nodename, protoID):
@@ -299,7 +312,7 @@ def get_image(nodename, protoID):
     return gr.Image(filepath)
 
 def species_change(species_name):
-    set_nodename_to_protoIDs(species_name)
+    # set_nodename_to_protoIDs(species_name)
     return [plot_tree_from_species(species_name), get_nodenames(species_name), get_nodenames(species_name)]
 
 with gr.Blocks() as demo:
@@ -342,7 +355,7 @@ with gr.Blocks() as demo:
             dropdown_2_protos = gr.Dropdown(label="Select a prototype ID", choices=[], allow_custom_value=True)
             image_output_2 = gr.Image()
     # dropdown_species.change(plot_tree_from_species, dropdown_species, tree_output)
-    dropdown_species.change(species_change, dropdown_species, [tree_output, dropdown_1_nodename, dropdown_1_nodename])
+    dropdown_species.change(species_change, dropdown_species, [tree_output, dropdown_1_nodename, dropdown_2_nodename])
     # dropdown_species.change(set_nodename_to_protoIDs)
     # dropdown_species.change(get_nodenames, dropdown_species, dropdown_1_nodename)
     # dropdown_species.change(get_nodenames, dropdown_species, dropdown_2_nodename)