Update pdf_generator.py

pdf_generator.py

@@ -1,9 +1,55 @@
 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
+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):
+        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()
@@ -85,6 +131,44 @@ def generate_pdf(session_state):
             story.append(t)
             story.append(Spacer(1, 12))
 
+            # Add quantitative criteria visualization if applicable
+            if page['input_key'] in ['technological_literacy', 'cognitive_mismatch']:
+                story.extend(process_quantitative_criteria(answer, styles))
+
     doc.build(story, onFirstPage=create_page_template, onLaterPages=create_page_template)
     buffer.seek(0)
-    return buffer
+    return buffer
+
+def process_quantitative_criteria(answer, styles):
+    story = []
+    lines = answer.split('\n')
+    for line in lines:
+        parsed = parse_quantitative_criteria(line)
+        if parsed:
+            name, value, min_val, max_val, is_percentage = parsed
+            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)
+    return story
+
+def parse_quantitative_criteria(input_string):
+    match = re.match(r'(.+):\s*([-+]?(?:\d*\.*\d+)(?:%)?)(?:\s*\[([-+]?(?:\d*\.*\d+)(?:%)?)\s*-\s*([-+]?(?:\d*\.*\d+)(?:%)?)?\])?', input_string)
+    if match:
+        name, value, min_val, max_val = match.groups()
+        name = name.strip()
+
+        # Handle percentage inputs
+        is_percentage = '%' in value or '%' in min_val or '%' in max_val
+        value = float(value.rstrip('%'))
+        min_val = float(min_val.rstrip('%') if min_val else 0)
+        max_val = float(max_val.rstrip('%') if max_val else 100)
+
+        if is_percentage:
+            value /= 100
+            min_val /= 100
+            max_val /= 100
+
+        return name, value, min_val, max_val, is_percentage
+    return None