Update app.py

app.py

@@ -204,7 +204,7 @@ Medication: {d['medication']}
 # ===============================
 # PDF GENERATION
 # ===============================
-def generate_pdf(*inputs):
+'''def generate_pdf(*inputs):
 
     global last_basic_data, last_llm_report
 
@@ -280,7 +280,147 @@ def generate_pdf(*inputs):
 
     doc.build(elements)
     return file_path
+'''
+def generate_pdf(*inputs):
+
+    global last_basic_data, last_llm_report
+
+    d = last_basic_data
+    file_path = f"{d['name']}_Typhoid_Report.pdf"
+    doc = SimpleDocTemplate(file_path)
+    elements = []
+    styles = getSampleStyleSheet()
+
+    # ---- Custom Styles ----
+    heading_style = ParagraphStyle(
+        name="CustomHeading",
+        parent=styles["Heading2"],
+        fontSize=14,
+        spaceAfter=6,
+        textColor=colors.black
+    )
+
+    bullet_style = ParagraphStyle(
+        name="BulletStyle",
+        parent=styles["Normal"],
+        leftIndent=15,
+        bulletIndent=5,
+        spaceAfter=4
+    )
+
+    disclaimer_heading_style = ParagraphStyle(
+        name="DisclaimerHeading",
+        parent=styles["Heading2"],
+        textColor=colors.red,
+        fontSize=13,
+        spaceBefore=15
+    )
+
+    disclaimer_text_style = ParagraphStyle(
+        name="DisclaimerText",
+        parent=styles["Normal"],
+        fontSize=10,
+        textColor=colors.grey
+    )
+
+    # ---- LOGO ----
+    if os.path.exists(LOGO_PATH):
+        elements.append(Image(LOGO_PATH, width=1.5*inch, height=1.5*inch))
+        elements.append(Spacer(1,0.2*inch))
+
+    elements.append(Paragraph("<b>TyphoidGuard AI Clinical Report</b>", styles["Title"]))
+    elements.append(Spacer(1, 0.3 * inch))
+
+    # ---- RISK BADGE (Improved Height & Padding) ----
+    risk_color = colors.HexColor(d["color"])
+    badge_style = ParagraphStyle(
+        name="Badge",
+        parent=styles["Normal"],
+        backColor=risk_color,
+        textColor=colors.white,
+        fontSize=13,
+        leading=18,
+        leftIndent=10,
+        rightIndent=10,
+        spaceBefore=10,
+        spaceAfter=15
+    )
 
+    elements.append(Paragraph(
+        f"<b>Risk: {d['prob_pct']}% - {d['risk_label']}</b>",
+        badge_style
+    ))
+    elements.append(Spacer(1,0.3*inch))
+
+    # ---- PATIENT TABLE ----
+    table_data = [
+        ["Patient Name", d["name"]],
+        ["Age", d["age"]],
+        ["Gender", d["gender"]],
+        ["Platelet Count", d["platelets"]],
+        ["Hemoglobin", d["hb"]],
+        ["Treatment Duration", d["duration"]],
+        ["Bacteria", d["bacteria"]],
+        ["Severity", d["severity"]],
+        ["Medication", d["medication"]],
+    ]
+
+    table = Table(table_data, colWidths=[2.5*inch, 3*inch])
+    table.setStyle(TableStyle([
+        ('GRID',(0,0),(-1,-1),1,colors.grey),
+        ('FONTNAME',(0,0),(-1,-1),'Helvetica')
+    ]))
+
+    elements.append(table)
+    elements.append(Spacer(1,0.4*inch))
+
+    # ---- AGENT RECOMMENDATION ----
+    elements.append(Paragraph("<b>Agent Recommendation</b>", heading_style))
+    elements.append(Paragraph(d["recommendation"], styles["Normal"]))
+    elements.append(Spacer(1,0.4*inch))
+
+    # ---- DETAILED AI REPORT (Formatted with Headings + Bullets) ----
+    if last_llm_report:
+        elements.append(Paragraph("<b>Detailed AI Report</b>", heading_style))
+        elements.append(Spacer(1,0.2*inch))
+
+        lines = last_llm_report.split("\n")
+
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+
+            # Detect headings (e.g., lines ending with ":" )
+            if line.endswith(":"):
+                elements.append(Spacer(1,0.2*inch))
+                elements.append(Paragraph(f"<b>{line}</b>", heading_style))
+            else:
+                elements.append(Paragraph(f"• {line}", bullet_style))
+
+    elements.append(Spacer(1,0.5*inch))
+
+    # ---- DISCLAIMER ----
+    elements.append(Paragraph("Disclaimer", disclaimer_heading_style))
+    elements.append(Spacer(1,0.1*inch))
+    elements.append(Paragraph(
+        "Remember to consult a qualified doctor for personalized advice and treatment. "
+        "They will be able to provide guidance tailored to the patient's specific needs and circumstances.",
+        disclaimer_text_style
+    ))
+
+    elements.append(Spacer(1,0.5*inch))
+
+    # ---- TEAM ----
+    elements.append(Paragraph("<b>Team Chinar AI (AJK)</b>", styles["Normal"]))
+    elements.append(Paragraph(
+        "Anees Qumar Abbasi | Sharafat Hussain | Salma Asghar | "
+        "Kaleem Hussain | Munazza Zahra | Kokub Khurishid",
+        styles["Normal"]
+    ))
+
+    doc.build(elements)
+    return file_path
 
 # ===============================
 # INTERFACE