pdf_generator.py

import io
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, inch
from reportlab.graphics.shapes import Drawing, Rect, String, Line

class SliderFlowable(Flowable):
    def __init__(self, name, value, min_val, max_val, is_percentage=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.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 = f"{self.min_val:.1f}%" if self.is_percentage else f"{self.min_val:.1f}"
        max_str = f"{self.max_val:.1f}%" if self.is_percentage else f"{self.max_val:.1f}"
        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 = f"{self.value:.1f}%" if self.is_percentage else f"{self.value:.1f}"
        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 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 = io.BytesIO()
    doc = SimpleDocTemplate(buffer, pagesize=A4, rightMargin=20*mm, leftMargin=20*mm, topMargin=20*mm, bottomMargin=20*mm)
    styles = create_styles()
    story = [Paragraph("Experimental Validation Method Plan", styles['Title'])]

    for page in session_state.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']))

            if page['input_key'] == 'idea_testing':
                story.extend(process_idea_testing(session_state, styles))
            elif page['input_key'] == 'capability_testing':
                story.extend(process_capability_testing(session_state, styles))
            elif page['input_key'] == 'approach_comparison':
                story.extend(process_approach_comparison(session_state, styles))
            elif page['input_key'] == 'mock_application':
                story.extend(process_mock_application(session_state, styles))

            story.append(Spacer(1, 12))

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

def process_idea_testing(session_state, styles):
    story = []
    answers = session_state.answers['idea_testing']

    story.append(Paragraph("Rapid Prototyping", styles['Heading2']))
    for field in ['input', 'process', 'output']:
        story.append(Paragraph(f"{field.capitalize()}:", styles['Heading2']))
        story.append(Paragraph(answers['rapid_prototyping'].get(field, ""), styles['Answer']))

    story.append(Paragraph("Idea Testing", styles['Heading2']))
    story.append(Paragraph("Framework for idea validation:", styles['Heading2']))
    story.append(Paragraph(answers.get('framework', ""), styles['Answer']))
    story.append(Paragraph("Useful libraries, tools, or assets:", styles['Heading2']))
    story.append(Paragraph(answers.get('tools', ""), styles['Answer']))

    return story

def process_capability_testing(session_state, styles):
    story = []
    answers = session_state.answers['capability_testing']

    story.append(Paragraph("Capability Testing", styles['Heading2']))
    for field in ['capability', 'assessment_method', 'success_definition']:
        story.append(Paragraph(f"{field.replace('_', ' ').capitalize()}:", styles['Heading2']))
        story.append(Paragraph(answers.get(field, ""), styles['Answer']))

    story.append(Paragraph("Validation Criteria", styles['Heading2']))
    story.extend(process_validation_criteria(answers['validation_criteria'], styles))

    return story

def process_validation_criteria(session_state, styles):
    story = []

    # Qualitative Criteria
    story.append(Paragraph("Qualitative Criteria:", styles['Heading2']))
    for i, criterion in enumerate(session_state.get('qualitative', [])):
        description = session_state.get(f'qual_desc_{i}', '')
        story.append(Paragraph(f"<b>{criterion}</b>: <font face='Helvetica' size=11 color='#350863'>{description}</font>", styles['Answer']))

    # Quantitative Criteria
    story.append(Paragraph("Quantitative Criteria:", styles['Heading2']))
    for i, criterion in enumerate(session_state.get('quantitative', [])):
        parsed = parse_quantitative_criteria(criterion)
        if parsed:
            name, min_val, max_val, is_percentage, is_integer = parsed
            value = session_state.get(f'quant_value_{i}', min_val)

            if is_percentage:
                slider = SliderFlowable(name, value*100, min_val*100, max_val*100, is_percentage=True)
            else:
                slider = SliderFlowable(name, value, min_val, max_val, is_percentage=False)

            story.append(slider)
            story.append(Paragraph(f"{name}: {value:.2f}", styles['Answer']))

    return story

def process_approach_comparison(session_state, styles):
    story = []
    answers = session_state.answers['approach_comparison']

    for field in ['standardization', 'experiment_overview']:
        story.append(Paragraph(f"{field.replace('_', ' ').capitalize()}:", styles['Heading2']))
        story.append(Paragraph(answers.get(field, ""), styles['Answer']))

    return story

def process_mock_application(session_state, styles):
    story = []
    answers = session_state.answers['mock_application']

    for field in ['user_testing', 'insights', 'presentation']:
        story.append(Paragraph(f"{field.replace('_', ' ').capitalize()}:", styles['Heading2']))
        story.append(Paragraph(answers.get(field, ""), styles['Answer']))

    return story

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

        # Handle percentage inputs
        is_percentage = '%' in min_val or '%' in max_val
        min_val = float(min_val.rstrip('%'))
        max_val = float(max_val.rstrip('%'))

        if is_percentage:
            min_val /= 100
            max_val /= 100

        is_integer = '.' not in input_string or (min_val.is_integer() and max_val.is_integer())

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