Create app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 
 from reportlab.platypus import (
     SimpleDocTemplate, Paragraph, Spacer,
-    Table, TableStyle, Image
+    Table, TableStyle
 )
 from reportlab.lib.styles import ParagraphStyle, getSampleStyleSheet
 from reportlab.lib import colors
@@ -15,14 +15,14 @@ from reportlab.lib.units import inch
 # ===============================
 # LOAD MODEL
 # ===============================
-model = joblib.load("typhoidguard_model_reduced2.pkl")
+model = joblib.load("typhoidguard_model_reduced.pkl")
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 last_llm_report = ""
 last_basic_data = {}
 
 # ===============================
-# LLM FUNCTION (UNCHANGED)
+# LLM FUNCTION
 # ===============================
 def generate_llm_report(patient_info, prob_pct, risk_label, recommendation, factors):
 
@@ -62,7 +62,7 @@ End with reminder to consult qualified doctor.
 
 
 # ===============================
-# BASIC PREDICTION (UNCHANGED)
+# BASIC PREDICTION
 # ===============================
 def predict_basic(name, age, gender, platelets, hb, duration,
                   blood_bacteria, severity, medication):
@@ -124,23 +124,78 @@ def predict_basic(name, age, gender, platelets, hb, duration,
         "prob_pct": prob_pct,
         "risk_label": risk_label,
         "recommendation": recommendation,
-        "factors": factors,
-        "color": color
+        "factors": factors
     }
 
+    urdu = f"""
+    <div style='background:#eef3fb;padding:10px;border-radius:8px;'>
+    <b>اردو خلاصہ:</b><br>
+    علاج ناکامی کا خطرہ <b>{prob_pct}%</b> ہے۔ درجہ: <b>{risk_label}</b>۔<br>
+    ڈاکٹر سے مشورہ ضرور کریں۔
+    </div>
+    """
+
     return f"""
     <div style="font-family:Arial;padding:15px;">
     <h2 style="color:{color};">Patient: {name}</h2>
     <h3>Risk of Treatment Failure: {prob_pct}%</h3>
     <h3>Status: {risk_label}</h3>
+
+    <h4>Clinical Risk Factors:</h4>
+    <ul>
+    {''.join([f"<li>{f}</li>" for f in factors]) if factors else "<li>None</li>"}
+    </ul>
+
     <h4>Agent Recommendation:</h4>
     <p><b>{recommendation}</b></p>
+
+    {urdu}
     </div>
     """
 
 
 # ===============================
-# PDF GENERATION (UPDATED)
+# DETAILED BUTTON
+# ===============================
+def generate_detailed(*inputs):
+    global last_basic_data, last_llm_report
+
+    basic_html = predict_basic(*inputs)
+
+    d = last_basic_data
+
+    patient_info = f"""
+Age: {d['age']}
+Gender: {d['gender']}
+Platelets: {d['platelets']}
+Hemoglobin: {d['hb']}
+Duration: {d['duration']}
+Bacteria: {d['bacteria']}
+Severity: {d['severity']}
+Medication: {d['medication']}
+"""
+
+    last_llm_report = generate_llm_report(
+        patient_info,
+        d["prob_pct"],
+        d["risk_label"],
+        d["recommendation"],
+        d["factors"]
+    )
+
+    formatted = last_llm_report.replace("\n", "<br>")
+
+    return basic_html + f"""
+    <hr>
+    <h3 style="color:#003366;">Detailed AI Report & Recommendations</h3>
+    <div style="background:#f9f9f9;padding:15px;border-radius:8px;">
+    {formatted}
+    </div>
+    """
+
+
+# ===============================
+# PDF GENERATION (PROFESSIONAL)
 # ===============================
 def generate_pdf(*inputs):
 
@@ -152,32 +207,11 @@ def generate_pdf(*inputs):
     elements = []
     styles = getSampleStyleSheet()
 
-    # ---- LOGO ----
-    if os.path.exists("logo.png"):
-        elements.append(Image("logo.png", width=1.5*inch, height=1.5*inch))
-        elements.append(Spacer(1,0.2*inch))
-
-    # ---- TITLE ----
+    # Title
     elements.append(Paragraph("<b>TyphoidGuard AI Clinical Report</b>", styles["Title"]))
     elements.append(Spacer(1, 0.3 * inch))
 
-    # ---- RISK BADGE ----
-    badge_color = colors.red if "HIGH" in d["risk_label"] else \
-                  colors.orange if "MODERATE" in d["risk_label"] else colors.green
-
-    badge = Table([[d["risk_label"]]], colWidths=[2*inch])
-    badge.setStyle(TableStyle([
-        ('BACKGROUND',(0,0),(-1,-1),badge_color),
-        ('TEXTCOLOR',(0,0),(-1,-1),colors.white),
-        ('ALIGN',(0,0),(-1,-1),'CENTER'),
-        ('FONTSIZE',(0,0),(-1,-1),14),
-        ('GRID',(0,0),(-1,-1),1,colors.black)
-    ]))
-
-    elements.append(badge)
-    elements.append(Spacer(1,0.3*inch))
-
-    # ---- PATIENT TABLE ----
+    # ---- TABLE ----
     table_data = [
         ["Patient Name", d["name"]],
         ["Age", d["age"]],
@@ -189,29 +223,39 @@ def generate_pdf(*inputs):
         ["Severity", d["severity"]],
         ["Medication", d["medication"]],
         ["Predicted Risk %", d["prob_pct"]],
+        ["Risk Level", d["risk_label"]],
     ]
 
     table = Table(table_data, colWidths=[2.5*inch, 3*inch])
     table.setStyle(TableStyle([
+        ('BACKGROUND',(0,0),(-1,0),colors.lightgrey),
         ('GRID',(0,0),(-1,-1),1,colors.grey),
-        ('BACKGROUND',(0,0),(-1,0),colors.lightgrey)
+        ('FONTNAME',(0,0),(-1,-1),'Helvetica')
     ]))
 
     elements.append(table)
     elements.append(Spacer(1,0.4*inch))
 
-    # ---- RECOMMENDATION ----
+    # ---- BASIC RECOMMENDATION ----
     elements.append(Paragraph("<b>Agent Recommendation</b>", styles["Heading2"]))
     elements.append(Spacer(1,0.2*inch))
     elements.append(Paragraph(d["recommendation"], styles["Normal"]))
     elements.append(Spacer(1,0.4*inch))
 
-    # ---- LLM ----
+    # ---- LLM SECTION ----
     if last_llm_report:
         elements.append(Paragraph("<b>Detailed AI Report & Recommendations</b>", styles["Heading2"]))
         elements.append(Spacer(1,0.2*inch))
         elements.append(Paragraph(last_llm_report.replace("\n","<br/>"), styles["Normal"]))
+        elements.append(Spacer(1,0.4*inch))
+
+    # Disclaimer
+    elements.append(Paragraph(
+        "<i>This report is AI-generated. Final medical decisions must be made by a qualified doctor.</i>",
+        styles["Normal"]
+    ))
 
+    # Team Footer
     elements.append(Spacer(1,0.5*inch))
     elements.append(Paragraph("<b>Team Chinar AI (AJK)</b>", styles["Normal"]))
     elements.append(Paragraph(
@@ -235,9 +279,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     #### Team Chinar AI (AJK)
     """)
 
-    if os.path.exists("logo.png"):
-        gr.Image("logo.png", width=150)
-
     with gr.Row():
         name = gr.Textbox(label="Patient Name")
         age = gr.Number(label="Age")
@@ -262,38 +303,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             label="Current Medication"
         )
 
-    # ---- BUTTONS (ALL COLORED) ----
+    # ---- BUTTONS ABOVE OUTPUT ----
     with gr.Row():
-        predict_btn = gr.Button("Basic Risk Assessment",
-                                elem_id="btn1")
-        detailed_btn = gr.Button("Detailed Report & Recommendations",
-                                 elem_id="btn2")
-        pdf_btn = gr.Button("Generate PDF Report",
-                            elem_id="btn3")
+        predict_btn = gr.Button("Basic Risk Assessment", variant="primary")
+        detailed_btn = gr.Button("Detailed Report & Recommendations", variant="secondary")
+        pdf_btn = gr.Button("Generate PDF Report")
 
     output = gr.HTML()
 
-    # ---- TEAM MEMBERS ON PAGE ----
-    gr.Markdown("""
-    ---
-    ### 👥 Team Members – Chinar AI
-    Anees Qumar Abbasi (Team Lead)  
-    Sharafat Hussain  
-    Salma Asghar  
-    Kaleem Hussain  
-    Munazza Zahra  
-    Kokub Khurishid  
-    """)
-
-    # ---- CUSTOM CSS ----
-    gr.HTML("""
-    <style>
-    #btn1 {background-color:#1565c0;color:white;}
-    #btn2 {background-color:#2e7d32;color:white;}
-    #btn3 {background-color:#455a64;color:white;}
-    </style>
-    """)
-
     predict_btn.click(predict_basic,
         inputs=[name,age,gender,platelets,hb,duration,
                 blood_bacteria,severity,medication],