Update app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 
 from reportlab.platypus import (
     SimpleDocTemplate, Paragraph, Spacer,
-    Table, TableStyle
+    Table, TableStyle, Image
 )
 from reportlab.lib.styles import ParagraphStyle, getSampleStyleSheet
 from reportlab.lib import colors
@@ -22,7 +22,7 @@ last_llm_report = ""
 last_basic_data = {}
 
 # ===============================
-# LLM FUNCTION
+# LLM FUNCTION (UNCHANGED)
 # ===============================
 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
+# BASIC PREDICTION (UNCHANGED)
 # ===============================
 def predict_basic(name, age, gender, platelets, hb, duration,
                   blood_bacteria, severity, medication):
@@ -124,78 +124,23 @@ def predict_basic(name, age, gender, platelets, hb, duration,
         "prob_pct": prob_pct,
         "risk_label": risk_label,
         "recommendation": recommendation,
-        "factors": factors
+        "factors": factors,
+        "color": color
     }
 
-    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>
     """
 
 
 # ===============================
-# 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)
+# PDF GENERATION (UPDATED)
 # ===============================
 def generate_pdf(*inputs):
 
@@ -207,11 +152,32 @@ def generate_pdf(*inputs):
     elements = []
     styles = getSampleStyleSheet()
 
-    # Title
+    # ---- 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 ----
     elements.append(Paragraph("<b>TyphoidGuard AI Clinical Report</b>", styles["Title"]))
     elements.append(Spacer(1, 0.3 * inch))
 
-    # ---- TABLE ----
+    # ---- 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_data = [
         ["Patient Name", d["name"]],
         ["Age", d["age"]],
@@ -223,39 +189,29 @@ 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),
-        ('FONTNAME',(0,0),(-1,-1),'Helvetica')
+        ('BACKGROUND',(0,0),(-1,0),colors.lightgrey)
     ]))
 
     elements.append(table)
     elements.append(Spacer(1,0.4*inch))
 
-    # ---- BASIC RECOMMENDATION ----
+    # ---- 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 SECTION ----
+    # ---- LLM ----
     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(
@@ -279,6 +235,9 @@ 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")
@@ -303,14 +262,38 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             label="Current Medication"
         )
 
-    # ---- BUTTONS ABOVE OUTPUT ----
+    # ---- BUTTONS (ALL COLORED) ----
     with gr.Row():
-        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")
+        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")
 
     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],