Update app.py

app.py

@@ -18,14 +18,13 @@ def get_pubmed(query):
         if not ids:
             return ""
         links = ["https://pubmed.ncbi.nlm.nih.gov/"+i for i in ids]
-        return "PubMed: " + " | ".join(links)
+        return "\n📚 PubMed: " + " | ".join(links)
     except:
         return ""
 
-def respond(message, history):
+def chat(message, history):
     if not GROQ_KEY:
-        history.append([message, "Error: Add GROQ_API_KEY to Space Settings under Secrets."])
-        return "", history
+        return "Error: Add GROQ_API_KEY to Space Settings under Secrets tab."
     try:
         client = Groq(api_key=GROQ_KEY)
         msgs = [
@@ -34,26 +33,24 @@ def respond(message, history):
                 "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 human_msg, ai_msg in history:
-            if human_msg:
-                msgs.append({"role":"user","content":str(human_msg)})
-            if ai_msg:
-                msgs.append({"role":"assistant","content":str(ai_msg)})
+        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])})
         pubmed = get_pubmed(message)
-        msgs.append({"role":"user","content":message+"\n\n"+pubmed})
+        msgs.append({"role": "user", "content": message})
         resp = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=msgs,
             max_tokens=600
         )
-        answer = resp.choices[0].message.content
-        if pubmed:
-            answer += "\n\n📚 " + pubmed
-        history.append([message, answer])
-        return "", history
+        return resp.choices[0].message.content + pubmed
     except Exception as e:
-        history.append([message, "Error: " + str(e)])
-        return "", history
+        return "Error: " + str(e)
 
 def piv_tool(velocity, shear, hr):
     v = "HIGH - stenosis risk" if float(velocity)>2.0 else "NORMAL"
@@ -74,16 +71,17 @@ 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")
-        chatbot = gr.Chatbot(label="CardioLab AI", height=500)
-        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")
-        send.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
-        msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
-        clear.click(lambda: [], None, chatbot)
+        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"
+        )
+
     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)
@@ -91,6 +89,7 @@ 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)
@@ -100,6 +99,7 @@ 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)