Update app.py

app.py

@@ -0,0 +1,68 @@
+import json
+import skia
+
+# Define the category dictionary
+category_dict = {
+    0: "Algorithm",
+    1: "Caption",
+    2: "Equation",
+    3: "Figure",
+    4: "Footnote",
+    5: "List",
+    7: "Table",
+    8: "Text",
+    9: "Text-EQ",
+    10: "Title",
+    12: "PaperTitle",
+    13: "Code",
+    14: "Abstract"
+}
+
+# Function to draw bounding box and label on the image using Skia
+def draw_annotation(canvas, bbox, category, source_code):
+    paint = skia.Paint(Color=skia.ColorRED, Style=skia.Paint.kStroke_Style, StrokeWidth=2)
+    text_paint = skia.Paint(Color=skia.ColorRED, TextSize=12)
+    
+    # Unpack the bounding box coordinates
+    x_min, y_min, x_max, y_max = bbox
+    
+    # Draw the bounding box
+    canvas.drawRect(skia.Rect.MakeLTRB(x_min, y_min, x_max, y_max), paint)
+    
+    # Draw the category label
+    label = f"{category_dict[category]}: {source_code}"
+    canvas.drawSimpleText(label, x_min, y_min - 10, skia.Font(None, 12), text_paint)
+
+# Load the annotations from a JSON file
+def load_annotations(file_path):
+    with open(file_path, 'r') as file:
+        annotations = json.load(file)
+    return annotations
+
+# Main function to render annotations onto images using Skia
+def render_annotations(annotations, image_dir):
+    for annotation in annotations:
+        page_index = annotation['page_index']
+        image_path = f"{image_dir}/page_{page_index:04d}.jpg"
+        
+        # Load the image using Skia
+        image = skia.Image.MakeFromEncoded(skia.Data.MakeFromFileName(image_path))
+        canvas = skia.Surface(image.width(), image.height()).getCanvas()
+        canvas.drawImage(image, 0, 0)
+        
+        # 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"
+        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
+    
+    annotations = load_annotations(annotations_file)
+    render_annotations(annotations, image_directory)