Upload app.py

app.py

@@ -14,6 +14,13 @@ def analyze(
     albumin, creatinine, glucose, crp, mcv, rdw, alp,
     wbc, lymph, age, gender, height, weight
 ):
+    # Calculate BMI (hidden from user, only passed to LLM)
+    try:
+        height_m = height / 100  # cm → m
+        bmi = round(weight / (height_m ** 2), 2)
+    except Exception:
+        bmi = "N/A"
+
     # System-style instruction
     system_prompt = (
         "You are an advanced AI medical assistant. "
@@ -29,8 +36,8 @@ def analyze(
     - Gender: {gender}
     - Height: {height} cm
     - Weight: {weight} kg
+    - BMI: {bmi}
     
-
     Lab Values:
     - Albumin: {albumin} g/dL
     - Creatinine: {creatinine} mg/dL
@@ -47,7 +54,7 @@ def analyze(
 
     # Call LLM
     result = pipe(prompt, max_new_tokens=1000, do_sample=True, temperature=0.6)
-    return result[1]["generated_text"]
+    return result[0]["generated_text"]
 
 
 # Build Gradio UI
@@ -84,11 +91,12 @@ with gr.Blocks() as demo:
     analyze_btn = gr.Button("🔎 Analyze")
     output = gr.Textbox(label="AI Medical Assessment", lines=12)
 
+    # Run analysis
     analyze_btn.click(
         fn=analyze,
         inputs=[albumin, creatinine, glucose, crp, mcv, rdw, alp,
-                wbc, lymph, age, gender, height, weight, bmi],
+                wbc, lymph, age, gender, height, weight],
         outputs=output
     )
 
-demo.launch()
+demo.launch()