Update report.py

report.py

@@ -63,6 +63,8 @@ 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):
@@ -134,18 +136,32 @@ class ReportGenerator:
         # Convert markdown to HTML
         html = markdown.markdown(recommendations)
         
-        # Parse HTML and convert to ReportLab paragraphs
-        root = ET.fromstring(html)
-        for elem in root.iter():
+        # 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, self.styles['Heading3']))
+                elements.append(Paragraph(elem.text or "", self.styles['Heading3']))
             elif elem.tag == 'h4':
-                elements.append(Paragraph(elem.text, self.styles['Heading4']))
+                elements.append(Paragraph(elem.text or "", self.styles['Heading4']))
             elif elem.tag == 'p':
-                elements.append(Paragraph(ET.tostring(elem, encoding='unicode', method='text'), self.styles['Justify']))
+                text = ''.join(elem.itertext())
+                elements.append(Paragraph(text, self.styles['Justify']))
             elif elem.tag == 'ul':
                 for li in elem.findall('li'):
-                    bullet_text = '• ' + ET.tostring(li, encoding='unicode', method='text').strip()
+                    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