Upload app.py with huggingface_hub

app.py

@@ -9,92 +9,98 @@ KNOWHOW = """SJSU CardioLab:
 MCL: Sylgard 184 PDMS 10:1 ratio 48hr cure green laser PIV 70bpm 5L/min.
 TGT: Arduino Uno Stepper Motor 150mL blood sampled at 0 20 40 60min measures TAT PF1.2 hemolysis platelets.
 uPAD: Jaffe reaction creatinine plus picric acid gives orange-red color normal 0.6-1.2 mg/dL CKD above 1.5.
-FSI: COMSOL ALE mesh blood 1060 kg/m3 0.0035 Pa.s St Jude geometry.
 MHV: 27mm SJM Regent bileaflet also trileaflet monoleaflet pediatric.
 Equipment: Heska HT5 hematology analyzer time-resolved PIV Tygon tubing Arduino Uno."""
 
 def get_pubmed(query):
     try:
-        r = requests.get(
-            "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi",
-            params={"db":"pubmed","term":query,"retmax":3,"retmode":"json","sort":"date"},
-            timeout=8
-        )
+        r = requests.get("https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi",
+            params={"db":"pubmed","term":query,"retmax":3,"retmode":"json","sort":"date"},timeout=8)
         ids = r.json()["esearchresult"]["idlist"]
-        if not ids:
-            return ""
-        links = ["https://pubmed.ncbi.nlm.nih.gov/"+i for i in ids]
-        return "\n\nPubMed: " + " | ".join(links)
-    except:
-        return ""
+        if not ids: return ""
+        return "\n\nPubMed: "+" | ".join(["https://pubmed.ncbi.nlm.nih.gov/"+i for i in ids])
+    except: return ""
 
 def chat(message, history):
     if not GROQ_KEY:
-        return "Error: Add GROQ_API_KEY to Space Settings under Secrets tab."
+        return "Error: Add GROQ_API_KEY to Space Settings Secrets tab."
     try:
         client = Groq(api_key=GROQ_KEY)
-        msgs = [{"role":"system","content":"You are CardioLab AI from SJSU Biomedical Engineering built on Biomni Stanford SNAP Lab. Expert in MHV MCL PIV TGT uPAD CKD FSI COMSOL. Remember the full conversation. Never invent paper URLs.\n\n"+KNOWHOW}]
+        msgs = [{"role":"system","content":"You are CardioLab AI from SJSU Biomedical Engineering built on Biomni Stanford. Expert in MHV MCL PIV TGT uPAD CKD FSI. Remember full conversation. Never invent URLs.\n\n"+KNOWHOW}]
         for item in history:
             if isinstance(item, dict):
                 msgs.append({"role":item["role"],"content":item["content"]})
-            elif isinstance(item, (list,tuple)) and len(item)==2:
+            elif isinstance(item,(list,tuple)) and len(item)==2:
                 if item[0]: msgs.append({"role":"user","content":str(item[0])})
                 if item[1]: msgs.append({"role":"assistant","content":str(item[1])})
-        pubmed = get_pubmed(message)
         msgs.append({"role":"user","content":message})
         resp = client.chat.completions.create(model="llama-3.3-70b-versatile",messages=msgs,max_tokens=600)
-        return resp.choices[0].message.content + pubmed
-    except Exception as e:
-        return "Error: "+str(e)
+        return resp.choices[0].message.content + get_pubmed(message)
+    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 "ELEVATED" if float(shear)>5 else "NORMAL"
-    return "Velocity: "+str(velocity)+" m/s - "+v+"\nShear: "+str(shear)+" Pa - "+s+"\nHeart Rate: "+str(hr)+" bpm"
+    v = "HIGH-stenosis" if float(velocity)>2.0 else "NORMAL"
+    s = "HIGH-thrombosis" if float(shear)>10 else "ELEVATED" if float(shear)>5 else "NORMAL"
+    return "Velocity:"+str(velocity)+"m/s - "+v+"\nShear:"+str(shear)+"Pa - "+s+"\nHR:"+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])
-    r = "HIGH THROMBOGENIC RISK" if risk>=3 else "MODERATE RISK" if risk>=2 else "LOW RISK"
-    return "TAT:"+str(tat)+" PF1.2:"+str(pf12)+"\nHemo:"+str(hemo)+" Platelets:"+str(platelets)+"\nTime:"+str(time)+" min\nResult: "+r
+def tgt_tool(tat,pf12,hemo,platelets,time):
+    risk=sum([float(tat)>15,float(pf12)>2.0,float(hemo)>50,float(platelets)<150])
+    r="HIGH RISK" if risk>=3 else "MODERATE" if risk>=2 else "LOW RISK"
+    return "TAT:"+str(tat)+" PF1.2:"+str(pf12)+"\nHemo:"+str(hemo)+" Plt:"+str(platelets)+"\nResult:"+r
 
-def upad_tool(r, g, b):
-    c = max(0,round(0.02*(float(r)-float(b))-0.5,2))
-    s = "Normal" if c<1.2 else "Borderline" if c<1.5 else "Stage 2 CKD" if c<3.0 else "Stage 3-4 CKD" if c<6.0 else "Stage 5 CKD"
-    return "Creatinine: "+str(c)+" mg/dL\nStage: "+s+"\nConfirm with: Heska Element HT5"
+def upad_tool(r,g,b):
+    c=max(0,round(0.02*(float(r)-float(b))-0.5,2))
+    s="Normal" if c<1.2 else "Borderline" if c<1.5 else "Stage2CKD" if c<3.0 else "Stage3-4" if c<6.0 else "Stage5"
+    return "Creatinine:"+str(c)+"mg/dL\nStage:"+s
 
-with gr.Blocks(title="CardioLab AI - SJSU") as demo:
+with gr.Blocks(title="CardioLab AI SJSU") as demo:
     gr.Markdown("# CardioLab AI Agent")
-    gr.Markdown("### SJSU Biomedical Engineering | Biomni Stanford + Llama 70B + Chat Memory + PubMed")
-    gr.Markdown("Open Source: github.com/pranatechsol/Cardio-Lab-Ai")
+    gr.Markdown("### SJSU Biomedical Engineering | Biomni Stanford + Llama 70B + PubMed")
+    gr.Markdown("github.com/pranatechsol/Cardio-Lab-Ai")
+
     with gr.Tab("Research Chat"):
-        gr.Markdown("### Chat with memory like ChatGPT for CardioLab research")
-        gr.ChatInterface(
-            fn=chat,
-            chatbot=gr.Chatbot(height=450, label="CardioLab AI"),
-            textbox=gr.Textbox(placeholder="Ask anything about CardioLab research...", lines=2, label="Your message")
-        )
+        gr.Markdown("### Chat like ChatGPT for CardioLab research")
+        chatbot = gr.Chatbot(height=400, label="CardioLab AI")
+        msg_box = gr.Textbox(placeholder="Ask anything about CardioLab...", label="Your message", lines=2)
+        with gr.Row():
+            send_btn = gr.Button("Send", variant="primary", scale=3)
+            clear_btn = gr.Button("Clear", scale=1)
+
+        def respond(message, history):
+            if not message.strip():
+                return "", history
+            reply = chat(message, history)
+            history = history + [[message, reply]]
+            return "", history
+
+        send_btn.click(respond, inputs=[msg_box, chatbot], outputs=[msg_box, chatbot])
+        msg_box.submit(respond, inputs=[msg_box, chatbot], outputs=[msg_box, chatbot])
+        clear_btn.click(lambda: ([], ""), outputs=[chatbot, msg_box])
+
     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)
+        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 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)
+        t1=gr.Number(label="TAT",value=18)
+        t2=gr.Number(label="PF1.2",value=2.5)
+        t3=gr.Number(label="Free Hemoglobin",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)
-        out3 = gr.Textbox(label="Result", lines=4)
-        gr.Button("Analyze uPAD").click(upad_tool, inputs=[r,g,b], outputs=out3)
+        r=gr.Number(label="R value",value=210)
+        g=gr.Number(label="G value",value=140)
+        b=gr.Number(label="B value",value=80)
+        out3=gr.Textbox(label="Result",lines=4)
+        gr.Button("Analyze uPAD").click(upad_tool,inputs=[r,g,b],outputs=out3)
 
 demo.launch()