Upload 2 files

pdf_generator.py

@@ -0,0 +1,218 @@
+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

requirements.txt

@@ -0,0 +1,7 @@
+streamlit
+pandas
+matplotlib
+base64
+streamlit-tags
+streamlit-vertical-slider
+reportlab