Update app.py

app.py

@@ -1,17 +1,3 @@
-# import gradio as gr
-# import time
-
-# def update_options(selected_option):
-#     new_options = [selected_option + str(x) for x in range(1, 4)]
-#     return gr.Dropdown(choices=new_options, interactive=True)
-
-# with gr.Blocks() as interface:
-#     dropdown = gr.Dropdown(choices=["Anna", "Christine", "Phoebe"])
-#     dropdown2 = gr.Dropdown(choices=["Anna", "Christine", "Phoebe"], allow_custom_value=True)
-#     dropdown.change(update_options, dropdown, dropdown2)
-
-# interface.launch()
-
 import gradio as gr
 import plotly.graph_objects as go
 import os
@@ -90,59 +76,18 @@ def display_tree():
     fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
     return fig
 
-def display_image_based_on_dropdown_1(dropdown_value):
-    # Create a white image
-    img = Image.new('RGB', (200, 100), color='white')
-    d = ImageDraw.Draw(img)
-    
-    # Specify a font. If you have a .ttf font file you can specify its path
-    # fnt = ImageFont.truetype('/path/to/font.ttf', 40)
-    # Otherwise, we'll use a default PIL font
-    fnt = ImageFont.load_default()
-    
-    # Position the text in the center
-    text = "Placeholder"
-    textwidth, textheight = d.textsize(text, font=fnt)
-    width, height = img.size
-    x = (width - textwidth) / 2
-    y = (height - textheight) / 2
 
-    # Draw the text onto the image
-    d.text((x,y), text, font=fnt, fill='black')
-    
-    # Save the image to a file in buffer to return
-    img.save('/tmp/dummy_image.png')
-    
-    return '/tmp/dummy_image.png'
-
-def display_image_based_on_dropdown_2(dropdown_value):
-    # Create a white image
-    img = Image.new('RGB', (200, 100), color='white')
-    d = ImageDraw.Draw(img)
-    
-    # Specify a font. If you have a .ttf font file you can specify its path
-    # fnt = ImageFont.truetype('/path/to/font.ttf', 40)
-    # Otherwise, we'll use a default PIL font
-    fnt = ImageFont.load_default()
-    
-    # Position the text in the center
-    text = "Placeholder"
-    textwidth, textheight = d.textsize(text, font=fnt)
-    width, height = img.size
-    x = (width - textwidth) / 2
-    y = (height - textheight) / 2
-
-    # Draw the text onto the image
-    d.text((x,y), text, font=fnt, fill='black')
-    
-    # Save the image to a file in buffer to return
-    img.save('/tmp/dummy_image.png')
-    
-    return '/tmp/dummy_image.png'
 
 def get_protoIDs(nodename):
     return gr.Dropdown(choices=nodename_to_protoIDs[nodename], interactive=True)
 
+
+def get_image(nodename, protoID):
+    imgname = '-'.join([nodename, protoID]) + '.png'
+    filepath = imgname_to_filepath[imgname]
+    return gr.Image(filepath)
+    
+
 with gr.Blocks() as demo:
     gr.Markdown("## Interactive Tree and Image Display")
     
@@ -160,6 +105,7 @@ with gr.Blocks() as demo:
             image_output_2 = gr.Image('new_teaser (3)-1.png')
 
         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)
 
 # Initialize with placeholder images