Update genesis/utils/pdf_export.py

genesis/utils/pdf_export.py

@@ -1,55 +1,41 @@
+# genesis/utils/pdf_export.py
 import os
 from reportlab.lib.pagesizes import letter
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
 from reportlab.lib.styles import getSampleStyleSheet
-
-def export_report_to_pdf(query, summary, citations, structures, funding_data, regulation_data, filename="output.pdf"):
-    """Export research report to PDF."""
+from reportlab.lib.utils import escape
+
+def _format_content(content):
+    """Convert string, list, or dict into clean string for PDF."""
+    if isinstance(content, str):
+        return content
+    elif isinstance(content, dict):
+        return "\n".join([f"<b>{escape(str(k))}</b>: {escape(str(v))}" for k, v in content.items()])
+    elif isinstance(content, list):
+        return "\n".join([escape(str(item)) for item in content])
+    return escape(str(content))
+
+def export_to_pdf(content, extra=None, filename="output.pdf"):
+    """Export given content to a PDF file."""
     filepath = os.path.join(os.getcwd(), filename)
     doc = SimpleDocTemplate(filepath, pagesize=letter)
     styles = getSampleStyleSheet()
     story = []
 
-    # Title
-    story.append(Paragraph(f"GENESIS-AI Research Report", styles["Title"]))
-    story.append(Spacer(1, 12))
-
-    # Query
-    story.append(Paragraph(f"<b>Query:</b> {query}", styles["Normal"]))
-    story.append(Spacer(1, 12))
-
-    # Summary
-    story.append(Paragraph("<b>Summary:</b>", styles["Heading2"]))
-    story.append(Paragraph(summary, styles["Normal"]))
-    story.append(Spacer(1, 12))
-
-    # Citations
-    if citations:
-        story.append(Paragraph("<b>Citations:</b>", styles["Heading2"]))
-        for c in citations:
-            story.append(Paragraph(f"{c.get('type', '')}: {c.get('url', '')}", styles["Normal"]))
-        story.append(Spacer(1, 12))
-
-    # Structures
-    if structures:
-        story.append(Paragraph("<b>Structures:</b>", styles["Heading2"]))
-        for s in structures:
-            story.append(Paragraph(str(s), styles["Normal"]))
+    # Main content
+    for line in _format_content(content).split("\n"):
+        story.append(Paragraph(line, styles["Normal"]))
         story.append(Spacer(1, 12))
 
-    # Funding
-    if funding_data:
-        story.append(Paragraph("<b>Funding Data:</b>", styles["Heading2"]))
-        for f in funding_data:
-            story.append(Paragraph(str(f), styles["Normal"]))
-        story.append(Spacer(1, 12))
-
-    # Regulation
-    if regulation_data:
-        story.append(Paragraph("<b>Regulatory Info:</b>", styles["Heading2"]))
-        for r in regulation_data:
-            story.append(Paragraph(str(r), styles["Normal"]))
-        story.append(Spacer(1, 12))
+    # Extra info (like citations)
+    if extra:
+        story.append(Paragraph("<b>Extra Information:</b>", styles["Heading2"]))
+        for line in _format_content(extra).split("\n"):
+            story.append(Paragraph(line, styles["Normal"]))
+            story.append(Spacer(1, 12))
 
     doc.build(story)
     return filepath
+
+# Backwards compatibility
+export_report_to_pdf = export_to_pdf