Create report_gen.py

report_gen.py

@@ -0,0 +1,44 @@
+from io import BytesIO
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.units import mm
+
+def generate_pdf(pages, answers):
+    buffer = BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=A4, rightMargin=20*mm, leftMargin=20*mm, topMargin=20*mm, bottomMargin=20*mm)
+
+    # Define custom colors
+    dark_purple = colors.Color(0.2, 0, 0.4)  # Very dark purple for titles
+    light_purple = colors.Color(0.6, 0.4, 0.8)  # Lighter purple for answers
+    medium_gray = colors.Color(0.5, 0.5, 0.5)  # Medium gray for descriptions
+
+    styles = getSampleStyleSheet()
+    styles.add(ParagraphStyle(name='SectionTitle', fontName='Helvetica', fontSize=16, spaceAfter=6, textColor=dark_purple))
+    styles.add(ParagraphStyle(name='Description', fontName='Helvetica', fontSize=12, spaceAfter=6, textColor=medium_gray))
+    styles.add(ParagraphStyle(name='Answer', fontName='Helvetica', fontSize=12, spaceAfter=12, textColor=light_purple))
+    styles.add(ParagraphStyle(name='MainTitle', fontName='Helvetica-Bold', fontSize=16, spaceAfter=24, textColor=dark_purple))
+
+    story = [Paragraph("AI Trust and Opacity Evaluation", styles['MainTitle'])]
+
+    for page in pages[:-1]:  # Skip the last page
+        if 'input_key' in page and page['input_key'] is not None:
+            story.append(Paragraph(page['title'], styles['SectionTitle']))
+            story.append(Paragraph(page['content'], styles['Description']))
+
+            answer = answers.get(page['input_key'], "")
+            if isinstance(answer, list):
+                answer = ', '.join(answer)
+
+            if page.get('input_type') == 'combined':
+                option = answers.get(page['input_key'], "")
+                conclusion = answers.get(f"{page['input_key']}_conclusion", "")
+                answer = f"Option selected: {option}\n\nConclusion: {conclusion}"
+
+            story.append(Paragraph(f"<b>Answer:</b> {answer}", styles['Answer']))
+            story.append(Spacer(1, 12))
+
+    doc.build(story)
+    buffer.seek(0)
+    return buffer