Upload app.py with huggingface_hub

app.py

@@ -5,20 +5,26 @@ from groq import Groq
 
 GROQ_KEY = os.environ.get("GROQ_API_KEY", "")
 
-KNOWHOW = """SJSU CardioLab: MCL uses Sylgard 184 PDMS 10:1 ratio 48hr cure green laser PIV 70bpm 5L/min. TGT uses Arduino Uno Stepper Motor 150mL blood sampled at 0 20 40 60min measures TAT PF1.2 hemolysis platelets. uPAD uses Jaffe reaction creatinine plus picric acid gives orange-red color normal 0.6-1.2 mg/dL CKD above 1.5. FSI uses COMSOL ALE mesh blood 1060 kg/m3 0.0035 Pa.s St Jude geometry. MHV is 27mm SJM Regent bileaflet also trileaflet monoleaflet pediatric. Equipment Heska HT5 hematology analyzer time-resolved PIV Tygon tubing Arduino Uno."""
+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+" mechanical heart valve OR microfluidic OR CKD","retmax":3,"retmode":"json","sort":"date"},
+            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📚 PubMed: " + " | ".join(links)
+        return "\n\nPubMed: " + " | ".join(links)
     except:
         return ""
 
@@ -27,27 +33,13 @@ def chat(message, history):
         return "Error: Add GROQ_API_KEY to Space Settings under 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
-            }
-        ]
-        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:
-                if item[0]:
-                    msgs.append({"role": "user", "content": str(item[0])})
-                if item[1]:
-                    msgs.append({"role": "assistant", "content": str(item[1])})
+        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}]
+        for h in history:
+            msgs.append({"role":"user","content":str(h[0])})
+            msgs.append({"role":"assistant","content":str(h[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
-        )
+        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)
@@ -71,17 +63,20 @@ 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")
-
     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),
-            submit_btn="Send",
-            clear_btn="Clear Chat"
-        )
-
+        chatbot = gr.Chatbot(label="CardioLab AI", height=450)
+        msg = gr.Textbox(label="Ask anything about CardioLab research", placeholder="e.g. How does the TGT measure thrombogenicity?", lines=2)
+        with gr.Row():
+            send = gr.Button("Send", variant="primary")
+            clear = gr.Button("Clear Chat")
+        def respond(message, history):
+            reply = chat(message, history)
+            history.append((message, reply))
+            return "", history
+        send.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
+        msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
+        clear.click(lambda: [], None, chatbot)
     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)
@@ -89,7 +84,6 @@ with gr.Blocks(title="CardioLab AI - SJSU") as demo:
         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)
@@ -99,7 +93,6 @@ with gr.Blocks(title="CardioLab AI - SJSU") as demo:
         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)
@@ -108,4 +101,4 @@ with gr.Blocks(title="CardioLab AI - SJSU") as demo:
         out3 = gr.Textbox(label="Result", lines=4)
         gr.Button("Analyze uPAD").click(upad_tool, inputs=[r,g,b], outputs=out3)
 
-demo.launch()
+demo.launch()