Update app.py

app.py

@@ -1,5 +1,6 @@
 import json
 import skia
+import gradio as gr
 
 # Define the category dictionary
 category_dict = {
@@ -41,6 +42,7 @@ def load_annotations(file_path):
 
 # Main function to render annotations onto images using Skia
 def render_annotations(annotations, image_dir):
+    annotated_images = []
     for annotation in annotations:
         page_index = annotation['page_index']
         image_path = f"{image_dir}/page_{page_index:04d}.jpg"
@@ -53,16 +55,30 @@ def render_annotations(annotations, image_dir):
         # Draw the annotation on the image
         draw_annotation(canvas, annotation['bbox'], annotation['category'], annotation['source_code'])
         
-        # Save the annotated image
-        output_path = f"{image_dir}/annotated_page_{page_index:04d}.jpg"
+        # Save the annotated image to a buffer
         surface = canvas.getSurface()
-        surface.makeImageSnapshot().save(output_path, skia.kJPEG_Codec)
-        print(f"Saved annotated image to {output_path}")
-
-# Example usage
-if __name__ == "__main__":
-    annotations_file = "annotations.json"  # Path to your annotations JSON file
-    image_directory = "images"  # Directory containing the images
+        image_buffer = surface.makeImageSnapshot().encode(skia.kPNG_Codec)
+        annotated_images.append((image_buffer, f"Page {page_index}"))
     
+    return annotated_images
+
+# Gradio interface function
+def gradio_interface(annotations_file, image_dir):
     annotations = load_annotations(annotations_file)
-    render_annotations(annotations, image_directory)
+    annotated_images = render_annotations(annotations, image_dir)
+    return annotated_images
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[
+        gr.inputs.File(label="Upload Annotations JSON File"),
+        gr.inputs.Textbox(label="Image Directory")
+    ],
+    outputs=[gr.outputs.Carousel(gr.outputs.Image(type="file"), label="Annotated Images")],
+    title="Paper Annotation Renderer",
+    description="Upload an annotations JSON file and specify the image directory to render annotations onto the images."
+)
+
+# Launch the Gradio interface
+iface.launch()