Upload app.py with huggingface_hub

app.py

@@ -1,57 +1,76 @@
 import gradio as gr
+import os
+from groq import Groq
+
+GROQ_KEY = os.environ.get("GROQ_API_KEY", "")
+client = Groq(api_key=GROQ_KEY)
+
+KNOWHOW = """
+CardioLab SJSU Know-How:
+MCL: Sylgard 184 at 10:1 ratio, 48hr cure, green laser PIV, 70bpm 5L/min flow
+TGT: Arduino Uno + Stepper Motor, 150mL blood, sample at 0/20/40/60min, measure TAT PF1.2 hemolysis platelets
+uPAD: Jaffe reaction creatinine + picric acid = orange-red, normal <1.2 mg/dL, CKD >1.5 mg/dL
+FSI: COMSOL ALE mesh, blood density 1060 kg/m3, viscosity 0.0035 Pa.s, St Jude geometry
+MHV: 27mm SJM Regent, bileaflet, trileaflet, monoleaflet, pediatric designs
+CKD: Stages 1-5 by creatinine, uPAD image RGB analysis 64x64 pixel
+"""
+
+def ask_biomni(question):
+    try:
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[
+                {"role": "system", "content": "You are CardioLab AI built on Biomni from Stanford SNAP Lab. Expert in SJSU Biomedical Engineering research. Always give detailed accurate answers.\n" + KNOWHOW},
+                {"role": "user", "content": question}
+            ],
+            max_tokens=500
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return "Error: " + str(e)
 
 def piv_tool(velocity, shear, hr):
     v = "HIGH - stenosis risk" if float(velocity) > 2.0 else "NORMAL"
-    s = "HIGH - thrombosis risk" if float(shear) > 10 else "NORMAL"
-    return "Velocity: " + str(velocity) + " m/s - " + v + "\nShear: " + str(shear) + " Pa - " + s
+    s = "HIGH - thrombosis risk" if float(shear) > 10 else "ELEVATED - monitor" if float(shear) > 5 else "NORMAL"
+    return "Velocity: " + str(velocity) + " m/s - " + v + "\nShear Stress: " + str(shear) + " Pa - " + s + "\nHeart Rate: " + str(hr) + " bpm"
 
 def tgt_tool(tat, pf12, hemo, platelets, time):
     risk = sum([float(tat)>15, float(pf12)>2.0, float(hemo)>50, float(platelets)<150])
     overall = "HIGH THROMBOGENIC RISK" if risk>=3 else "MODERATE RISK" if risk>=2 else "LOW RISK"
-    return "TAT:" + str(tat) + " PF1.2:" + str(pf12) + " Hemo:" + str(hemo) + " Platelets:" + str(platelets) + "\nResult: " + overall
+    return "TAT: " + str(tat) + "\nPF1.2: " + str(pf12) + "\nFree Hemoglobin: " + str(hemo) + "\nPlatelets: " + str(platelets) + "\nTime: " + str(time) + " min\nOverall: " + overall
 
 def upad_tool(r, g, b):
     creatinine = max(0, round(0.02*(float(r)-float(b))-0.5, 2))
     stage = "Normal" if creatinine<1.2 else "Borderline" if creatinine<1.5 else "Stage 2 CKD" if creatinine<3.0 else "Stage 3-4 CKD" if creatinine<6.0 else "Stage 5 CKD"
-    return "Creatinine: " + str(creatinine) + " mg/dL\nStage: " + stage
-
-def ask(q):
-    answers = {
-        "sylgard": "Sylgard 184 is used in the MCL to create transparent test sections for PIV imaging. Mixed at 10:1 ratio, cured 48hrs.",
-        "tgt": "TGT uses Arduino Uno + Stepper Motor at 70 bpm. Blood sampled at 0,20,40,60 min. Measures TAT, PF1.2, hemolysis.",
-        "jaffe": "Jaffe reaction: creatinine + picric acid = orange-red color. Used in uPAD for CKD creatinine detection.",
-        "piv": "PIV = Particle Image Velocimetry. Green laser + high speed camera measures velocity and shear stress in MCL.",
-        "ckd": "CKD detected via uPAD Jaffe reaction. Normal creatinine < 1.2 mg/dL. CKD threshold > 1.5 mg/dL.",
-        "mhv": "Lab uses 27mm SJM Regent MHV. Tested in MCL for hemodynamics and TGT for thrombogenicity.",
-    }
-    for key, answer in answers.items():
-        if key in q.lower():
-            return answer
-    return "CardioLab AI covers MCL, PIV, TGT, uPAD, CKD, FSI, MHV. Ask about any topic!"
+    return "Creatinine: " + str(creatinine) + " mg/dL\nCKD Stage: " + stage + "\nConfirm with: Heska Element HT5"
 
 with gr.Blocks(title="CardioLab AI - SJSU") as demo:
     gr.Markdown("# CardioLab AI Agent")
-    gr.Markdown("### SJSU Biomedical Engineering | Open Source")
+    gr.Markdown("### SJSU Biomedical Engineering | Built on Biomni (Stanford) | Powered by Llama 70B")
     gr.Markdown("GitHub: github.com/pranatechsol/Cardio-Lab-Ai")
     with gr.Tab("Ask Agent"):
-        q = gr.Textbox(label="Ask about CardioLab research")
-        a = gr.Textbox(label="Answer", lines=4)
-        gr.Button("Ask CardioLab AI").click(ask, inputs=q, outputs=a)
+        gr.Markdown("Ask anything about MCL, PIV, TGT, uPAD, CKD, FSI, MHV")
+        q = gr.Textbox(label="Your question", placeholder="e.g. What is Sylgard 184 used for?")
+        a = gr.Textbox(label="Answer", lines=6)
+        gr.Button("Ask CardioLab AI").click(ask_biomni, inputs=q, outputs=a)
     with gr.Tab("PIV Analysis"):
+        gr.Markdown("### Analyze PIV flow data from Mock Circulatory Loop")
         v = gr.Number(label="Max Velocity m/s", value=1.8)
         s = gr.Number(label="Shear Stress Pa", value=6.5)
         h = gr.Number(label="Heart Rate bpm", value=72)
         out = gr.Textbox(label="Result", lines=4)
         gr.Button("Analyze PIV").click(piv_tool, inputs=[v,s,h], outputs=out)
     with gr.Tab("TGT Results"):
+        gr.Markdown("### Interpret Thrombogenicity Tester blood results")
         t1 = gr.Number(label="TAT", value=18)
         t2 = gr.Number(label="PF1.2", value=2.5)
-        t3 = gr.Number(label="Free Hemoglobin", value=60)
+        t3 = gr.Number(label="Free Hemoglobin mg/L", value=60)
         t4 = gr.Number(label="Platelet Count", value=140)
         t5 = gr.Number(label="Time minutes", value=40)
         out2 = gr.Textbox(label="Result", lines=5)
         gr.Button("Analyze TGT").click(tgt_tool, inputs=[t1,t2,t3,t4,t5], outputs=out2)
     with gr.Tab("uPAD CKD"):
+        gr.Markdown("### Analyze uPAD colorimetric result - Jaffe Reaction")
         r = gr.Number(label="R value", value=210)
         g = gr.Number(label="G value", value=140)
         b = gr.Number(label="B value", value=80)