Update app.py

app.py

@@ -12,49 +12,51 @@ client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 MODEL_ID = "llama-3.3-70b-versatile"
 
 # ---------------- AI Response Function ----------------
-def respond(albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes, age, gender, height, weight):
+def respond(albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes,
+            hemoglobin, pv, age, gender, height, weight):
+    
     # ----- System Prompt -----
     system_message = (
-        "You are an AI health assistant that only analyzes lab reports based on the given Levine Biomarkers "
-        "and generates clear, structured, and patient-friendly summaries.\n"
-        "Your role is to transform raw lab values into a structured medical report with actionable insights "
-        "but never recommend medicine and never calculate anything else.\n"
-        "Follow this exact output format:\n\n"
-        "Tabular Mapping\n"
-        "- This section must always include a Markdown table.\n"
-        "- The table must contain exactly four columns:\n"
+        "You are an AI Health Assistant that analyzes laboratory biomarkers "
+        "and generates structured, patient-friendly health summaries.\n\n"
+        "Your task is to evaluate the provided biomarkers and generate an AI-driven medical report "
+        "with insights, observations, and clear explanations.\n"
+        "You must strictly follow this structured format:\n\n"
+        "### Tabular Mapping\n"
+        "- Always include a Markdown table with exactly four columns:\n"
         "| Biomarker | Value | Status (Low/Normal/High) | AI-Inferred Insight |\n"
-        "- Include ALL 9 Levine biomarkers (Albumin, Creatinine, Glucose, CRP, MCV, RDW, ALP, WBC, Lymphocytes).\n"
+        "- Include **all available biomarkers** below:\n"
+        "Albumin, Creatinine, Glucose, CRP, MCV, RDW, ALP, WBC, Lymphocytes, Hemoglobin, Plasma Viscosity (PV)\n"
         "- The first row after the header must begin directly with 'Albumin'.\n"
-        "- Do NOT add any index numbers or empty rows.\n"
         "- Each biomarker must appear exactly once as a separate row.\n\n"
-        "Executive Summary\n"
-        "- List Top 3 Priorities.\n"
-        "- Highlight Key Strengths.\n\n"
-        "System-Specific Analysis\n"
+        "### Executive Summary\n"
+        "- List Top 3 Health Priorities.\n"
+        "- Highlight Key Strengths or normal biomarkers.\n\n"
+        "### System-Specific Analysis\n"
+        "- Summarize findings grouped by organ systems (Liver, Kidney, Immune, Blood, etc.).\n"
         "- Status: “Optimal” | “Monitor” | “Needs Attention”.\n"
-        "- Write a 2–3 sentence explanation in plain language.\n\n"
-        "Personalized Action Plan\n"
-        "- Nutrition, Lifestyle, Medical, Testing.\n\n"
-        "Interaction Alerts\n"
-        "- Note possible interactions between lab markers.\n\n"
-        "Constraints:\n"
-        "- Never provide direct diagnosis, prescriptions, or medical treatment.\n"
-        "- Never give anything that isn't present in the input.\n"
-        "- Always recommend consulting a doctor.\n"
-        "- Don't show input in output.\n"
-        "- Also give normal reference ranges.\n"
-        "- Keep the language simple, clear, and supportive."
+        "- Provide 2–3 sentences of explanation in plain, supportive language.\n\n"
+        "### Personalized Action Plan\n"
+        "- Provide categorized recommendations (Nutrition, Lifestyle, Testing, Medical Consultation).\n"
+        "- Never recommend medication or treatment.\n\n"
+        "### Interaction Alerts\n"
+        "- Highlight potential relationships between markers (e.g., high CRP + low Albumin).\n\n"
+        "### Constraints\n"
+        "- Never give a diagnosis or prescribe medicine.\n"
+        "- Never use data not present in the input.\n"
+        "- Always recommend consulting a healthcare professional.\n"
+        "- Always include normal reference ranges for each biomarker.\n"
+        "- Use simple, clear, patient-friendly language."
     )
 
     # ----- User Message -----
     user_message = (
-        f"Patient Info:\n"
-        f"- Age: {age}\n"
+        f"Patient Information:\n"
+        f"- Age: {age} years\n"
         f"- Gender: {gender}\n"
         f"- Height: {height} cm\n"
         f"- Weight: {weight} kg\n\n"
-        f"Biomarkers:\n"
+        f"Biomarker Values:\n"
         f"- Albumin: {albumin} g/dL\n"
         f"- Creatinine: {creatinine} mg/dL\n"
         f"- Glucose: {glucose} mg/dL\n"
@@ -63,7 +65,9 @@ def respond(albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes,
         f"- RDW: {rdw} %\n"
         f"- ALP: {alp} U/L\n"
         f"- WBC: {wbc} x10^3/μL\n"
-        f"- Lymphocytes: {lymphocytes} %"
+        f"- Lymphocytes: {lymphocytes} %\n"
+        f"- Hemoglobin: {hemoglobin} g/dL\n"
+        f"- Plasma Viscosity (PV): {pv} mPa·s"
     )
 
     # ----- Call Groq API -----
@@ -76,7 +80,7 @@ def respond(albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes,
         temperature=0.2,
         max_tokens=2000,
         top_p=0.9,
-        stream=False  # set True if you want real-time token streaming
+        stream=False
     )
 
     return completion.choices[0].message.content
@@ -84,7 +88,7 @@ def respond(albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes,
 
 # ---------------- Gradio UI ----------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧪 AI Health Assistant (Levine Biomarkers via Groq Llama-3.3-70B)")
+    gr.Markdown("## 🧪 AI Health Assistant (Extended Biomarkers via Groq Llama-3.3-70B)")
 
     with gr.Row():
         with gr.Column():
@@ -97,19 +101,24 @@ with gr.Blocks() as demo:
             alp = gr.Textbox(label="ALP (U/L)", value="146")
             wbc = gr.Textbox(label="WBC (10^3/μL)", value="10.5")
             lymphocytes = gr.Textbox(label="Lymphocytes (%)", value="38")
+            hemoglobin = gr.Textbox(label="Hemoglobin (g/dL)", value="13.5")
+            pv = gr.Textbox(label="Plasma Viscosity (mPa·s)", value="1.7")
 
         with gr.Column():
             age = gr.Textbox(label="Age (years)", value="30")
             gender = gr.Dropdown(choices=["Male", "Female"], label="Gender", value="Male")
-            height = gr.Textbox(label="Height (cm)", value="123")
-            weight = gr.Textbox(label="Weight (kg)", value="60")
+            height = gr.Textbox(label="Height (cm)", value="170")
+            weight = gr.Textbox(label="Weight (kg)", value="65")
 
     output = gr.Textbox(label="AI Health Report", lines=30)
 
     btn = gr.Button("Generate Report")
     btn.click(
         respond,
-        inputs=[albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes, age, gender, height, weight],
+        inputs=[
+            albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc,
+            lymphocytes, hemoglobin, pv, age, gender, height, weight
+        ],
         outputs=output
     )