Upload app.py with huggingface_hub

app.py

@@ -1,409 +1,259 @@
 import gradio as gr
-import os
-import requests
+import os, requests, io
 import matplotlib
-matplotlib.use("Agg")
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import matplotlib.patches as mpatches
 import numpy as np
 from groq import Groq
-import io
 from PIL import Image
 
-GROQ_KEY = os.environ.get("GROQ_API_KEY", "")
+GROQ_KEY = os.environ.get('GROQ_API_KEY', '')
+KNOWHOW = '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. MHV: 27mm SJM Regent bileaflet also trileaflet monoleaflet pediatric.'
 
-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. "
-    "MHV: 27mm SJM Regent bileaflet also trileaflet monoleaflet pediatric. "
-    "Equipment: Heska HT5 hematology analyzer time-resolved PIV Tygon tubing Arduino Uno.")
-
-CSS = """
+CSS = '''
 body, .gradio-container { background: #f0f4f8 !important; color: #1a202c !important; }
 .tab-nav { background: #ffffff !important; border-bottom: 2px solid #e2e8f0 !important; padding: 0 10px !important; }
-.tab-nav button { background: #f7fafc !important; color: #2d3748 !important; border: 1px solid #e2e8f0 !important; border-radius: 8px 8px 0 0 !important; padding: 12px 18px !important; font-size: 0.9em !important; font-weight: 600 !important; margin-top: 6px !important; transition: all 0.2s !important; }
-.tab-nav button:hover { background: #ebf4ff !important; color: #1a237e !important; border-color: #4361ee !important; }
-.tab-nav button.selected { background: linear-gradient(135deg, #e63946, #c1121f) !important; color: #ffffff !important; border-color: #e63946 !important; font-weight: 700 !important; }
-.tabitem { background: #ffffff !important; padding: 20px !important; border: 1px solid #e2e8f0 !important; border-top: none !important; border-radius: 0 0 12px 12px !important; }
-.main-header { background: linear-gradient(135deg, #1a237e 0%, #b71c1c 100%); border-bottom: 3px solid #e63946; padding: 25px 20px; text-align: center; border-radius: 12px 12px 0 0; }
-textarea, input[type=number], input[type=text] { background: #f7fafc !important; color: #1a202c !important; border: 1px solid #cbd5e0 !important; border-radius: 8px !important; }
-textarea::placeholder { color: #a0aec0 !important; }
-button.primary { background: linear-gradient(135deg, #e63946 0%, #c1121f 100%) !important; color: white !important; border: none !important; border-radius: 8px !important; font-weight: 700 !important; padding: 12px 24px !important; }
+.tab-nav button { background: #f7fafc !important; color: #2d3748 !important; border: 1px solid #e2e8f0 !important; border-radius: 8px 8px 0 0 !important; padding: 12px 18px !important; font-size: 0.9em !important; font-weight: 600 !important; margin-top: 6px !important; }
+.tab-nav button:hover { background: #ebf4ff !important; color: #1a237e !important; }
+.tab-nav button.selected { background: linear-gradient(135deg, #e63946, #c1121f) !important; color: #ffffff !important; font-weight: 700 !important; }
+button.primary { background: linear-gradient(135deg, #e63946 0%, #c1121f 100%) !important; color: white !important; border: none !important; border-radius: 8px !important; font-weight: 700 !important; }
 button.secondary { background: #edf2f7 !important; color: #4a5568 !important; border: 1px solid #cbd5e0 !important; border-radius: 8px !important; }
-label span, .label-wrap span { color: #2b6cb0 !important; font-weight: 600 !important; font-size: 0.85em !important; text-transform: uppercase !important; }
-.message.user { background: linear-gradient(135deg, #e63946, #c1121f) !important; color: white !important; border-radius: 12px 12px 4px 12px !important; }
-.message.bot { background: #ebf4ff !important; color: #1a202c !important; border: 1px solid #bee3f8 !important; border-radius: 12px 12px 12px 4px !important; }
-"""
+textarea, input[type=number] { background: #f7fafc !important; color: #1a202c !important; border: 1px solid #cbd5e0 !important; border-radius: 8px !important; }
+.message.user { background: linear-gradient(135deg, #e63946, #c1121f) !important; color: white !important; }
+.message.bot { background: #ebf4ff !important; color: #1a202c !important; border: 1px solid #bee3f8 !important; }
+label span { color: #2b6cb0 !important; font-weight: 600 !important; font-size: 0.85em !important; text-transform: uppercase !important; }
+'''
 
 def get_pubmed(query, n=5):
     try:
-        r = requests.get("https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi",
-            params={"db":"pubmed","term":query+" AND (mechanical heart valve OR microfluidic OR CKD OR thrombogenicity)",
-            "retmax":n,"retmode":"json","sort":"date"},timeout=10)
-        ids = r.json()["esearchresult"]["idlist"]
-        if not ids: return ""
-        return chr(10).join(["https://pubmed.ncbi.nlm.nih.gov/"+i for i in ids])
-    except: return ""
+        r = requests.get('https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi',
+            params={'db':'pubmed','term':query+' AND (mechanical heart valve OR microfluidic OR CKD OR thrombogenicity)','retmax':n,'retmode':'json','sort':'date'},timeout=10)
+        ids = r.json()['esearchresult']['idlist']
+        if not ids: return ''
+        return chr(10).join(['https://pubmed.ncbi.nlm.nih.gov/'+i for i in ids])
+    except: return ''
 
 def get_scholar(query, n=5):
     try:
-        r = requests.get("https://api.semanticscholar.org/graph/v1/paper/search",
-            params={"query":query+" biomedical","limit":n,"fields":"title,year,url,citationCount"},timeout=10)
-        papers = r.json().get("data",[])
+        r = requests.get('https://api.semanticscholar.org/graph/v1/paper/search',
+            params={'query':query+' biomedical','limit':n,'fields':'title,year,url,citationCount'},timeout=10)
+        papers = r.json().get('data',[])
         out = []
         for p in papers:
-            title = p.get("title","")
-            year = str(p.get("year",""))
-            url = p.get("url","")
-            citations = str(p.get("citationCount",0))
-            if url:
-                out.append(title[:80]+" ("+year+") - "+citations+" citations"+chr(10)+"   "+url)
-        return chr(10)+chr(10).join(out)
-    except: return ""
+            url = p.get('url','')
+            if url: out.append(p.get('title','')[:80]+' ('+str(p.get('year',''))+') - '+str(p.get('citationCount',0))+' citations'+chr(10)+'   '+url)
+        return chr(10).join(out)
+    except: return ''
 
 def quick_search(query):
-    if not query.strip(): return "Please enter a research topic."
+    if not query.strip(): return 'Please enter a research topic.'
     pubmed = get_pubmed(query, n=8)
     scholar = get_scholar(query, n=5)
-    result = "PAPERS FOR: " + query + chr(10)+chr(10)
-    result += "PUBMED (verified links):"+chr(10)+pubmed+chr(10)+chr(10)
-    result += "SEMANTIC SCHOLAR:"+chr(10)+scholar
-    return result
+    return 'PUBMED RESULTS:'+chr(10)+pubmed+chr(10)+chr(10)+'SEMANTIC SCHOLAR:'+chr(10)+scholar
 
 def research_chat(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 Secrets tab."})
-        return "", history
+        history.append({'role':'user','content':message})
+        history.append({'role':'assistant','content':'Error: Add GROQ_API_KEY to Space Settings Secrets.'})
+        return '', history
     try:
         client = Groq(api_key=GROQ_KEY)
         pubmed = get_pubmed(message, n=3)
-        msgs = [{"role":"system","content":"You are CardioLab AI assistant. Expert in MHV MCL PIV TGT uPAD CKD FSI. Remember full conversation. Never invent URLs. "+KNOWHOW}]
+        msgs = [{'role':'system','content':'You are CardioLab AI. Expert in MHV MCL PIV TGT uPAD CKD FSI. Remember full conversation. Never invent URLs. '+KNOWHOW}]
         for item in history:
-            if isinstance(item, dict):
-                msgs.append({"role":item["role"],"content":item["content"]})
-        msgs.append({"role":"user","content":message})
-        resp = client.chat.completions.create(model="llama-3.3-70b-versatile",messages=msgs,max_tokens=700)
+            if isinstance(item, dict): msgs.append({'role':item['role'],'content':item['content']})
+        msgs.append({'role':'user','content':message})
+        resp = client.chat.completions.create(model='llama-3.3-70b-versatile',messages=msgs,max_tokens=700)
         answer = resp.choices[0].message.content
-        if pubmed: answer += chr(10)+chr(10)+"PUBMED LINKS:"+chr(10)+pubmed
-        history.append({"role":"user","content":message})
-        history.append({"role":"assistant","content":answer})
-        return "", history
+        if pubmed: answer += chr(10)+chr(10)+'PUBMED LINKS:'+chr(10)+pubmed
+        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
+        history.append({'role':'user','content':message})
+        history.append({'role':'assistant','content':'Error: '+str(e)})
+        return '', history
 
 def voice_chat(audio, history):
     if audio is None:
-        history.append({"role":"assistant","content":"Please record your question first."})
+        history.append({'role':'assistant','content':'Please record your question first.'})
         return history
     try:
         client = Groq(api_key=GROQ_KEY)
-        with open(audio, "rb") as f:
-            transcription = client.audio.transcriptions.create(
-                file=("audio.wav", f, "audio/wav"),
-                model="whisper-large-v3"
-            )
-        text = transcription.text
-        msgs = [{"role":"system","content":"You are CardioLab AI assistant. Expert in MHV MCL PIV TGT uPAD CKD FSI. "+KNOWHOW}]
+        with open(audio, 'rb') as f:
+            tx = client.audio.transcriptions.create(file=('audio.wav', f, 'audio/wav'), model='whisper-large-v3')
+        text = tx.text
+        msgs = [{'role':'system','content':'You are CardioLab AI. '+KNOWHOW}]
         for item in history:
-            if isinstance(item, dict):
-                msgs.append({"role":item["role"],"content":item["content"]})
-        msgs.append({"role":"user","content":text})
-        resp = client.chat.completions.create(model="llama-3.3-70b-versatile",messages=msgs,max_tokens=500)
-        answer = resp.choices[0].message.content
-        history.append({"role":"user","content":"[Voice] "+text})
-        history.append({"role":"assistant","content":answer})
+            if isinstance(item, dict): msgs.append({'role':item['role'],'content':item['content']})
+        msgs.append({'role':'user','content':text})
+        resp = client.chat.completions.create(model='llama-3.3-70b-versatile',messages=msgs,max_tokens=500)
+        history.append({'role':'user','content':'[Voice] '+text})
+        history.append({'role':'assistant','content':resp.choices[0].message.content})
         return history
     except Exception as e:
-        history.append({"role":"assistant","content":"Voice error: "+str(e)})
+        history.append({'role':'assistant','content':'Voice error: '+str(e)})
         return history
 
 def generate_diagram(topic):
-    topic_lower = topic.lower()
-    fig, ax = plt.subplots(1, 1, figsize=(12, 8))
-    ax.set_xlim(0, 12)
-    ax.set_ylim(0, 8)
-    ax.axis("off")
-    fig.patch.set_facecolor("#0d1b3e")
-    ax.set_facecolor("#0d1b3e")
-
-    if "tgt" in topic_lower or "thrombogen" in topic_lower:
-        ax.set_title("Thrombogenicity Tester (TGT) — SJSU CardioLab", color="white", fontsize=14, fontweight="bold", pad=15)
-        components = [
-            (1, 6, 2, 1, "#e63946", "Arduino Uno
-(Controller)"),
-            (4, 6, 2, 1, "#4361ee", "Motor Driver"),
-            (7, 6, 2, 1, "#2ecc71", "Stepper Motor"),
-            (4, 3.5, 4, 2, "#1a2744", "Test Chamber
-27mm SJM Regent MHV
-(150mL Blood)"),
-            (1, 1, 2, 1, "#e67e22", "Heska HT5
-Analyzer"),
-            (9, 1, 2, 1, "#9b59b6", "Data Logger"),
+    t = topic.lower()
+    fig, ax = plt.subplots(figsize=(12, 8))
+    ax.set_xlim(0,12); ax.set_ylim(0,8); ax.axis('off')
+    fig.patch.set_facecolor('#0d1b3e'); ax.set_facecolor('#0d1b3e')
+    if 'tgt' in t or 'thrombogen' in t:
+        ax.set_title('Thrombogenicity Tester (TGT) - SJSU CardioLab', color='white', fontsize=14, fontweight='bold', pad=15)
+        boxes = [
+            (0.5, 6.0, 2.0, 0.9, '#e63946', 'Arduino Uno'),
+            (3.5, 6.0, 2.0, 0.9, '#4361ee', 'Motor Driver'),
+            (6.5, 6.0, 2.0, 0.9, '#2ecc71', 'Stepper Motor'),
+            (3.5, 3.5, 4.0, 2.0, '#1a2744', 'Test Chamber - 27mm SJM MHV - 150mL Blood'),
+            (0.5, 1.0, 2.0, 0.9, '#e67e22', 'Heska HT5 Analyzer'),
+            (9.0, 1.0, 2.0, 0.9, '#9b59b6', 'Data Logger'),
         ]
-        for x, y, w, h, color, label in components:
-            rect = mpatches.FancyBboxPatch((x, y), w, h, boxstyle="round,pad=0.1", facecolor=color, edgecolor="white", linewidth=2)
-            ax.add_patch(rect)
-            ax.text(x+w/2, y+h/2, label, ha="center", va="center", color="white", fontsize=8, fontweight="bold")
-        arrows = [(2, 6.5, 4, 6.5), (5, 6.5, 7, 6.5), (8, 6, 8, 5.5), (6, 3.5, 6, 2), (4, 4.5, 2, 2), (8, 3.5, 10, 2)]
-        for x1,y1,x2,y2 in arrows:
-            ax.annotate("", xy=(x2,y2), xytext=(x1,y1), arrowprops=dict(arrowstyle="->", color="#7eb8f7", lw=2))
-        labels = [(3, 7, "Signal"), (5.5, 7, "Power"), (8.5, 5.7, "Rotate"), (6.5, 2.8, "Sample"), (2.8, 3.5, "Analyze"), (9, 2.8, "Record")]
-        for x, y, t in labels:
-            ax.text(x, y, t, color="#a8b2d8", fontsize=7, ha="center")
-        times = ["0 min", "20 min", "40 min", "60 min"]
-        for i, t in enumerate(times):
-            ax.text(1+i*2.5, 0.5, t, color="#e63946", fontsize=8, ha="center", fontweight="bold")
-        ax.text(6, 0.5, "Blood Sampling Times", color="#7eb8f7", fontsize=8, ha="center")
-
-    elif "mcl" in topic_lower or "circulatory" in topic_lower or "piv" in topic_lower:
-        ax.set_title("Mock Circulatory Loop (MCL) + PIV System — SJSU CardioLab", color="white", fontsize=13, fontweight="bold", pad=15)
-        loop_x = [2, 4, 6, 8, 10, 10, 8, 6, 4, 2, 2]
-        loop_y = [4, 6, 6.5, 6, 4, 3, 2, 1.5, 2, 3, 4]
-        ax.plot(loop_x, loop_y, color="#4361ee", linewidth=4, label="Flow Loop")
-        components = [
-            (5, 5.8, 2, 0.8, "#e63946", "MHV Test Section
-(Sylgard 184)"),
-            (1, 3.5, 1.5, 0.8, "#2ecc71", "Pump"),
-            (8.5, 3.5, 1.5, 0.8, "#e67e22", "Compliance
-Chamber"),
-            (4.5, 1.2, 2, 0.8, "#9b59b6", "Reservoir"),
-        ]
-        for x, y, w, h, color, label in components:
-            rect = mpatches.FancyBboxPatch((x, y), w, h, boxstyle="round,pad=0.1", facecolor=color, edgecolor="white", linewidth=2)
-            ax.add_patch(rect)
-            ax.text(x+w/2, y+h/2, label, ha="center", va="center", color="white", fontsize=7, fontweight="bold")
-        ax.annotate("", xy=(6, 7.5), xytext=(6, 6.5), arrowprops=dict(arrowstyle="->", color="#00ff88", lw=2))
-        ax.text(6, 7.7, "Green Laser (PIV)", color="#00ff88", fontsize=8, ha="center", fontweight="bold")
-        rect_laser = mpatches.FancyBboxPatch((5, 7.5), 2, 0.4, boxstyle="round,pad=0.05", facecolor="#00ff44", edgecolor="white", linewidth=1)
-        ax.add_patch(rect_laser)
-        ax.text(6, 7.7, "LASER", ha="center", va="center", color="black", fontsize=7, fontweight="bold")
-        rect_cam = mpatches.FancyBboxPatch((9.5, 5.5), 1.5, 0.8, boxstyle="round,pad=0.05", facecolor="#1a2744", edgecolor="#7eb8f7", linewidth=2)
-        ax.add_patch(rect_cam)
-        ax.text(10.25, 5.9, "Camera
-(PIV)", ha="center", va="center", color="white", fontsize=7, fontweight="bold")
-
-    elif "upad" in topic_lower or "ckd" in topic_lower or "creatinine" in topic_lower:
-        ax.set_title("uPAD Fabrication and CKD Detection Flow — SJSU CardioLab", color="white", fontsize=13, fontweight="bold", pad=15)
-        steps = [
-            (0.5, 6, "#e63946", "1. Whatman
-Paper"),
-            (2.5, 6, "#e67e22", "2. Wax
-Printer"),
-            (4.5, 6, "#f1c40f", "3. Heat
-120°C"),
-            (6.5, 6, "#2ecc71", "4. Add
-Reagents"),
-            (8.5, 6, "#4361ee", "5. Apply
-Sample"),
-            (10, 6, "#9b59b6", "6. Image
-Analysis"),
-        ]
-        for x, y, color, label in steps:
-            circle = plt.Circle((x+0.75, y+0.5), 0.7, color=color, zorder=5)
-            ax.add_patch(circle)
-            ax.text(x+0.75, y+0.5, label, ha="center", va="center", color="white", fontsize=7, fontweight="bold", zorder=6)
-        for i in range(len(steps)-1):
-            x1 = steps[i][0]+1.45
-            x2 = steps[i+1][0]+0.05
-            y = 6.5
-            ax.annotate("", xy=(x2,y), xytext=(x1,y), arrowprops=dict(arrowstyle="->", color="white", lw=2))
-        colors_ckd = ["#2ecc71", "#f1c40f", "#e67e22", "#e74c3c", "#c0392b"]
-        stages = ["Normal
-<1.2", "Borderline
-1.2-1.5", "Stage 2
-1.5-3.0", "Stage 3-4
-3.0-6.0", "Stage 5
->6.0"]
-        for i, (color, stage) in enumerate(zip(colors_ckd, stages)):
-            rect = mpatches.FancyBboxPatch((0.5+i*2.2, 2), 2, 2.5, boxstyle="round,pad=0.1", facecolor=color, edgecolor="white", linewidth=2, alpha=0.8)
-            ax.add_patch(rect)
-            ax.text(1.5+i*2.2, 3.2, stage, ha="center", va="center", color="white", fontsize=8, fontweight="bold")
-        ax.text(6, 1.5, "Creatinine Level (mg/dL)", color="#7eb8f7", fontsize=9, ha="center", fontweight="bold")
-        ax.text(6, 4.8, "Jaffe Reaction: Creatinine + Picric Acid → Orange-Red Color", color="#a8b2d8", fontsize=8, ha="center")
-
-    elif "mhv" in topic_lower or "valve" in topic_lower or "bileaflet" in topic_lower:
-        ax.set_title("Mechanical Heart Valve (MHV) — Bileaflet Design", color="white", fontsize=14, fontweight="bold", pad=15)
-        circle_outer = plt.Circle((6, 4), 3, color="#2d3a5a", fill=True, zorder=1)
-        circle_inner = plt.Circle((6, 4), 2.8, color="#1a2744", fill=True, zorder=2)
-        ax.add_patch(circle_outer)
-        ax.add_patch(circle_inner)
-        border = plt.Circle((6, 4), 3, color="#7eb8f7", fill=False, linewidth=3, zorder=3)
-        ax.add_patch(border)
-        leaflet1 = mpatches.FancyBboxPatch((4.5, 3), 1.3, 2, boxstyle="round,pad=0.1", facecolor="#e63946", edgecolor="white", linewidth=2, zorder=4)
-        leaflet2 = mpatches.FancyBboxPatch((6.2, 3), 1.3, 2, boxstyle="round,pad=0.1", facecolor="#e63946", edgecolor="white", linewidth=2, zorder=4)
-        ax.add_patch(leaflet1)
-        ax.add_patch(leaflet2)
-        ax.text(5.15, 4, "Leaflet
-1", ha="center", va="center", color="white", fontsize=8, fontweight="bold", zorder=5)
-        ax.text(6.85, 4, "Leaflet
-2", ha="center", va="center", color="white", fontsize=8, fontweight="bold", zorder=5)
-        ax.annotate("", xy=(6, 7.5), xytext=(6, 7), arrowprops=dict(arrowstyle="->", color="#4361ee", lw=3))
-        ax.text(6, 7.7, "Blood Flow", color="#4361ee", ha="center", fontsize=9, fontweight="bold")
-        ax.annotate("", xy=(6, 0.5), xytext=(6, 1), arrowprops=dict(arrowstyle="->", color="#4361ee", lw=3))
-        specs = ["Diameter: 27mm (SJM Regent)", "Material: Pyrolytic Carbon", "Leaflets: 2 (Bileaflet)", "Opening Angle: 85°"]
-        for i, spec in enumerate(specs):
-            ax.text(0.3, 7-i*0.7, "• "+spec, color="#a8b2d8", fontsize=8)
-
+        for x,y,w,h,c,lbl in boxes:
+            ax.add_patch(mpatches.FancyBboxPatch((x,y),w,h,boxstyle='round,pad=0.1',facecolor=c,edgecolor='white',linewidth=2))
+            ax.text(x+w/2, y+h/2, lbl, ha='center', va='center', color='white', fontsize=8, fontweight='bold')
+        for x1,y1,x2,y2 in [(2.5,6.45,3.5,6.45),(5.5,6.45,6.5,6.45),(7.5,6.0,7.5,5.5),(5.5,3.5,5.5,2.3),(4.0,3.5,2.0,1.9),(8.0,3.5,10.0,1.9)]:
+            ax.annotate('',xy=(x2,y2),xytext=(x1,y1),arrowprops=dict(arrowstyle='->',color='#7eb8f7',lw=2))
+        for x,y,txt in [(2.9,6.7,'Signal'),(6.0,6.7,'Drive'),(7.8,5.7,'Rotate'),(6.0,2.9,'Sample'),(2.8,2.7,'Analyze'),(9.2,2.7,'Record')]:
+            ax.text(x,y,txt,color='#a8b2d8',fontsize=7,ha='center')
+        ax.text(6, 0.4, 'Sample at: 0 min  |  20 min  |  40 min  |  60 min', color='#e63946', fontsize=9, ha='center', fontweight='bold')
+    elif 'mcl' in t or 'piv' in t or 'circulatory' in t:
+        ax.set_title('Mock Circulatory Loop + PIV System - SJSU CardioLab', color='white', fontsize=13, fontweight='bold', pad=15)
+        lx = [2,4,6,8,10,10,8,6,4,2,2]; ly = [4,6,6.5,6,4,3,2,1.5,2,3,4]
+        ax.plot(lx,ly,color='#4361ee',linewidth=4)
+        for x,y,w,h,c,lbl in [(4.8,5.8,2.0,0.8,'#e63946','MHV Test Section'),(0.8,3.5,1.5,0.8,'#2ecc71','Pump'),(9.5,3.5,1.5,0.8,'#e67e22','Compliance'),(4.2,1.2,2.0,0.8,'#9b59b6','Reservoir')]:
+            ax.add_patch(mpatches.FancyBboxPatch((x,y),w,h,boxstyle='round,pad=0.1',facecolor=c,edgecolor='white',linewidth=2))
+            ax.text(x+w/2,y+h/2,lbl,ha='center',va='center',color='white',fontsize=7,fontweight='bold')
+        ax.add_patch(mpatches.FancyBboxPatch((4.8,7.4),2.0,0.5,boxstyle='round,pad=0.05',facecolor='#00cc44',edgecolor='white',linewidth=1))
+        ax.text(5.8,7.65,'GREEN LASER',ha='center',va='center',color='black',fontsize=8,fontweight='bold')
+        ax.annotate('',xy=(5.8,6.6),xytext=(5.8,7.4),arrowprops=dict(arrowstyle='->',color='#00ff44',lw=3))
+        ax.add_patch(mpatches.FancyBboxPatch((9.5,5.3),1.8,0.9,boxstyle='round,pad=0.05',facecolor='#1a2744',edgecolor='#7eb8f7',linewidth=2))
+        ax.text(10.4,5.75,'PIV Camera',ha='center',va='center',color='white',fontsize=7,fontweight='bold')
+        ax.text(6,0.4,'70 bpm  |  5 L/min  |  80-120 mmHg',color='#7eb8f7',fontsize=9,ha='center',fontweight='bold')
+    elif 'upad' in t or 'ckd' in t or 'creatinine' in t:
+        ax.set_title('uPAD Fabrication and CKD Detection - SJSU CardioLab', color='white', fontsize=13, fontweight='bold', pad=15)
+        steps = [('1.Whatman Paper','#4361ee'),('2.Wax Print','#e67e22'),('3.Heat 120C','#f1c40f'),('4.Add Reagent','#2ecc71'),('5.Apply Sample','#e63946'),('6.RGB Capture','#9b59b6')]
+        for i,(lbl,c) in enumerate(steps):
+            cx = 1.0+i*1.8
+            ax.add_patch(plt.Circle((cx,6.2),0.75,color=c,zorder=5))
+            ax.text(cx,6.2,lbl,ha='center',va='center',color='white',fontsize=6.5,fontweight='bold',zorder=6)
+            if i<len(steps)-1: ax.annotate('',xy=(cx+1.05,6.2),xytext=(cx+0.75,6.2),arrowprops=dict(arrowstyle='->',color='white',lw=2))
+        stages = [('Normal','<1.2','#27ae60'),('Borderline','1.2-1.5','#f1c40f'),('Stage 2','1.5-3.0','#e67e22'),('Stage 3-4','3.0-6.0','#e74c3c'),('Stage 5','>6.0','#922b21')]
+        for i,(s,r,c) in enumerate(stages):
+            ax.add_patch(mpatches.FancyBboxPatch((0.3+i*2.3,1.5),2.0,2.8,boxstyle='round,pad=0.1',facecolor=c,edgecolor='white',linewidth=2,alpha=0.85))
+            ax.text(1.3+i*2.3,3.1,s,ha='center',color='white',fontsize=9,fontweight='bold')
+            ax.text(1.3+i*2.3,2.6,r+' mg/dL',ha='center',color='white',fontsize=8)
+        ax.text(6,1.0,'Creatinine Level (mg/dL)',color='#7eb8f7',fontsize=9,ha='center',fontweight='bold')
+    elif 'mhv' in t or 'valve' in t:
+        ax.set_title('Mechanical Heart Valve (MHV) - 27mm SJM Regent Bileaflet', color='white', fontsize=13, fontweight='bold', pad=15)
+        ax.add_patch(plt.Circle((6,4),3.0,color='#2d3a5a',zorder=1))
+        ax.add_patch(plt.Circle((6,4),2.8,color='#1a2744',zorder=2))
+        ax.add_patch(plt.Circle((6,4),3.0,color='#7eb8f7',fill=False,linewidth=4,zorder=3))
+        ax.add_patch(mpatches.FancyBboxPatch((4.3,2.8),1.4,2.4,boxstyle='round,pad=0.1',facecolor='#e63946',edgecolor='white',linewidth=2,zorder=4))
+        ax.add_patch(mpatches.FancyBboxPatch((6.3,2.8),1.4,2.4,boxstyle='round,pad=0.1',facecolor='#e63946',edgecolor='white',linewidth=2,zorder=4))
+        ax.text(5.0,4.0,'Leaflet 1',ha='center',va='center',color='white',fontsize=8,fontweight='bold',zorder=5)
+        ax.text(7.0,4.0,'Leaflet 2',ha='center',va='center',color='white',fontsize=8,fontweight='bold',zorder=5)
+        ax.annotate('',xy=(6,7.5),xytext=(6,7.0),arrowprops=dict(arrowstyle='->',color='#4361ee',lw=4))
+        ax.text(6,7.7,'Blood Flow',color='#4361ee',ha='center',fontsize=10,fontweight='bold')
+        for i,(spec) in enumerate(['Diameter: 27mm','Material: Pyrolytic Carbon','Type: Bileaflet','Opening Angle: 85 deg']):
+            ax.text(0.3,2.0+i*0.7,'* '+spec,color='#a8b2d8',fontsize=8)
     else:
-        ax.set_title("CardioLab AI — Research Overview", color="white", fontsize=14, fontweight="bold", pad=15)
-        pillars = [
-            (2, 5, "#e63946", "Pillar 1
-MHV Design
-& Testing", ["MCL", "PIV", "FSI"]),
-            (5, 5, "#4361ee", "Pillar 2
-Thrombosis
-Research", ["TGT", "Blood Clots", "Hemolysis"]),
-            (8, 5, "#2ecc71", "Pillar 3
-CKD
-Diagnostics", ["uPAD", "Jaffe", "Creatinine"]),
-        ]
-        for x, y, color, title, items in pillars:
-            rect = mpatches.FancyBboxPatch((x-1, y-0.5), 2.5, 2.5, boxstyle="round,pad=0.2", facecolor=color, edgecolor="white", linewidth=2, alpha=0.9)
-            ax.add_patch(rect)
-            ax.text(x+0.25, y+1.7, title, ha="center", va="center", color="white", fontsize=9, fontweight="bold")
-            for j, item in enumerate(items):
-                ax.text(x+0.25, y+1-j*0.5, "• "+item, ha="center", color="white", fontsize=8)
-        equipment = ["27mm SJM Regent MHV", "Arduino Uno + Stepper Motor", "Time-resolved PIV (Green Laser)", "Sylgard 184 Transparent Sections", "Heska Element HT5 Analyzer"]
-        rect_eq = mpatches.FancyBboxPatch((1, 1), 10, 2.5, boxstyle="round,pad=0.2", facecolor="#1a2744", edgecolor="#7eb8f7", linewidth=2)
-        ax.add_patch(rect_eq)
-        ax.text(6, 3.2, "KEY EQUIPMENT", ha="center", color="#7eb8f7", fontsize=10, fontweight="bold")
-        for i, eq in enumerate(equipment):
-            col = i % 2
-            row = i // 2
-            ax.text(2+col*5, 2.7-row*0.7, "• "+eq, color="#e2e8f0", fontsize=7.5)
-
+        ax.set_title('CardioLab AI - Research Overview - SJSU Biomedical Engineering', color='white', fontsize=12, fontweight='bold', pad=15)
+        for i,(x,c,title,items) in enumerate([(2.0,'#e63946','MHV + PIV',['MCL Loop','PIV Laser','FSI COMSOL']),(5.5,'#4361ee','Thrombosis',['TGT Circuit','TAT PF1.2','Hemolysis']),(9.0,'#2ecc71','CKD Diag.',['uPAD Paper','Jaffe Rxn','Creatinine'])]):
+            ax.add_patch(mpatches.FancyBboxPatch((x-1.2,4.5),2.8,3.0,boxstyle='round,pad=0.2',facecolor=c,edgecolor='white',linewidth=2,alpha=0.9))
+            ax.text(x+0.2,7.0,title,ha='center',color='white',fontsize=10,fontweight='bold')
+            for j,it in enumerate(items): ax.text(x+0.2,6.3-j*0.6,'* '+it,ha='center',color='white',fontsize=8)
+        ax.add_patch(mpatches.FancyBboxPatch((0.5,1.0),11.0,3.0,boxstyle='round,pad=0.2',facecolor='#1a2744',edgecolor='#7eb8f7',linewidth=2))
+        ax.text(6,3.7,'KEY EQUIPMENT',ha='center',color='#7eb8f7',fontsize=10,fontweight='bold')
+        equip = ['27mm SJM Regent MHV','Arduino Uno + Stepper Motor','Time-resolved PIV Green Laser','Sylgard 184 Transparent MCL','Heska Element HT5 Analyzer']
+        for i,eq in enumerate(equip): ax.text(1.0+(i%2)*6,3.0-(i//2)*0.7,'* '+eq,color='#e2e8f0',fontsize=8)
     buf = io.BytesIO()
     plt.tight_layout()
-    plt.savefig(buf, format="png", facecolor=fig.get_facecolor(), bbox_inches="tight", dpi=120)
+    plt.savefig(buf,format='png',facecolor=fig.get_facecolor(),bbox_inches='tight',dpi=120)
     buf.seek(0)
     img = Image.open(buf)
     plt.close()
     return img
 
 def piv_tool(velocity, shear, hr):
-    v = "HIGH - stenosis risk" if float(velocity)>2.0 else "NORMAL"
-    s = "HIGH - thrombosis risk" if float(shear)>10 else "ELEVATED - monitor" if float(shear)>5 else "NORMAL"
-    hr_s = "ABNORMAL" if float(hr)<60 or float(hr)>100 else "NORMAL"
-    return ("PIV ANALYSIS RESULTS"+chr(10)+
-        "━━━━━━━━━━━━━━━━━━━━"+chr(10)+
-        "Velocity:   "+str(velocity)+" m/s  →  "+v+chr(10)+
-        "Shear:      "+str(shear)+" Pa   →  "+s+chr(10)+
-        "Heart Rate: "+str(hr)+" bpm →  "+hr_s)
+    v = 'HIGH - stenosis risk' if float(velocity)>2.0 else 'NORMAL'
+    s = 'HIGH - thrombosis risk' if float(shear)>10 else 'ELEVATED' if float(shear)>5 else 'NORMAL'
+    hr_s = 'ABNORMAL' if float(hr)<60 or float(hr)>100 else 'NORMAL'
+    return 'PIV RESULTS'+chr(10)+'Velocity: '+str(velocity)+' m/s - '+v+chr(10)+'Shear: '+str(shear)+' Pa - '+s+chr(10)+'HR: '+str(hr)+' bpm - '+hr_s
 
 def tgt_tool(tat,pf12,hemo,platelets,time):
     risk=sum([float(tat)>15,float(pf12)>2.0,float(hemo)>50,float(platelets)<150])
-    r="HIGH THROMBOGENIC RISK" if risk>=3 else "MODERATE RISK" if risk>=2 else "LOW RISK"
-    return ("TGT BLOOD ANALYSIS"+chr(10)+"━━━━━━━━━━━━━━━━━━━━"+chr(10)+
-        "Time:       "+str(time)+" min"+chr(10)+
-        "TAT:        "+str(tat)+"  →  "+("HIGH" if float(tat)>15 else "NORMAL")+chr(10)+
-        "PF1.2:      "+str(pf12)+"  →  "+("HIGH" if float(pf12)>2.0 else "NORMAL")+chr(10)+
-        "Hemoglobin: "+str(hemo)+"  →  "+("HIGH" if float(hemo)>50 else "NORMAL")+chr(10)+
-        "Platelets:  "+str(platelets)+"  →  "+("LOW" if float(platelets)<150 else "NORMAL")+chr(10)+
-        "━━━━━━━━━━━━━━━━━━━━"+chr(10)+"OVERALL: "+r)
+    r='HIGH THROMBOGENIC RISK' if risk>=3 else 'MODERATE RISK' if risk>=2 else 'LOW RISK'
+    return 'TGT ANALYSIS'+chr(10)+'Time: '+str(time)+' min'+chr(10)+'TAT: '+str(tat)+(' HIGH' if float(tat)>15 else ' NORMAL')+chr(10)+'PF1.2: '+str(pf12)+(' HIGH' if float(pf12)>2.0 else ' NORMAL')+chr(10)+'Hemo: '+str(hemo)+(' HIGH' if float(hemo)>50 else ' NORMAL')+chr(10)+'Platelets: '+str(platelets)+(' LOW' if float(platelets)<150 else ' NORMAL')+chr(10)+'RESULT: '+r
 
 def upad_tool(r,g,b):
     c=max(0,round(0.02*(float(r)-float(b))-0.5,2))
-    if c<1.2: s="Normal - No CKD"
-    elif c<1.5: s="Borderline - Monitor"
-    elif c<3.0: s="Stage 2 CKD"
-    elif c<6.0: s="Stage 3-4 CKD"
-    else: s="Stage 5 CKD - Kidney Failure"
-    return ("uPAD CKD ANALYSIS"+chr(10)+"━━━━━━━━━━━━━━━━━━━━"+chr(10)+
-        "RGB:        R="+str(r)+" G="+str(g)+" B="+str(b)+chr(10)+
-        "Creatinine: "+str(c)+" mg/dL"+chr(10)+
-        "━━━━━━━━━━━━━━━━━━━━"+chr(10)+"CKD STAGE: "+s+chr(10)+
-        "Confirm with: Heska Element HT5")
-
-with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
-    gr.HTML("""
-    <div class="main-header">
-        <div style="font-size:2.8em;font-weight:900;color:#ffffff;letter-spacing:3px;text-shadow:0 0 30px rgba(255,255,255,0.3);">
-            ❤️ CardioLab AI
-        </div>
-    </div>
-    """)
+    s='Normal' if c<1.2 else 'Borderline' if c<1.5 else 'Stage 2 CKD' if c<3.0 else 'Stage 3-4 CKD' if c<6.0 else 'Stage 5 CKD'
+    return 'uPAD RESULT'+chr(10)+'Creatinine: '+str(c)+' mg/dL'+chr(10)+'CKD Stage: '+s
 
+with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
+    gr.HTML('<div style="background:linear-gradient(135deg,#1a237e,#b71c1c);padding:25px;text-align:center;border-radius:12px 12px 0 0;"><div style="font-size:2.8em;font-weight:900;color:#fff;letter-spacing:3px;">❤️ CardioLab AI</div></div>')
     with gr.Tabs():
-        with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(label="", height=450)
+        with gr.Tab('💬 Chat'):
+            chatbot = gr.Chatbot(label='', height=450)
             with gr.Row():
-                msg_box = gr.Textbox(placeholder="Ask anything about CardioLab research...", label="", lines=2, scale=4)
+                msg_box = gr.Textbox(placeholder='Ask anything about CardioLab research...', label='', lines=2, scale=4)
                 with gr.Column(scale=1, min_width=100):
-                    send_btn = gr.Button("Send ➤", variant="primary")
-                    clear_btn = gr.Button("Clear", variant="secondary")
+                    send_btn = gr.Button('Send', variant='primary')
+                    clear_btn = gr.Button('Clear', variant='secondary')
             send_btn.click(research_chat, inputs=[msg_box, chatbot], outputs=[msg_box, chatbot])
             msg_box.submit(research_chat, inputs=[msg_box, chatbot], outputs=[msg_box, chatbot])
-            clear_btn.click(lambda: ([], ""), outputs=[chatbot, msg_box])
-
-        with gr.Tab("🎙️ Voice"):
-            gr.Markdown("### Speak your research question — powered by Groq Whisper AI")
-            voice_chatbot = gr.Chatbot(label="", height=350)
-            audio_input = gr.Audio(sources=["microphone"], type="filepath", label="Record Question")
+            clear_btn.click(lambda: ([], ''), outputs=[chatbot, msg_box])
+        with gr.Tab('🎙️ Voice'):
+            gr.Markdown('### Speak your question - Groq Whisper AI')
+            voice_chatbot = gr.Chatbot(label='', height=350)
+            audio_input = gr.Audio(sources=['microphone'], type='filepath', label='Record Question')
             with gr.Row():
-                voice_btn = gr.Button("Ask by Voice ➤", variant="primary")
-                voice_clear = gr.Button("Clear", variant="secondary")
+                voice_btn = gr.Button('Ask by Voice', variant='primary')
+                voice_clear = gr.Button('Clear', variant='secondary')
             voice_btn.click(voice_chat, inputs=[audio_input, voice_chatbot], outputs=voice_chatbot)
             voice_clear.click(lambda: [], outputs=voice_chatbot)
-
-        with gr.Tab("🔍 Papers"):
-            gr.Markdown("### Search latest research papers with verified working links")
+        with gr.Tab('🔍 Papers'):
+            gr.Markdown('### Search research papers - verified links only')
             with gr.Row():
-                search_input = gr.Textbox(placeholder="e.g. mechanical heart valve thrombogenicity", label="Research Topic", scale=4)
-                search_btn = gr.Button("Search ➤", variant="primary", scale=1)
-            search_output = gr.Textbox(label="Results — Verified Links Only", lines=18)
+                search_input = gr.Textbox(placeholder='e.g. mechanical heart valve thrombogenicity', label='Research Topic', scale=4)
+                search_btn = gr.Button('Search', variant='primary', scale=1)
+            search_output = gr.Textbox(label='Verified Results', lines=18)
             search_btn.click(quick_search, inputs=search_input, outputs=search_output)
             search_input.submit(quick_search, inputs=search_input, outputs=search_output)
-            gr.Markdown("**Try:** `TGT thrombogenicity` | `creatinine uPAD` | `bileaflet MHV PIV` | `CKD point-of-care`")
-
-        with gr.Tab("🎨 Diagrams"):
-            gr.Markdown("### Generate real visual diagrams of CardioLab equipment and systems")
+        with gr.Tab('🎨 Diagrams'):
+            gr.Markdown('### Real visual diagrams - try: TGT setup | MCL PIV | uPAD CKD | MHV valve | CardioLab overview')
             with gr.Row():
-                diagram_input = gr.Textbox(placeholder="e.g. TGT setup | MCL PIV system | uPAD CKD | MHV bileaflet valve | CardioLab overview", label="What to diagram", scale=4)
-                diagram_btn = gr.Button("Generate ➤", variant="primary", scale=1)
-            diagram_output = gr.Image(label="Visual Diagram", type="pil")
+                diagram_input = gr.Textbox(placeholder='e.g. TGT setup', label='What to diagram', scale=4)
+                diagram_btn = gr.Button('Generate', variant='primary', scale=1)
+            diagram_output = gr.Image(label='Visual Diagram', type='pil')
             diagram_btn.click(generate_diagram, inputs=diagram_input, outputs=diagram_output)
-            gr.Markdown("**Try these:** `TGT setup` | `MCL PIV system` | `uPAD CKD flow` | `MHV bileaflet valve` | `CardioLab overview`")
-
-        with gr.Tab("📊 PIV"):
-            gr.Markdown("### Analyze Particle Image Velocimetry data from Mock Circulatory Loop")
+        with gr.Tab('📊 PIV'):
             with gr.Row():
                 with gr.Column():
-                    v=gr.Number(label="Max Velocity (m/s)", value=1.8, info="Normal: 0.5-2.0 m/s")
-                    s=gr.Number(label="Wall Shear Stress (Pa)", value=6.5, info="Normal: below 5 Pa")
-                    h=gr.Number(label="Heart Rate (bpm)", value=72, info="Normal: 60-100 bpm")
-                    piv_btn = gr.Button("Analyze PIV ➤", variant="primary")
-                with gr.Column():
-                    piv_out=gr.Textbox(label="Analysis Result", lines=8)
-            piv_btn.click(piv_tool, inputs=[v,s,h], outputs=piv_out)
-
-        with gr.Tab("🩸 TGT"):
-            gr.Markdown("### Interpret Thrombogenicity Tester blood analysis results")
+                    v=gr.Number(label='Max Velocity m/s', value=1.8)
+                    s=gr.Number(label='Wall Shear Stress Pa', value=6.5)
+                    h=gr.Number(label='Heart Rate bpm', value=72)
+                    piv_out=gr.Textbox(label='Result', lines=5)
+                    gr.Button('Analyze PIV', variant='primary').click(piv_tool,inputs=[v,s,h],outputs=piv_out)
+        with gr.Tab('🩸 TGT'):
             with gr.Row():
                 with gr.Column():
-                    t1=gr.Number(label="TAT (ng/mL)", value=18, info="Normal: below 8")
-                    t2=gr.Number(label="PF1.2 (nmol/L)", value=2.5, info="Normal: below 2.0")
-                    t3=gr.Number(label="Free Hemoglobin (mg/L)", value=60, info="Normal: below 20")
-                    t4=gr.Number(label="Platelet Count", value=140, info="Normal: above 150")
-                    t5=gr.Number(label="Time (minutes)", value=40)
-                    tgt_btn = gr.Button("Analyze TGT ➤", variant="primary")
-                with gr.Column():
-                    out2=gr.Textbox(label="Analysis Result", lines=12)
-            tgt_btn.click(tgt_tool, inputs=[t1,t2,t3,t4,t5], outputs=out2)
-
-        with gr.Tab("🧪 uPAD"):
-            gr.Markdown("### Analyze uPAD colorimetric result using Jaffe Reaction for CKD diagnosis")
+                    t1=gr.Number(label='TAT ng/mL', value=18)
+                    t2=gr.Number(label='PF1.2 nmol/L', value=2.5)
+                    t3=gr.Number(label='Free Hemoglobin mg/L', value=60)
+                    t4=gr.Number(label='Platelet Count', value=140)
+                    t5=gr.Number(label='Time minutes', value=40)
+                    out2=gr.Textbox(label='Result', lines=8)
+                    gr.Button('Analyze TGT', variant='primary').click(tgt_tool,inputs=[t1,t2,t3,t4,t5],outputs=out2)
+        with gr.Tab('🧪 uPAD'):
             with gr.Row():
                 with gr.Column():
-                    r=gr.Number(label="R (Red)", value=210, info="Range: 0-255")
-                    g=gr.Number(label="G (Green)", value=140, info="Range: 0-255")
-                    b=gr.Number(label="B (Blue)", value=80, info="Range: 0-255")
-                    upad_btn = gr.Button("Analyze uPAD ➤", variant="primary")
-                with gr.Column():
-                    out3=gr.Textbox(label="CKD Analysis Result", lines=10)
-            upad_btn.click(upad_tool, inputs=[r,g,b], outputs=out3)
-
-demo.launch()
+                    r=gr.Number(label='R value', value=210)
+                    g=gr.Number(label='G value', value=140)
+                    b=gr.Number(label='B value', value=80)
+                    out3=gr.Textbox(label='Result', lines=5)
+                    gr.Button('Analyze uPAD', variant='primary').click(upad_tool,inputs=[r,g,b],outputs=out3)
+demo.launch()