| import gradio as gr | |
| # from gradio_image_prompter import ImagePrompter | |
| from typing import List | |
| from shapely.geometry import Point | |
| from boxes_define import gdf | |
| # Function to find the matching bounding box for a given point and return the image with boxes | |
| def find_bounding_box(evt: gr.SelectData, img): | |
| x, y = evt.index[0], evt.index[1] | |
| point = Point(x, y) | |
| match = gdf[gdf.contains(point)] | |
| if not match.empty: | |
| matched_box = match.iloc[0]['name'] | |
| else: | |
| matched_box = "No bounding box found" | |
| return matched_box | |
| # Gradio app | |
| with gr.Blocks() as demo: | |
| # Draw bounding boxes on the image | |
| img_with_boxes = gr.Image(value='assets/images/bird_boxed.png', label=None) | |
| bbox_label = gr.Label() | |
| img_with_boxes.select(find_bounding_box, | |
| inputs=[img_with_boxes], | |
| outputs=[bbox_label]) | |
| demo.launch(server_name="0.0.0.0") | |