Update app.py

app.py

@@ -88,3 +88,64 @@ def annotate_images(images):
         draw_canvas(canvas, state['image'], state['annotations'])
 
     #
+
+            # Define the draw annotation function
+    def draw_annotation(canvas, start_x, start_y, end_x, end_y, annotation_type):
+        canvas.clear()
+        draw_canvas(canvas, state['image'], state['annotations'])
+        width = np.abs(start_x - end_x)
+        height = np.abs(start_y - end_y)
+        if annotation_type == 'rect':
+            canvas.draw_rect(start_x, start_y, width, height, stroke_color='red')
+        elif annotation_type == 'circle':
+            radius = np.sqrt(np.power(width, 2) + np.power(height, 2)) / 2
+            center_x, center_y = start_x + width / 2, start_y + height / 2
+            canvas.draw_circle(center_x, center_y, radius, stroke_color='red')
+
+    # Define the annotation type dropdown event handler
+    def annotation_type_changed(value):
+        state['annotation_type'] = value
+
+    # Define the download annotations button click event handler
+    def download_annotations_clicked():
+        # Define the csv headers
+        headers = ['x', 'y', 'width', 'height', 'type']
+
+        # Define the csv data
+        rows = [[str(annotation.x), str(annotation.y), str(annotation.width), str(annotation.height), annotation.type]
+                for annotation in state['annotations']]
+
+        # Create the csv string
+        csv_string = StringIO()
+        csv_writer = csv.writer(csv_string)
+        csv_writer.writerow(headers)
+        for row in rows:
+            csv_writer.writerow(row)
+
+        # Download the csv file
+        b64_csv = base64.b64encode(csv_string.getvalue().encode()).decode()
+        href = f'data:text/csv;base64,{b64_csv}'
+        download_link = f'<a href="{href}" download="annotations.csv">Download Annotations CSV</a>'
+        gr.Interface.show(download_link)
+
+    # Define the interface components
+    image = gr.inputs.Image(label='Image')
+    annotation_type = gr.inputs.Dropdown(ANNOTATION_TYPES, label='Annotation Type', default=ANNOTATION_TYPES[0], onchange=annotation_type_changed)
+    download_annotations = gr.outputs.Button(label='Download Annotations', type='button', onclick=download_annotations_clicked)
+    canvas = gr.outputs.Canvas(draw_event_handlers={
+        'mousedown': canvas_mousedown,
+        'mousemove': canvas_mousemove,
+        'mouseup': canvas_mouseup
+    })
+
+    # Define the interface function
+    def annotate_images(images):
+        state['image'] = images[0]
+        draw_canvas(canvas, state['image'], state['annotations'])
+        return canvas, annotation_type, download_annotations
+
+    # Create the interface
+    interface = gr.Interface(annotate_images, inputs=image, outputs=[canvas, annotation_type, download_annotations], capture_session=True)
+
+    return interface
+