Update app.py

app.py

@@ -93,31 +93,11 @@ def analyze(
     "- Creatinine: 0.7 – 1.3 mg/dL (Adult Male), 0.6 – 1.1 mg/dL (Adult Female)\n"
     "- Glucose (Fasting): 70 – 100 mg/dL (Adults)\n"
     "- CRP: 0.3 – 10 mg/L (Adults)\n"
-    "- Hemoglobin (Hb): 13.5 – 17.5 g/dL (Adult Male), 12.0 – 15.5 g/dL (Adult Female)\n"
-    "- Hematocrit (Hct): 41 – 53% (Adult Male), 36 – 46% (Adult Female)\n"
     "- MCV: 80 – 100 fL (Adults)\n"
-    "- MCH: 27 – 33 pg (Adults)\n"
-    "- MCHC: 32 – 36 g/dL (Adults)\n"
     "- RDW: 11 – 15% (Adults)\n"
-    "- Platelets: 150,000 – 450,000 /µL (Adults)\n"
     "- WBC: 4,000 – 11,000 /µL (Adults)\n"
-    "- Neutrophils: 40 – 60% (Adults)\n"
     "- Lymphocytes: 20 – 40% (Adults)\n"
-    "- Monocytes: 2 – 8% (Adults)\n"
-    "- Eosinophils: 1 – 4% (Adults)\n"
-    "- Basophils: 0.5 – 1% (Adults)\n"
-    "- Total Cholesterol: < 200 mg/dL (Desirable)\n"
-    "- HDL Cholesterol: > 40 mg/dL (Male), > 50 mg/dL (Female)\n"
-    "- LDL Cholesterol: < 100 mg/dL (Optimal)\n"
-    "- Triglycerides: < 150 mg/dL (Normal)\n"
-    "- ALT: 7 – 56 U/L (Adults)\n"
-    "- AST: 10 – 40 U/L (Adults)\n"
     "- ALP: 44 – 147 U/L (Adults)\n"
-    "- Bilirubin (Total): 0.1 – 1.2 mg/dL (Adults)\n"
-    "- Urea (BUN): 7 – 20 mg/dL (Adults)\n"
-    "- Sodium (Na+): 135 – 145 mmol/L (Adults)\n"
-    "- Potassium (K+): 3.5 – 5.0 mmol/L (Adults)\n"
-    "- Calcium (Ca2+): 8.5 – 10.5 mg/dL (Adults)\n\n"
 
     "RULES:\n"
     "- For each biomarker, compare the patient’s value strictly against the ranges above.\n"
@@ -161,12 +141,11 @@ def analyze(
     "- Highlight important biomarker interactions (e.g., MCV+RDW for anemia, CRP+WBC for inflammation).\n\n"
 
     "*Tabular Mapping*\n"
-    "- Markdown table with columns: | Biomarker | Value | Status | AI-Inferred Insight |\n"
+    "- Markdown table with columns: | Biomarker |Reference range | Value | Status | AI-Inferred Insight |\n"
     "- Include all available biomarkers in order.\n\n"
 
     "*Enhanced AI Insights & Longitudinal Risk*\n"
-    "- Provide cautious trends only if patterns justify.\n"
-    "- If all normal: 'No abnormalities detected from current biomarkers.'\n\n"
+  
 
     "STYLE:\n"
     "- Professional, concise, medically accurate.\n"
@@ -201,12 +180,17 @@ def analyze(
 
     # Generate
     # Keep generation parameters conservative for Spaces
-    gen = pipe(prompt,
-               max_new_tokens=2500,
-               do_sample=True,
-               temperature=0.4,
-               top_p=0.9,
-               return_full_text=False)
+    gen = pipe(
+        prompt,
+        max_new_tokens=2500,       # enough for executive summary
+        do_sample=False,          # greedy decoding (no randomness)
+        temperature=0.0,          # fully deterministic
+        top_p=1.0,                # consider full probability mass
+        repetition_penalty=1.0,   # neutral repetition handling
+        early_stopping=True,      # stop cleanly once done
+        return_full_text=False    # only assistant output
+    )
+
 
     # Extract generated text
     generated = gen[0].get("generated_text") or gen[0].get("text") or str(gen[0])