Update app.py

app.py

@@ -297,7 +297,10 @@ def get_image(nodename, protoID):
     imgname = '-'.join([nodename, protoID]) + '.png'
     filepath = imgname_to_filepath[imgname]
     return gr.Image(filepath)
-    
+
+def species_change(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:
 
@@ -338,10 +341,11 @@ with gr.Blocks() as demo:
             dropdown_2_nodename = gr.Dropdown(label="Select a node name", choices=[])
             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(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(set_nodename_to_protoIDs)
-    dropdown_species.change(get_nodenames, dropdown_species, dropdown_1_nodename)
-    dropdown_species.change(get_nodenames, dropdown_species, dropdown_2_nodename)
+    # dropdown_species.change(get_nodenames, dropdown_species, dropdown_1_nodename)
+    # dropdown_species.change(get_nodenames, dropdown_species, dropdown_2_nodename)
     dropdown_1_nodename.change(get_protoIDs, dropdown_1_nodename, dropdown_1_protos)
     dropdown_1_protos.change(get_image, [dropdown_1_nodename, dropdown_1_protos], image_output_1)
     dropdown_2_nodename.change(get_protoIDs, dropdown_2_nodename, dropdown_2_protos)