Upload app.py with huggingface_hub

app.py

@@ -1,7 +1,14 @@
 import gradio as gr
 import os
 import requests
+import matplotlib
+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", "")
 
@@ -13,176 +20,20 @@ KNOWHOW = ("SJSU CardioLab: "
     "Equipment: Heska HT5 hematology analyzer time-resolved PIV Tygon tubing Arduino Uno.")
 
 CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700;900&display=swap');
-
-* { font-family: 'Inter', sans-serif !important; box-sizing: border-box; }
-
-body, .gradio-container {
-    background: #f0f4f8 !important;
-    color: #1a202c !important;
-}
-
-.gradio-container {
-    max-width: 1200px !important;
-    margin: 0 auto !important;
-    padding: 0 !important;
-}
-
-/* HEADER */
-.main-header {
-    background: linear-gradient(135deg, #1a237e 0%, #b71c1c 100%);
-    border-bottom: 3px solid #e63946;
-    padding: 30px 20px;
-    text-align: center;
-    margin-bottom: 0;
-}
-
-/* TABS - ALWAYS VISIBLE */
-.tab-nav {
-    background: #ffffff !important;
-    border-bottom: 2px solid #e2e8f0 !important;
-    padding: 0 10px !important;
-    display: flex !important;
-    gap: 5px !important;
-    overflow-x: auto !important;
-}
-
-.tab-nav button {
-    background: #f7fafc !important;
-    color: #2d3748 !important;
-    border: 1px solid #e2e8f0 !important;
-    border-radius: 8px 8px 0 0 !important;
-    padding: 12px 20px !important;
-    font-size: 0.9em !important;
-    font-weight: 600 !important;
-    cursor: pointer !important;
-    transition: all 0.2s !important;
-    white-space: nowrap !important;
-    margin-top: 8px !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;
-}
-
-/* CONTENT AREA */
-.tabitem {
-    background: #ffffff !important;
-    padding: 20px !important;
-    border: 1px solid #e2e8f0 !important;
-    border-top: none !important;
-}
-
-/* INPUTS */
-textarea, input[type=number], input[type=text] {
-    background: #f7fafc !important;
-    color: #1a202c !important;
-    border: 1px solid #cbd5e0 !important;
-    border-radius: 8px !important;
-    font-size: 0.95em !important;
-}
-
-textarea:focus, input:focus {
-    border-color: #e63946 !important;
-    outline: none !important;
-}
-
-textarea::placeholder {
-    color: #a0aec0 !important;
-}
-
-/* BUTTONS */
-button.primary, .btn-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;
-    font-size: 0.95em !important;
-    cursor: pointer !important;
-    transition: transform 0.1s !important;
-}
-
-button.primary:hover {
-    transform: translateY(-1px) !important;
-    box-shadow: 0 4px 15px rgba(230, 57, 70, 0.4) !important;
-}
-
-button.secondary {
-    background: #edf2f7 !important;
-    color: #4a5568 !important;
-    border: 1px solid #cbd5e0 !important;
-    border-radius: 8px !important;
-    font-weight: 600 !important;
-}
-
-/* LABELS */
-label span, label, .label-wrap span {
-    color: #2b6cb0 !important;
-    font-weight: 600 !important;
-    font-size: 0.85em !important;
-    text-transform: uppercase !important;
-    letter-spacing: 0.5px !important;
-}
-
-/* MARKDOWN TEXT */
-.md h1, .md h2, .md h3 { color: #ffffff !important; }
-.md p, .md li { color: #a0b3d6 !important; }
-
-/* CHATBOT */
-.chatbot {
-    background: #0a1628 !important;
-    border: 1px solid #1e3a6e !important;
-    border-radius: 12px !important;
-}
-
-.message.user {
-    background: linear-gradient(135deg, #e63946, #c1121f) !important;
-    color: #ffffff !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;
-}
-
-/* RESULT BOXES */
-.gr-box, .block {
-    background: #ffffff !important;
-    border: 1px solid #e2e8f0 !important;
-    border-radius: 12px !important;
-}
-
-/* NUMBERS */
-input[type=number] {
-    color: #7eb8f7 !important;
-    font-size: 1.1em !important;
-    font-weight: 600 !important;
-}
-
-/* INFO TEXT */
-.info {
-    color: #5a7aaa !important;
-    font-size: 0.8em !important;
-}
-
-/* SCROLLBAR */
-::-webkit-scrollbar { width: 6px; }
-::-webkit-scrollbar-track { background: #f7fafc; }
-::-webkit-scrollbar-thumb { background: #cbd5e0; border-radius: 3px; }
+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; }
+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; }
 """
 
 def get_pubmed(query, n=5):
@@ -228,8 +79,7 @@ def research_chat(message, history):
     try:
         client = Groq(api_key=GROQ_KEY)
         pubmed = get_pubmed(message, n=3)
-        system_msg = "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":system_msg}]
+        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}]
         for item in history:
             if isinstance(item, dict):
                 msgs.append({"role":item["role"],"content":item["content"]})
@@ -272,15 +122,167 @@ def voice_chat(audio, history):
         return history
 
 def generate_diagram(topic):
-    if not topic.strip(): return "Please enter a topic."
-    try:
-        client = Groq(api_key=GROQ_KEY)
-        msgs = [{"role":"system","content":"You are a biomedical engineering expert for SJSU CardioLab. Generate a detailed ASCII diagram or technical schematic. Make it clear, labeled, and useful for researchers."}]
-        msgs.append({"role":"user","content":"Create a detailed labeled diagram for: "+topic})
-        resp = client.chat.completions.create(model="llama-3.3-70b-versatile",messages=msgs,max_tokens=600)
-        return resp.choices[0].message.content
-    except Exception as e:
-        return "Error: "+str(e)
+    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"),
+        ]
+        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)
+
+    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)
+
+    buf = io.BytesIO()
+    plt.tight_layout()
+    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"
@@ -295,15 +297,13 @@ def piv_tool(velocity, shear, hr):
 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)+
+    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)
+        "━━━━━━━━━━━━━━━━━━━━"+chr(10)+"OVERALL: "+r)
 
 def upad_tool(r,g,b):
     c=max(0,round(0.02*(float(r)-float(b))-0.5,2))
@@ -312,20 +312,16 @@ def upad_tool(r,g,b):
     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 Input:   R="+str(r)+" G="+str(g)+" B="+str(b)+chr(10)+
-        "Orange Score: "+str(int(float(r)-float(b)))+chr(10)+
-        "Creatinine:  "+str(c)+" mg/dL"+chr(10)+
-        "━━━━━━━━━━━━━━━━━━━━"+chr(10)+
-        "CKD STAGE: "+s+chr(10)+
+    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:#e63946; letter-spacing:3px; text-shadow: 0 0 30px rgba(230,57,70,0.5);">
+        <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>
@@ -335,12 +331,7 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
         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")
@@ -361,38 +352,30 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
         with gr.Tab("🔍 Papers"):
             gr.Markdown("### Search latest research papers with verified working links")
             with gr.Row():
-                search_input = gr.Textbox(
-                    placeholder="e.g. mechanical heart valve thrombogenicity",
-                    label="Research Topic",
-                    scale=4
-                )
+                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_btn.click(quick_search, inputs=search_input, outputs=search_output)
             search_input.submit(quick_search, inputs=search_input, outputs=search_output)
-            gr.Markdown("**Try:** `mechanical heart valve PIV` | `creatinine uPAD microfluidic` | `bileaflet MHV thrombogenicity`")
+            gr.Markdown("**Try:** `TGT thrombogenicity` | `creatinine uPAD` | `bileaflet MHV PIV` | `CKD point-of-care`")
 
         with gr.Tab("🎨 Diagrams"):
-            gr.Markdown("### Generate technical diagrams and schematics for CardioLab")
+            gr.Markdown("### Generate real visual diagrams of CardioLab equipment and systems")
             with gr.Row():
-                diagram_input = gr.Textbox(
-                    placeholder="e.g. Mock Circulatory Loop with PIV system",
-                    label="What to diagram",
-                    scale=4
-                )
+                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.Textbox(label="Technical Diagram", lines=18)
+            diagram_output = gr.Image(label="Visual Diagram", type="pil")
             diagram_btn.click(generate_diagram, inputs=diagram_input, outputs=diagram_output)
-            gr.Markdown("**Try:** `TGT circuit diagram` | `uPAD fabrication steps` | `PIV optical setup` | `MCL flow loop`")
+            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.Row():
                 with gr.Column():
-                    v=gr.Number(label="Max Velocity (m/s)", value=1.8, info="Normal range: 0.5 - 2.0 m/s")
+                    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 Data ➤", variant="primary")
+                    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)
@@ -404,9 +387,9 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
                     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 (10³/μL)", value=140, info="Normal: above 150")
+                    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 Results ➤", variant="primary")
+                    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)
@@ -415,10 +398,9 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
             gr.Markdown("### Analyze uPAD colorimetric result using Jaffe Reaction for CKD diagnosis")
             with gr.Row():
                 with gr.Column():
-                    gr.Markdown("**Enter RGB values from 64x64 pixel detection zone**")
-                    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")
+                    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)