Update main.py

main.py

@@ -313,174 +313,102 @@ Root Causes:\n{root_causes}""",
 
 # ── PDF Generator ─────────────────────────────────────────────────────────────
 
-def strip_markdown(text: str) -> str:
-    """Remove markdown bold/italic markers for plain PDF text."""
+def strip_md(text: str) -> str:
     text = re.sub(r"\*\*(.*?)\*\*", r"\1", text)
     text = re.sub(r"\*(.*?)\*", r"\1", text)
     return text
 
 
 def build_pdf(analysis: FullAnalysis, name: str, role: str) -> bytes:
-    """Generate a professional PDF report from the analysis."""
+    """Clean black-and-white PDF report."""
     buf = io.BytesIO()
+    INK   = colors.HexColor("#111827")
+    INK2  = colors.HexColor("#374151")
+    GRAY  = colors.HexColor("#6b7280")
+    LGRAY = colors.HexColor("#d1d5db")
+    W = A4[0] - 48 * mm
 
-    # ── Page setup
     doc = SimpleDocTemplate(
-        buf,
-        pagesize=A4,
-        rightMargin=20 * mm,
-        leftMargin=20 * mm,
-        topMargin=22 * mm,
-        bottomMargin=22 * mm,
+        buf, pagesize=A4,
+        rightMargin=24 * mm, leftMargin=24 * mm,
+        topMargin=24 * mm, bottomMargin=24 * mm,
     )
-
     styles = getSampleStyleSheet()
-    W = A4[0] - 40 * mm  # usable width
-
-    # ── Custom styles
-    s_title = ParagraphStyle(
-        "title",
-        parent=styles["Normal"],
-        fontSize=22,
-        fontName="Helvetica-Bold",
-        textColor=colors.HexColor("#1a1a2e"),
-        spaceAfter=4,
-    )
-    s_sub = ParagraphStyle(
-        "sub",
-        parent=styles["Normal"],
-        fontSize=11,
-        fontName="Helvetica",
-        textColor=colors.HexColor("#6b7280"),
-        spaceAfter=12,
-    )
-    s_section = ParagraphStyle(
-        "section",
-        parent=styles["Normal"],
-        fontSize=13,
-        fontName="Helvetica-Bold",
-        textColor=colors.HexColor("#2563eb"),
-        spaceBefore=16,
-        spaceAfter=6,
-    )
-    s_h3 = ParagraphStyle(
-        "h3",
-        parent=styles["Normal"],
-        fontSize=11,
-        fontName="Helvetica-Bold",
-        textColor=colors.HexColor("#374151"),
-        spaceBefore=8,
-        spaceAfter=3,
-    )
-    s_body = ParagraphStyle(
-        "body",
-        parent=styles["Normal"],
-        fontSize=10,
-        fontName="Helvetica",
-        textColor=colors.HexColor("#374151"),
-        leading=15,
-        spaceAfter=4,
-    )
-    s_label = ParagraphStyle(
-        "label",
-        parent=styles["Normal"],
-        fontSize=8,
-        fontName="Helvetica-Bold",
-        textColor=colors.white,
-    )
 
-    def agent_badge(label: str, color: str) -> Table:
-        """Small colored badge showing which agent produced this section."""
-        data = [[Paragraph(f"◉  {label}", s_label)]]
-        t = Table(data, colWidths=[W])
-        t.setStyle(
-            TableStyle(
-                [
-                    ("BACKGROUND", (0, 0), (-1, -1), colors.HexColor(color)),
-                    ("TOPPADDING", (0, 0), (-1, -1), 5),
-                    ("BOTTOMPADDING", (0, 0), (-1, -1), 5),
-                    ("LEFTPADDING", (0, 0), (-1, -1), 10),
-                    ("ROUNDEDCORNERS", [4, 4, 4, 4]),
-                ]
-            )
-        )
-        return t
-
-    def render_markdown_block(md_text: str) -> list:
-        """Convert basic markdown to ReportLab flowables."""
-        flowables = []
+    s_title = ParagraphStyle("t", parent=styles["Normal"],
+        fontSize=20, fontName="Helvetica-Bold", textColor=INK, spaceAfter=3)
+    s_meta = ParagraphStyle("m", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica", textColor=GRAY, spaceAfter=14)
+    s_section_head = ParagraphStyle("sh", parent=styles["Normal"],
+        fontSize=11, fontName="Helvetica-Bold", textColor=INK,
+        spaceBefore=20, spaceAfter=2)
+    s_section_num = ParagraphStyle("sn", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica", textColor=GRAY, spaceAfter=8)
+    s_h2 = ParagraphStyle("h2", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica-Bold", textColor=INK,
+        spaceBefore=10, spaceAfter=3)
+    s_h3 = ParagraphStyle("h3", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica-Bold", textColor=INK2,
+        spaceBefore=7, spaceAfter=2)
+    s_body = ParagraphStyle("b", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica", textColor=INK2,
+        leading=15, spaceAfter=3)
+    s_bullet = ParagraphStyle("bl", parent=styles["Normal"],
+        fontSize=10, fontName="Helvetica", textColor=INK2,
+        leading=15, spaceAfter=2, leftIndent=12, firstLineIndent=-12)
+    s_footer = ParagraphStyle("f", parent=styles["Normal"],
+        fontSize=8, fontName="Helvetica", textColor=GRAY, alignment=1)
+
+    SECTIONS = [
+        ("Problem Analyst",    analysis.problem_statement),
+        ("Root Cause Analyst", analysis.root_causes),
+        ("Solutions",          analysis.solutions),
+        ("Action Plan",        analysis.action_plan),
+        ("Thinking Coach",     analysis.thinking_feedback),
+    ]
+
+    def render_block(md_text: str) -> list:
+        out = []
         for line in md_text.splitlines():
             line = line.strip()
             if not line:
-                flowables.append(Spacer(1, 4))
-                continue
-            if line.startswith("## "):
-                flowables.append(Paragraph(line[3:], s_section))
+                out.append(Spacer(1, 3))
+            elif line.startswith("## "):
+                out.append(Paragraph(strip_md(line[3:]), s_h2))
             elif line.startswith("### "):
-                flowables.append(Paragraph(line[4:], s_h3))
+                out.append(Paragraph(strip_md(line[4:]), s_h3))
             elif line.startswith("- ") or line.startswith("* "):
-                clean = strip_markdown(line[2:])
-                flowables.append(Paragraph(f"• {clean}", s_body))
+                out.append(Paragraph(f"– {strip_md(line[2:])}", s_bullet))
             elif re.match(r"^\d+\.", line):
-                clean = strip_markdown(line)
-                flowables.append(Paragraph(clean, s_body))
-            elif line.startswith("**") and line.endswith("**"):
-                flowables.append(Paragraph(line[2:-2], s_h3))
+                out.append(Paragraph(strip_md(line), s_body))
             else:
-                clean = strip_markdown(line)
-                flowables.append(Paragraph(clean, s_body))
-        return flowables
+                out.append(Paragraph(strip_md(line), s_body))
+        return out
 
     story = []
 
-    # ── Header
-    story.append(Paragraph("Problem Analysis Report", s_title))
-    story.append(
-        Paragraph(
-            f"{name}  ·  {role}  ·  {datetime.now().strftime('%d %B %Y')}",
-            s_sub,
-        )
-    )
-    story.append(HRFlowable(width=W, thickness=1.5, color=colors.HexColor("#2563eb"), spaceAfter=10))
-
-    # ── Section 1: Problem Statement
-    story.append(agent_badge("AGENT 1  —  Problem Analyst", "#1e3a5f"))
-    story.append(Spacer(1, 6))
-    story += render_markdown_block(analysis.problem_statement)
-
-    # ── Section 2: Root Cause
-    story.append(Spacer(1, 8))
-    story.append(agent_badge("AGENT 2  —  Root Cause Analyst", "#1a4731"))
-    story.append(Spacer(1, 6))
-    story += render_markdown_block(analysis.root_causes)
-
-    # ── Section 3: Solutions
-    story.append(Spacer(1, 8))
-    story.append(agent_badge("AGENT 3  —  Solution Brainstorm", "#4a1942"))
-    story.append(Spacer(1, 6))
-    story += render_markdown_block(analysis.solutions)
-
-    # ── Section 4: Action Plan
-    story.append(Spacer(1, 8))
-    story.append(agent_badge("AGENT 4  —  Action Planner", "#7c2d12"))
-    story.append(Spacer(1, 6))
-    story += render_markdown_block(analysis.action_plan)
-
-    # ── Section 5: Thinking Coach
-    story.append(Spacer(1, 8))
-    story.append(agent_badge("AGENT 5  —  Thinking Coach", "#312e81"))
-    story.append(Spacer(1, 6))
-    story += render_markdown_block(analysis.thinking_feedback)
-
-    # ── Footer note
-    story.append(Spacer(1, 16))
-    story.append(HRFlowable(width=W, thickness=0.5, color=colors.HexColor("#e5e7eb"), spaceAfter=6))
-    story.append(
-        Paragraph(
-            "Generated by the AI Intern Problem-Solving System  ·  Confidential",
-            ParagraphStyle("footer", parent=styles["Normal"], fontSize=8, textColor=colors.HexColor("#9ca3af"), alignment=1),
-        )
-    )
+    # Header
+    story.append(Paragraph("Problem Analysis", s_title))
+    story.append(Paragraph(
+        f"{name}  ·  {datetime.now().strftime('%d %B %Y')}",
+        s_meta,
+    ))
+    story.append(HRFlowable(width=W, thickness=0.5, color=LGRAY, spaceAfter=4))
+
+    # Sections
+    for i, (title, text) in enumerate(SECTIONS, 1):
+        story.append(Spacer(1, 4))
+        story.append(Paragraph(title, s_section_head))
+        story.append(HRFlowable(width=W, thickness=0.5, color=LGRAY, spaceAfter=6))
+        story += render_block(text)
+
+    # Footer
+    story.append(Spacer(1, 20))
+    story.append(HRFlowable(width=W, thickness=0.5, color=LGRAY, spaceAfter=6))
+    story.append(Paragraph(
+        f"Problem Solver  ·  {datetime.now().strftime('%d %B %Y')}",
+        s_footer,
+    ))
 
     doc.build(story)
     buf.seek(0)