Update app.py

app.py

@@ -59,9 +59,12 @@ def render_annotations(annotations, image_dir):
         draw_annotation(canvas, annotation['bbox'], annotation['category'], annotation['source_code'])
         
         # Save the annotated image to a buffer
+        output_path = f"annotated_page_{page_index:04d}.png"
         surface = canvas.getSurface()
-        image_buffer = surface.makeImageSnapshot().encode(skia.kPNG_Codec)
-        annotated_images.append((image_buffer, f"Page {page_index}"))
+        # image_buffer = surface.makeImageSnapshot().encode(skia.kPNG_Codec)
+        # annotated_images.append((image_buffer, f"Page {page_index}"))
+        surface.makeImageSnapshot().save(output_path, skia.kPNG_Codec)
+        annotated_images.append(output_path)
     
     return annotated_images