Upload app.py with huggingface_hub

app.py

@@ -34,15 +34,18 @@ def chat(message, 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}]
-        for h in history:
-            msgs.append({"role":"user","content":str(h[0])})
-            msgs.append({"role":"assistant","content":str(h[1])})
+        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})
         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 "Error: "+str(e)
 
 def piv_tool(velocity, shear, hr):
     v = "HIGH - stenosis risk" if float(velocity)>2.0 else "NORMAL"
@@ -65,18 +68,14 @@ 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=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)
+        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"),
+            submit_btn="Send",
+            clear_btn="Clear Chat",
+            title="CardioLab Research Assistant"
+        )
     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)