Upload app.py with huggingface_hub

app.py

@@ -21,22 +21,27 @@ def get_pubmed(query):
         return "\n\nPubMed: "+" | ".join(["https://pubmed.ncbi.nlm.nih.gov/"+i for i in ids])
     except: return ""
 
-def chat(message, history):
+def respond(message, history):
     if not GROQ_KEY:
-        return "Error: Add GROQ_API_KEY to Space Settings Secrets tab."
+        history.append({"role":"user","content":message})
+        history.append({"role":"assistant","content":"Error: Add GROQ_API_KEY to Space Settings Secrets tab."})
+        return "", history
     try:
         client = Groq(api_key=GROQ_KEY)
         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:
-                if item[0]: msgs.append({"role":"user","content":str(item[0])})
-                if item[1]: msgs.append({"role":"assistant","content":str(item[1])})
         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 + get_pubmed(message)
-    except Exception as e: return "Error: "+str(e)
+        answer = resp.choices[0].message.content + get_pubmed(message)
+        history.append({"role":"user","content":message})
+        history.append({"role":"assistant","content":answer})
+        return "", history
+    except Exception as e:
+        history.append({"role":"user","content":message})
+        history.append({"role":"assistant","content":"Error: "+str(e)})
+        return "", history
 
 def piv_tool(velocity, shear, hr):
     v = "HIGH-stenosis" if float(velocity)>2.0 else "NORMAL"
@@ -57,26 +62,16 @@ with gr.Blocks(title="CardioLab AI SJSU") as demo:
     gr.Markdown("# CardioLab AI Agent")
     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 like ChatGPT for CardioLab research")
-        chatbot = gr.Chatbot(height=400, label="CardioLab AI")
+        chatbot = gr.Chatbot(height=400, label="CardioLab AI", type="messages")
         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)
@@ -84,7 +79,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)
@@ -94,7 +88,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)