Update report.py

report.py

@@ -60,13 +60,27 @@ 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
 
 class ReportGenerator:
     def __init__(self):
         self.styles = getSampleStyleSheet()
         self.styles.add(ParagraphStyle(name='Justify', alignment=TA_JUSTIFY))
 
-    # ... (other methods remain the same)
+    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 Statistics"))
+        elements.extend(self._add_chart(student_metrics_fig, "Student Metrics"))
+        elements.extend(self._add_recommendations(recommendations))
+        
+        doc.build(elements)
+        buffer.seek(0)
+        return buffer
 
     def _add_chart(self, fig, title):
         elements = []
@@ -105,4 +119,27 @@ class ReportGenerator:
         
         elements.append(img)
         elements.append(Spacer(1, 12))
+        return elements
+
+    def _add_recommendations(self, recommendations):
+        elements = []
+        elements.append(Paragraph("AI Recommendations", self.styles['Heading1']))
+        
+        # Convert markdown to HTML
+        html = markdown.markdown(recommendations)
+        
+        # Parse HTML and convert to ReportLab paragraphs
+        root = ET.fromstring(html)
+        for elem in root.iter():
+            if elem.tag == 'h3':
+                elements.append(Paragraph(elem.text, self.styles['Heading3']))
+            elif elem.tag == 'h4':
+                elements.append(Paragraph(elem.text, self.styles['Heading4']))
+            elif elem.tag == 'p':
+                elements.append(Paragraph(ET.tostring(elem, encoding='unicode', method='text'), self.styles['Justify']))
+            elif elem.tag == 'ul':
+                for li in elem.findall('li'):
+                    bullet_text = '• ' + ET.tostring(li, encoding='unicode', method='text').strip()
+                    elements.append(Paragraph(bullet_text, self.styles['BodyText']))
+        
         return elements