Update app.py

app.py

@@ -51,7 +51,8 @@ Preventative Measures
 Follow-up Recommendations
 
 Keep headings clear.
-End with reminder to consult qualified doctor.
+End with this line: "Remember, this advice is generated based on the provided information and should not replace the judgment of a qualified doctor. 
+It is essential to consult a qualified doctor for personalized advice and treatment reminder to consult qualified doctor."
 """
 
     response = client.chat.completions.create(
@@ -240,83 +241,6 @@ Medication: {d['medication']}
 # ===============================
 # PDF GENERATION
 # ===============================
-'''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()
-
-    # ---- 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 ----
-    risk_color = colors.HexColor(d["color"])
-    badge_style = ParagraphStyle(
-        name="Badge",
-        parent=styles["Normal"],
-        backColor=risk_color,
-        textColor=colors.white,
-        fontSize=12,
-        leftIndent=5,
-        rightIndent=5,
-        spaceAfter=10
-    )
-
-    elements.append(Paragraph(
-        f"Risk: {d['prob_pct']}% - {d['risk_label']}",
-        badge_style
-    ))
-    elements.append(Spacer(1,0.3*inch))
-
-    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))
-
-    elements.append(Paragraph("<b>Agent Recommendation</b>", styles["Heading2"]))
-    elements.append(Paragraph(d["recommendation"], styles["Normal"]))
-    elements.append(Spacer(1,0.4*inch))
-
-    if last_llm_report:
-        elements.append(Paragraph("<b>Detailed AI Report</b>", styles["Heading2"]))
-        elements.append(Paragraph(last_llm_report.replace("\n","<br/>"), styles["Normal"]))
-
-    elements.append(Spacer(1,0.5*inch))
-    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
-'''
 def generate_pdf(*inputs):
 
     global last_basic_data, last_llm_report
@@ -415,9 +339,7 @@ def generate_pdf(*inputs):
     elements.append(Paragraph(d["recommendation"], styles["Normal"]))
     elements.append(Spacer(1,0.4*inch))
 
-    # ---- DETAILED AI REPORT (Formatted with Headings + Bullets) ----
-    # ---- DETAILED AI REPORT (Formatted with ## Headings) ----
-        
+    # ---- DETAILED AI REPORT ----
     if last_llm_report:
         elements.append(Paragraph("<b>Detailed AI Report</b>", heading_style))
         elements.append(Spacer(1,0.2*inch))
@@ -430,11 +352,9 @@ def generate_pdf(*inputs):
             if not line:
                 continue
 
-            # ❌ Remove disclaimer coming from LLM
             if "consult a qualified doctor" in line.lower():
                 continue
 
-            # ✅ Detect headings starting with ##
             if line.startswith("##"):
                 clean_heading = line.replace("##", "").strip()
                 elements.append(Spacer(1,0.2*inch))
@@ -443,7 +363,6 @@ def generate_pdf(*inputs):
                 )
             else:
                 elements.append(Paragraph(line, styles["Normal"]))
-                #elements.append(Paragraph(f"• {line}", bullet_style))
 
         elements.append(Spacer(1,0.5*inch))
 
@@ -462,13 +381,17 @@ def generate_pdf(*inputs):
     elements.append(Paragraph("<b>Team Chinar AI (AJK)</b>", styles["Normal"]))
     elements.append(Paragraph(
         "Anees Qumar Abbasi (Team Lead) | Sharafat Hussain | Salma Asghar | "
-        "Kaleem Hussain | Munazza Zahra | Kokub Khurishid",
+        "Kaleem Hussain | Munazza Zahra | Kokub Khhurishid",
         styles["Normal"]
     ))
 
     doc.build(elements)
-    return file_path
 
+    # ✅ SUCCESS POPUP MESSAGE (Added Feature)
+    
+    gr.Info("Report Generated Successfully! Please download it below.")
+
+    return file_path
 # ===============================
 # INTERFACE
 # ===============================