Update report.py

report.py

@@ -62,6 +62,7 @@ from reportlab.lib.units import inch
 import matplotlib.pyplot as plt
 import markdown
 from xml.etree import ElementTree as ET
+from PIL import Image as PILImage
 
 class ReportGenerator:
     def __init__(self):
@@ -97,13 +98,15 @@ class ReportGenerator:
             raise ValueError(f"Unsupported figure type: {type(fig)}")
         
         img_buffer.seek(0)
-        img = Image(img_buffer)
+        
+        # Use PIL to get image dimensions
+        with PILImage.open(img_buffer) as img:
+            img_width, img_height = img.size
         
         # Calculate width and height to maintain aspect ratio
         max_width = 6.5 * inch  # Maximum width (letter width is 8.5 inches, leaving margins)
         max_height = 4 * inch   # Maximum height
         
-        img_width, img_height = img.getSize()
         aspect = img_width / float(img_height)
         
         if img_width > max_width:
@@ -114,8 +117,11 @@ class ReportGenerator:
             img_height = max_height
             img_width = img_height * aspect
         
-        img.drawWidth = img_width
-        img.drawHeight = img_height
+        # Reset buffer position
+        img_buffer.seek(0)
+        
+        # Create ReportLab Image with calculated dimensions
+        img = Image(img_buffer, width=img_width, height=img_height)
         
         elements.append(img)
         elements.append(Spacer(1, 12))