Update utils/pdf_export.py

utils/pdf_export.py

@@ -1,95 +1,55 @@
 import os
-from datetime import datetime
-from typing import List, Dict
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
 from reportlab.lib.styles import getSampleStyleSheet
-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_report_to_pdf(query, summary, citations, structures, funding_data, regulation_data, filename="output.pdf"):
+    """Export research report to PDF."""
     filepath = os.path.join(os.getcwd(), filename)
     doc = SimpleDocTemplate(filepath, pagesize=letter)
     styles = getSampleStyleSheet()
     story = []
 
     # Title
-    story.append(Paragraph(f"<b>GENESIS-AI Research Report</b>", styles["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 or "No summary generated.", styles["Normal"]))
+    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 cite in citations:
-            url = cite.get("url", "")
-            story.append(Paragraph(url, styles["Normal"]))
+        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 struct in structures:
-            story.append(Paragraph(str(struct), styles["Normal"]))
+        story.append(Paragraph("<b>Structures:</b>", styles["Heading2"]))
+        for s in structures:
+            story.append(Paragraph(str(s), styles["Normal"]))
         story.append(Spacer(1, 12))
 
-    # Funding Data (Table)
+    # Funding
     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(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 Data
+    # Regulation
     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(Paragraph("<b>Regulatory Info:</b>", styles["Heading2"]))
+        for r in regulation_data:
+            story.append(Paragraph(str(r), styles["Normal"]))
         story.append(Spacer(1, 12))
 
-    # Build PDF
     doc.build(story)
     return filepath