Update app.py

app.py

@@ -1,125 +1,171 @@
-import gradio as gr
-import plotly.graph_objects as go
-
-def display_tree():
-    # This function should create and return a Plotly figure of the tree
-    # Currently returns a simple string, but should be replaced with actual graph
-
-    # Define the nodes and edges for the graph
-    nodes = ['Node 1', 'Node 2', 'Node 3', 'Node 4']
-    edges = [(0, 1), (0, 2), (2, 3)]  # Edges are tuples of node indices
-    
-    # Define positions for the nodes (you can use a layout algorithm for more complex graphs)
-    positions = [(0, 0), (1, 2), (1, -2), (2, 0)]
-    
-    # Create traces for nodes and edges
-    edge_x = []
-    edge_y = []
-    for edge in edges:
-        x0, y0 = positions[edge[0]]
-        x1, y1 = positions[edge[1]]
-        edge_x.extend([x0, x1, None])
-        edge_y.extend([y0, y1, None])
-    
-    edge_trace = go.Scatter(
-        x=edge_x, y=edge_y,
-        line=dict(width=2, color='Black'),
-        hoverinfo='none',
-        mode='lines')
-    
-    node_x = [pos[0] for pos in positions]
-    node_y = [pos[1] for pos in positions]
-    
-    node_trace = go.Scatter(
-        x=node_x, y=node_y,
-        mode='markers+text',
-        hoverinfo='text',
-        marker=dict(showscale=False, size=10, color='Goldenrod'),
-        text=nodes,
-        textposition="top center"
-    )
-    
-    # Define the layout of the graph
-    layout = go.Layout(
-        showlegend=False,
-        hovermode='closest',
-        margin=dict(b=0, l=0, r=0, t=0),
-        xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
-    )
-    
-    # Create the figure
-    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'
+# import gradio as gr
+# import plotly.graph_objects as go
+# import os
+# from collections import defaultdict
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Interactive Tree and Image Display")
+# species_to_imgpath = {'bird': '/descendent_specific_topk=10_heatmap_withbb_ep=last_024+051'}
+
+# # this has to be there for each species
+# imgname_to_filepath = {} # this ignores the extension such as .png
+# nodename_to_protoIDs = defaultdict()
+
+# for species, imgpath in species_to_imgpath.items():
+#     for foldername in os.listdir(imgpath):
+#         if os.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 = filepath.split('.')[0].split('-')[0]
+#                     protoID = filepath.split('.')[0].split('-')[1]
+#                     nodename_to_protoIDs[nodename].append(protoID)
+                    
+
+# def display_tree():
+#     # This function should create and return a Plotly figure of the tree
+#     # Currently returns a simple string, but should be replaced with actual graph
+
+#     # Define the nodes and edges for the graph
+#     nodes = ['Node 1', 'Node 2', 'Node 3', 'Node 4']
+#     edges = [(0, 1), (0, 2), (2, 3)]  # Edges are tuples of node indices
+    
+#     # Define positions for the nodes (you can use a layout algorithm for more complex graphs)
+#     positions = [(0, 0), (1, 2), (1, -2), (2, 0)]
+    
+#     # Create traces for nodes and edges
+#     edge_x = []
+#     edge_y = []
+#     for edge in edges:
+#         x0, y0 = positions[edge[0]]
+#         x1, y1 = positions[edge[1]]
+#         edge_x.extend([x0, x1, None])
+#         edge_y.extend([y0, y1, None])
+    
+#     edge_trace = go.Scatter(
+#         x=edge_x, y=edge_y,
+#         line=dict(width=2, color='Black'),
+#         hoverinfo='none',
+#         mode='lines')
+    
+#     node_x = [pos[0] for pos in positions]
+#     node_y = [pos[1] for pos in positions]
+    
+#     node_trace = go.Scatter(
+#         x=node_x, y=node_y,
+#         mode='markers+text',
+#         hoverinfo='text',
+#         marker=dict(showscale=False, size=10, color='Goldenrod'),
+#         text=nodes,
+#         textposition="top center"
+#     )
+    
+#     # Define the layout of the graph
+#     layout = go.Layout(
+#         showlegend=False,
+#         hovermode='closest',
+#         margin=dict(b=0, l=0, r=0, t=0),
+#         xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
+#         yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
+#     )
+    
+#     # Create the figure
+#     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')
     
-    with gr.Row():
-        tree_output = gr.Plot(display_tree)  # Connect the function directly
+#     # 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'
+
+# with gr.Blocks() as demo:
+#     gr.Markdown("## Interactive Tree and Image Display")
+    
+#     with gr.Row():
+#         tree_output = gr.Plot(display_tree)  # Connect the function directly
+    
+#     with gr.Row():
+#         with gr.Column():
+#             dropdown_1_nodename = gr.Dropdown(label="Select any node name", choices=["Option 1", "Option 2", "Option 3"])
+#             dropdown_1_protos = gr.Dropdown(label="Select a prototype ID", choices=["Option 1", "Option 2", "Option 3"])
+#             image_output_1 = gr.Image('new_teaser (3)-1.png')
+#         with gr.Column():
+#             dropdown_2_nodename = gr.Dropdown(label="Select any node name", choices=["Option 1", "Option 2", "Option 3"])
+#             dropdown_2_protos = gr.Dropdown(label="Select a prototype ID", choices=["Option 1", "Option 2", "Option 3"])
+#             image_output_2 = gr.Image('new_teaser (3)-1.png')
+
+# # Initialize with placeholder images
+# # image_output_1.update(display_image_based_on_dropdown_1)
+# # image_output_2.update(display_image_based_on_dropdown_2)
+
+# demo.launch()
+
+
+import gradio as gr
+
+def update_options(selected_option):
+    # Logic to determine new options based on selected option
+    if selected_option == "Option 1":
+        new_options = ["Suboption 1.1", "Suboption 1.2"]
+    elif selected_option == "Option 2":
+        new_options = ["Suboption 2.1", "Suboption 2.2"]
+    else:
+        new_options = ["Suboption 3.1", "Suboption 3.2"]
+
+    # Return the new options to update the second dropdown
+    return new_options, selected_option  # Return current selection to persist it in the first dropdown
+
+with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column():
-            dropdown_1_nodename = gr.Dropdown(label="Select Option for Field 2", choices=["Option 1", "Option 2", "Option 3"])
-            dropdown_1_protos = gr.Dropdown(label="Select Option for Field 2", choices=["Option 1", "Option 2", "Option 3"])
-            image_output_1 = gr.Image('new_teaser (3)-1.png')
-        with gr.Column():
-            dropdown_2_nodename = gr.Dropdown(label="Select Option for Field 2", choices=["Option 1", "Option 2", "Option 3"])
-            dropdown_2_protos = gr.Dropdown(label="Select Option for Field 2", choices=["Option 1", "Option 2", "Option 3"])
-            image_output_2 = gr.Image('new_teaser (3)-1.png')
-
-# Initialize with placeholder images
-# image_output_1.update(display_image_based_on_dropdown_1)
-# image_output_2.update(display_image_based_on_dropdown_2)
+        dropdown_1 = gr.Dropdown(["Option 1", "Option 2", "Option 3"], label="Main Options")
+        dropdown_2 = gr.Dropdown([], label="Sub Options")
+
+    # When the first dropdown changes, update the options in the second dropdown
+    dropdown_1.change(fn=update_options, inputs=dropdown_1, outputs=[dropdown_2, dropdown_1])
 
 demo.launch()