pdf_generator.py

from io import BytesIO
import re
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, Table, TableStyle, Flowable
from reportlab.lib.units import mm
from reportlab.graphics.shapes import Drawing, Rect, String, Line

class SliderFlowable(Flowable):
    def __init__(self, name, value, min_val, max_val, is_percentage=False, is_integer=False):
        Flowable.__init__(self)
        self.name = name
        self.value = value
        self.min_val = min_val
        self.max_val = max_val
        self.is_percentage = is_percentage
        self.is_integer = is_integer
        self.width = 400
        self.height = 80

    def draw(self):
        drawing = Drawing(self.width, self.height)

        # Draw slider bar
        bar = Rect(50, 30, 300, 20, fillColor=colors.HexColor("#f7fbfd"), strokeColor=colors.HexColor("#9999ff"))
        drawing.add(bar)

        # Draw slider value
        if self.max_val == self.min_val:
            value_width = 50  # or some default width
        else:
            value_width = 50 + ((self.value - self.min_val) / (self.max_val - self.min_val) * 300)
        value_bar = Rect(50, 30, value_width - 50, 20, fillColor=colors.HexColor("#9999ff"), strokeColor=None)
        drawing.add(value_bar)

        # Add slider name
        drawing.add(String(0, 60, self.name, fontSize=12, fillColor=colors.HexColor("#26004d")))

        # Add range labels
        min_str = self.format_label(self.min_val)
        max_str = self.format_label(self.max_val)
        drawing.add(String(40, 10, min_str, fontSize=10, fillColor=colors.HexColor("#26004d")))
        drawing.add(String(340, 10, max_str, fontSize=10, fillColor=colors.HexColor("#26004d")))
        
        # Add value label
        value_str = self.format_label(self.value)
        drawing.add(String(value_width - 20, 55, value_str, fontSize=10, fillColor=colors.HexColor("#26004d")))

        # Add value marker
        drawing.add(Line(value_width, 25, value_width, 55, strokeColor=colors.HexColor("#26004d"), strokeWidth=2))

        drawing.drawOn(self.canv, 0, 0)

    def format_label(self, value):
        if self.is_percentage:
            return f"{value:.2f}%"
        elif self.is_integer:
            return f"{int(value)}"
        else:
            return f"{value:.2f}"

def create_styles():
    styles = getSampleStyleSheet()
    styles['Title'].fontName = 'Helvetica-Bold'
    styles['Title'].fontSize = 18
    styles['Title'].spaceAfter = 16
    styles['Title'].textColor = colors.HexColor("#26004d")

    styles['Heading1'].fontName = 'Helvetica-Bold'
    styles['Heading1'].fontSize = 16
    styles['Heading1'].spaceAfter = 10
    styles['Heading1'].textColor = colors.HexColor("#3b0b75")

    styles['Heading2'].fontName = 'Helvetica'
    styles['Heading2'].fontSize = 14
    styles['Heading2'].spaceAfter = 12
    styles['Heading2'].textColor = colors.HexColor("#52176a")

    styles['BodyText'].fontName = 'Helvetica'
    styles['BodyText'].fontSize = 12
    styles['BodyText'].spaceAfter = 12
    styles['BodyText'].textColor = colors.HexColor("#26004d")

    styles.add(ParagraphStyle(
        name='Answer',
        parent=styles['BodyText'],
        backColor=colors.HexColor("#f0f2fd"),
        borderColor=colors.HexColor("#9999ff"),
        borderWidth=0.5,
        borderPadding=(5, 5, 5, 5),
        spaceAfter=10
    ))

    return styles

def create_page_template(canvas, doc):
    canvas.saveState()
    canvas.setFillColor(colors.HexColor("#e6ebfb"))
    canvas.rect(0, 0, doc.pagesize[0], doc.pagesize[1], fill=1)
    canvas.setFillColor(colors.HexColor("#26004d"))
    canvas.setFont('Helvetica', 9)
    canvas.drawString(30, 20, f"Page {doc.page}")
    canvas.restoreState()

def generate_pdf(session_state):
    buffer = BytesIO()
    doc = SimpleDocTemplate(buffer, pagesize=A4, rightMargin=20*mm, leftMargin=20*mm, topMargin=20*mm, bottomMargin=20*mm)
    styles = create_styles()
    story = [Paragraph("AI Trust and Opacity Evaluation", styles['Title'])]

    pages = session_state.pages
    answers = session_state.answers

    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['Heading1']))
            story.append(Paragraph(page['content'], styles['BodyText']))

            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}"

            # Create a table for the answer to allow multi-line text within a block
            data = [[Paragraph(answer, styles['Answer'])]]
            t = Table(data, colWidths=[450])
            t.setStyle(TableStyle([
                ('BACKGROUND', (0, 0), (-1, -1), colors.HexColor("#f0f2fd")),
                ('TEXTCOLOR', (0, 0), (-1, -1), colors.HexColor("#26004d")),
                ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
                ('VALIGN', (0, 0), (-1, -1), 'TOP'),
                ('INNERGRID', (0, 0), (-1, -1), 0.25, colors.HexColor("#9999ff")),
                ('BOX', (0, 0), (-1, -1), 0.25, colors.HexColor("#9999ff")),
            ]))
            story.append(t)
            story.append(Spacer(1, 12))

            # Add quantitative criteria visualization if applicable
            if page['input_key'] == 'useful_assets':
                story.extend(process_quantitative_criteria(answers['useful_assets'], styles))

    doc.build(story, onFirstPage=create_page_template, onLaterPages=create_page_template)
    buffer.seek(0)
    return buffer

def process_quantitative_criteria(answers, styles):
    story = []
    quantitative_criteria = answers.get('quantitative', [])
    for i, criterion in enumerate(quantitative_criteria):
        parsed = parse_quantitative_criteria(criterion)
        if parsed:
            name, min_val, max_val, is_percentage, is_integer = parsed
            value = answers.get(f'quant_value_{i}', min_val)
            slider = SliderFlowable(name, value, min_val, max_val, is_percentage=is_percentage, is_integer=is_integer)
            story.append(slider)
            story.append(Paragraph(f"{name}: {value:.1f}%" if is_percentage else (f"{name}: {int(value)}" if is_integer else f"{name}: {value:.2f}"), styles['BodyText']))
    return story

def parse_quantitative_criteria(input_string):
    match = re.match(r'(.+?)\s*\[([-+]?(?:\d*\.*\d+)(?:%)?)\s*-\s*([-+]?(?:\d*\.*\d+)(?:%)?)\]', input_string)
    if match:
        name, min_val, max_val = match.groups()
        name = name.strip()

        def parse_value(val):
            if '%' in val:
                return float(val.rstrip('%')), True
            elif '.' in val:
                return float(val), False
            else:
                return int(val), False

        min_val, is_min_percent = parse_value(min_val)
        max_val, is_max_percent = parse_value(max_val)

        is_percentage = is_min_percent or is_max_percent
        is_integer = isinstance(min_val, int) and isinstance(max_val, int) and not is_percentage

        return name, min_val, max_val, is_percentage, is_integer
    return None