Create pdf_generator.py

pdf_generator.py

@@ -0,0 +1,75 @@
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import (
+    SimpleDocTemplate,
+    Paragraph,
+    Spacer,
+)
+from reportlab.lib.styles import getSampleStyleSheet
+from io import BytesIO
+
+
+def generate_pdf(user_data, roi_data, synopsis):
+
+    buffer = BytesIO()
+
+    doc = SimpleDocTemplate(
+        buffer,
+        pagesize=letter,
+    )
+
+    styles = getSampleStyleSheet()
+
+    elements = []
+
+    title = Paragraph(
+        "<b>AI ROI Savings Report</b>",
+        styles['Title']
+    )
+
+    elements.append(title)
+    elements.append(Spacer(1, 20))
+
+    user_heading = Paragraph(
+        "<b>User Details</b>",
+        styles['Heading2']
+    )
+
+    elements.append(user_heading)
+
+    for key, value in user_data.items():
+        text = Paragraph(f"<b>{key}:</b> {value}", styles['BodyText'])
+        elements.append(text)
+
+    elements.append(Spacer(1, 20))
+
+    roi_heading = Paragraph(
+        "<b>ROI Calculation Results</b>",
+        styles['Heading2']
+    )
+
+    elements.append(roi_heading)
+
+    for key, value in roi_data.items():
+        text = Paragraph(f"<b>{key}:</b> INR {value}", styles['BodyText'])
+        elements.append(text)
+
+    elements.append(Spacer(1, 20))
+
+    synopsis_heading = Paragraph(
+        "<b>Project Synopsis</b>",
+        styles['Heading2']
+    )
+
+    elements.append(synopsis_heading)
+
+    synopsis_text = Paragraph(synopsis, styles['BodyText'])
+
+    elements.append(synopsis_text)
+
+    doc.build(elements)
+
+    pdf = buffer.getvalue()
+
+    buffer.close()
+
+    return pdf