File size: 9,568 Bytes
4f91af3 1ce8e69 9105915 4f91af3 1806672 b8415d6 2acbd28 4f91af3 de16828 766ec45 1ce8e69 41388fe 4f91af3 1ce8e69 766ec45 2acbd28 4f91af3 2acbd28 a1ba9de 2acbd28 1ce8e69 1ee1fe7 2acbd28 a1ba9de 2acbd28 4f91af3 766ec45 4f91af3 766ec45 4f91af3 2acbd28 766ec45 1ee1fe7 766ec45 1ee1fe7 766ec45 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 |
# import io
# from reportlab.lib.pagesizes import letter
# from reportlab.platypus import SimpleDocTemplate, Image, Paragraph, Spacer
# from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
# from reportlab.lib.enums import TA_JUSTIFY
# 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
# from xml.parsers.expat import ExpatError
# from html import escape
# class ReportGenerator:
# def __init__(self):
# self.styles = getSampleStyleSheet()
# self.styles.add(ParagraphStyle(name='Justify', alignment=TA_JUSTIFY))
# def create_combined_pdf(self, intervention_fig, student_metrics_fig, recommendations):
# buffer = io.BytesIO()
# doc = SimpleDocTemplate(buffer, pagesize=letter)
# elements = []
# elements.extend(self._add_chart(intervention_fig, "Intervention Dosage"))
# elements.extend(self._add_chart(student_metrics_fig, "Student Attendance and Engagement"))
# elements.extend(self._add_recommendations(recommendations))
# doc.build(elements)
# buffer.seek(0)
# return buffer
# def _add_chart(self, fig, title):
# elements = []
# elements.append(Paragraph(title, self.styles['Heading2']))
# img_buffer = io.BytesIO()
# if hasattr(fig, 'write_image'): # Plotly figure
# fig.write_image(img_buffer, format="png", width=700, height=400)
# elif isinstance(fig, plt.Figure): # Matplotlib figure
# fig.set_size_inches(10, 6) # Set a consistent size
# fig.savefig(img_buffer, format='png', dpi=100, bbox_inches='tight')
# plt.close(fig)
# else:
# raise ValueError(f"Unsupported figure type: {type(fig)}")
# img_buffer.seek(0)
# # 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
# aspect = img_width / float(img_height)
# if img_width > max_width:
# img_width = max_width
# img_height = img_width / aspect
# if img_height > max_height:
# img_height = max_height
# img_width = img_height * aspect
# # 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))
# return elements
# def _add_recommendations(self, recommendations):
# elements = []
# elements.append(Paragraph("MTSS.ai Analysis", self.styles['Heading1']))
# # Convert markdown to HTML
# html = markdown.markdown(recommendations)
# # Wrap the HTML in a root element to ensure valid XML
# wrapped_html = f"<root>{html}</root>"
# try:
# root = ET.fromstring(wrapped_html)
# except ExpatError:
# # If parsing fails, fallback to treating the entire content as plain text
# elements.append(Paragraph(escape(recommendations), self.styles['BodyText']))
# return elements
# for elem in root:
# if elem.tag == 'h3':
# elements.append(Paragraph(elem.text or "", self.styles['Heading3']))
# elif elem.tag == 'h4':
# elements.append(Paragraph(elem.text or "", self.styles['Heading4']))
# elif elem.tag == 'p':
# text = ''.join(elem.itertext())
# elements.append(Paragraph(text, self.styles['Justify']))
# elif elem.tag == 'ul':
# for li in elem.findall('li'):
# bullet_text = '• ' + ''.join(li.itertext()).strip()
# elements.append(Paragraph(bullet_text, self.styles['BodyText']))
# else:
# # For any other tags, just extract the text
# text = ''.join(elem.itertext())
# if text.strip():
# elements.append(Paragraph(text, self.styles['BodyText']))
# return elements
import io
from reportlab.lib.pagesizes import letter
from reportlab.platypus import SimpleDocTemplate, Image, Paragraph, Spacer
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.enums import TA_JUSTIFY, TA_LEFT
from reportlab.lib.units import inch
import matplotlib.pyplot as plt
import markdown
from bs4 import BeautifulSoup
from PIL import Image as PILImage
from reportlab.lib.colors import black
class ReportGenerator:
def __init__(self):
self.styles = getSampleStyleSheet()
self.styles['BodyText'].alignment = TA_JUSTIFY
self.styles['Bullet'].leftIndent = 20
self.styles['Bullet'].firstLineIndent = 0
self.styles['Bullet'].alignment = TA_LEFT
self.styles['Heading1'].fontSize = 18
self.styles['Heading2'].fontSize = 16
self.styles['Heading3'].fontSize = 14
self.styles['Heading4'].fontSize = 12
# Add a new style for bold text
self.styles.add(ParagraphStyle(name='Bold', parent=self.styles['BodyText'], fontName='Helvetica-Bold'))
def create_combined_pdf(self, intervention_fig, student_metrics_fig, recommendations):
buffer = io.BytesIO()
doc = SimpleDocTemplate(buffer, pagesize=letter, topMargin=0.5*inch, bottomMargin=0.5*inch)
elements = []
elements.extend(self._add_chart(intervention_fig, "Intervention Dosage"))
elements.extend(self._add_chart(student_metrics_fig, "Student Attendance and Engagement"))
elements.extend(self._add_recommendations(recommendations))
doc.build(elements)
buffer.seek(0)
return buffer
def _add_chart(self, fig, title):
elements = []
elements.append(Paragraph(title, self.styles['Heading2']))
img_buffer = io.BytesIO()
if hasattr(fig, 'write_image'): # Plotly figure
fig.write_image(img_buffer, format="png", width=700, height=400)
elif isinstance(fig, plt.Figure): # Matplotlib figure
fig.set_size_inches(10, 6) # Set a consistent size
fig.savefig(img_buffer, format='png', dpi=100, bbox_inches='tight')
plt.close(fig)
else:
raise ValueError(f"Unsupported figure type: {type(fig)}")
img_buffer.seek(0)
# 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
aspect = img_width / float(img_height)
if img_width > max_width:
img_width = max_width
img_height = img_width / aspect
if img_height > max_height:
img_height = max_height
img_width = img_height * aspect
# 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))
return elements
def _add_recommendations(self, recommendations):
elements = []
elements.append(Paragraph("MTSS.ai Analysis", self.styles['Heading1']))
html = markdown.markdown(recommendations)
soup = BeautifulSoup(html, 'html.parser')
for element in soup.find_all(['h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'p', 'ul']):
if element.name.startswith('h'):
level = int(element.name[1])
style = f'Heading{min(level, 4)}'
elements.append(Paragraph(element.text, self.styles[style]))
elif element.name == 'p':
elements.append(self._create_paragraph_with_inline_styles(element))
elif element.name == 'ul':
for li in element.find_all('li'):
bullet_text = '• ' + self._get_text_with_inline_styles(li)
elements.append(Paragraph(bullet_text, self.styles['Bullet']))
elements.append(Spacer(1, 6))
return elements
def _create_paragraph_with_inline_styles(self, element):
text = self._get_text_with_inline_styles(element)
return Paragraph(text, self.styles['BodyText'])
def _get_text_with_inline_styles(self, element):
text = ""
for child in element.children:
if isinstance(child, str):
text += child
elif child.name in ['strong', 'b']:
text += f'<b>{child.text}</b>'
elif child.name in ['em', 'i']:
text += f'<i>{child.text}</i>'
elif child.name == 'u':
text += f'<u>{child.text}</u>'
return text |