Update app.py

app.py

@@ -24,20 +24,21 @@ def get_pubmed(query):
 
 def respond(message, history):
     if not GROQ_KEY:
-        history.append({"role":"user","content":message})
-        history.append({"role":"assistant","content":"Error: Add GROQ_API_KEY to Space Settings tab under Secrets."})
+        history.append([message, "Error: Add GROQ_API_KEY to Space Settings under Secrets."])
         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 SNAP Lab. Expert in MHV MCL PIV TGT uPAD CKD FSI COMSOL. Remember the full conversation. Never invent paper URLs.\n\n"+KNOWHOW
+                "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 msg in history:
-            if isinstance(msg, dict):
-                msgs.append({"role":msg["role"],"content":msg["content"]})
+        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)})
         pubmed = get_pubmed(message)
         msgs.append({"role":"user","content":message+"\n\n"+pubmed})
         resp = client.chat.completions.create(
@@ -47,13 +48,11 @@ def respond(message, history):
         )
         answer = resp.choices[0].message.content
         if pubmed:
-            answer += "\n\n📚 "+pubmed
-        history.append({"role":"user","content":message})
-        history.append({"role":"assistant","content":answer})
+            answer += "\n\n📚 " + pubmed
+        history.append([message, answer])
         return "", history
     except Exception as e:
-        history.append({"role":"user","content":message})
-        history.append({"role":"assistant","content":"Error: "+str(e)})
+        history.append([message, "Error: " + str(e)])
         return "", history
 
 def piv_tool(velocity, shear, hr):
@@ -77,7 +76,7 @@ with gr.Blocks(title="CardioLab AI - SJSU") as demo:
     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, type="messages")
+        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")