Update utils/pdf_export.py

utils/pdf_export.py

@@ -1,19 +1,95 @@
-# genesis/utils/pdf_export.py
+import os
+from datetime import datetime
+from typing import List, Dict
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
 from reportlab.lib.styles import getSampleStyleSheet
-import os
+from reportlab.lib import colors
+
+def export_report_to_pdf(
+    query: str,
+    summary: str,
+    citations: List[Dict],
+    structures: List[Dict],
+    funding_data: List[Dict],
+    regulation_data: List[Dict],
+    filename: str = None
+) -> str:
+    """
+    Export a structured GENESIS-AI research report to a PDF file.
+
+    Args:
+        query: The original research query.
+        summary: AI-generated research summary.
+        citations: List of citations (dicts with 'url' or 'id').
+        structures: List of molecular structures info.
+        funding_data: Funding records.
+        regulation_data: Regulatory insights.
+        filename: Optional custom filename.
+    Returns:
+        str: Full path to generated PDF.
+    """
+    if not filename:
+        filename = f"genesis_report_{datetime.utcnow().strftime('%Y%m%d_%H%M%S')}.pdf"
 
-def export_to_pdf(content: str, filename: str = "output.pdf"):
-    """Export given text content to a PDF file."""
     filepath = os.path.join(os.getcwd(), filename)
     doc = SimpleDocTemplate(filepath, pagesize=letter)
     styles = getSampleStyleSheet()
     story = []
 
-    for line in content.split("\n"):
-        story.append(Paragraph(line, styles["Normal"]))
+    # Title
+    story.append(Paragraph(f"<b>GENESIS-AI Research Report</b>", styles["Title"]))
+    story.append(Spacer(1, 12))
+    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 or "No summary generated.", styles["Normal"]))
+    story.append(Spacer(1, 12))
+
+    # Citations
+    if citations:
+        story.append(Paragraph("<b>Citations</b>", styles["Heading2"]))
+        for cite in citations:
+            url = cite.get("url", "")
+            story.append(Paragraph(url, styles["Normal"]))
+        story.append(Spacer(1, 12))
+
+    # Structures
+    if structures:
+        story.append(Paragraph("<b>Structures</b>", styles["Heading2"]))
+        for struct in structures:
+            story.append(Paragraph(str(struct), styles["Normal"]))
+        story.append(Spacer(1, 12))
+
+    # Funding Data (Table)
+    if funding_data:
+        story.append(Paragraph("<b>Funding Data</b>", styles["Heading2"]))
+        table_data = [["Company", "Amount", "Investor"]]
+        for fund in funding_data:
+            table_data.append([
+                fund.get("company", ""),
+                fund.get("amount", ""),
+                ", ".join(fund.get("investors", []))
+            ])
+        table = Table(table_data, repeatRows=1)
+        table.setStyle(TableStyle([
+            ("BACKGROUND", (0,0), (-1,0), colors.grey),
+            ("TEXTCOLOR", (0,0), (-1,0), colors.whitesmoke),
+            ("ALIGN", (0,0), (-1,-1), "LEFT"),
+            ("GRID", (0,0), (-1,-1), 0.5, colors.black),
+        ]))
+        story.append(table)
+        story.append(Spacer(1, 12))
+
+    # Regulation Data
+    if regulation_data:
+        story.append(Paragraph("<b>Regulatory Insights</b>", styles["Heading2"]))
+        for reg in regulation_data:
+            story.append(Paragraph(str(reg), styles["Normal"]))
         story.append(Spacer(1, 12))
 
+    # Build PDF
     doc.build(story)
     return filepath