Upload app.py with huggingface_hub

app.py

@@ -21,206 +21,149 @@ KNOWHOW = ("MCL: Sylgard 184 PDMS 10:1 ratio 48hr cure green laser PIV 70bpm 5L/
     "Equipment: Heska HT5 hematology analyzer time-resolved PIV Tygon tubing Arduino Uno.")
 
 CSS = """
-/* Reset and base */
 body, .gradio-container {
     background: #f7f7f8 !important;
     font-family: -apple-system, BlinkMacSystemFont, Segoe UI, sans-serif !important;
-    margin: 0 !important;
-    padding: 0 !important;
+    margin: 0 !important; padding: 0 !important;
 }
-
-/* Hide default gradio header */
-.gradio-container > .main > .wrap > .panel {
-    border: none !important;
-    box-shadow: none !important;
-}
-
-/* CHATGPT STYLE SIDEBAR */
-.sidebar {
-    background: #202123 !important;
-    min-height: 100vh !important;
-    padding: 10px !important;
-    border-right: 1px solid #3a3a3a !important;
-}
-
-.sidebar-title {
-    color: white !important;
-    font-size: 1.1em !important;
-    font-weight: 700 !important;
-    padding: 10px 5px !important;
-    border-bottom: 1px solid #3a3a3a !important;
-    margin-bottom: 10px !important;
-}
-
-.session-item {
-    background: #2d2d30 !important;
-    color: #ececf1 !important;
-    border-radius: 6px !important;
-    padding: 8px 12px !important;
-    margin-bottom: 4px !important;
-    cursor: pointer !important;
-    font-size: 0.85em !important;
-}
-
-.session-item:hover {
-    background: #3a3a3c !important;
-}
-
-/* TABS - TOP NAV STYLE */
 .tab-nav {
     background: #ffffff !important;
     border-bottom: 1px solid #e5e7eb !important;
     padding: 0 16px !important;
     display: flex !important;
-    flex-wrap: nowrap !important;
-    overflow-x: auto !important;
+    flex-wrap: wrap !important;
     gap: 0 !important;
 }
-
 .tab-nav button {
     background: transparent !important;
     color: #6b7280 !important;
     border: none !important;
     border-bottom: 2px solid transparent !important;
-    padding: 12px 16px !important;
+    padding: 12px 14px !important;
     font-weight: 500 !important;
-    font-size: 0.85em !important;
+    font-size: 0.82em !important;
     white-space: nowrap !important;
     border-radius: 0 !important;
-    margin: 0 !important;
-    transition: all 0.15s !important;
-}
-
-.tab-nav button:hover {
-    color: #111827 !important;
-    background: #f9fafb !important;
-}
-
-.tab-nav button.selected {
-    color: #e63946 !important;
-    border-bottom: 2px solid #e63946 !important;
-    font-weight: 600 !important;
-    background: transparent !important;
-}
-
-/* MAIN CHAT AREA */
-.chat-container {
-    background: #ffffff !important;
-    min-height: 80vh !important;
-}
-
-/* CHATBOT MESSAGES */
-.chatbot {
-    background: #ffffff !important;
-    border: none !important;
-    border-radius: 0 !important;
-}
-
-.message.user {
-    background: #f7f7f8 !important;
-    color: #1a202c !important;
-    border-radius: 12px !important;
-    padding: 12px 16px !important;
-    margin: 4px 0 !important;
-}
-
-.message.bot {
-    background: #ffffff !important;
-    color: #1a202c !important;
-    border-radius: 12px !important;
-    padding: 12px 16px !important;
-    margin: 4px 0 !important;
-    border-left: 3px solid #e63946 !important;
-}
-
-/* INPUT AREA */
-textarea {
-    background: #ffffff !important;
-    color: #1a202c !important;
-    border: 1px solid #d1d5db !important;
-    border-radius: 12px !important;
-    font-size: 0.95em !important;
-    padding: 12px !important;
-    box-shadow: 0 1px 3px rgba(0,0,0,0.1) !important;
-}
-
-textarea:focus {
-    border-color: #e63946 !important;
-    box-shadow: 0 0 0 2px rgba(230,57,70,0.1) !important;
-    outline: none !important;
-}
-
-/* BUTTONS */
-button.primary {
-    background: #e63946 !important;
-    color: white !important;
-    border: none !important;
-    border-radius: 8px !important;
-    font-weight: 600 !important;
-    padding: 10px 20px !important;
-    transition: background 0.15s !important;
-}
-
-button.primary:hover {
-    background: #c1121f !important;
-}
-
-button.secondary {
-    background: #f3f4f6 !important;
-    color: #374151 !important;
-    border: 1px solid #d1d5db !important;
-    border-radius: 8px !important;
-    font-weight: 500 !important;
-}
-
-/* NEW CHAT BUTTON */
-.new-chat-btn {
-    background: transparent !important;
-    color: #ececf1 !important;
-    border: 1px solid #3a3a3c !important;
-    border-radius: 6px !important;
-    padding: 8px 12px !important;
-    width: 100% !important;
-    text-align: left !important;
-    margin-bottom: 8px !important;
-    font-size: 0.85em !important;
-}
-
-/* DROPDOWN */
-select, .gr-dropdown {
-    background: #2d2d30 !important;
-    color: #ececf1 !important;
-    border: 1px solid #3a3a3c !important;
-    border-radius: 6px !important;
-}
-
-/* INPUT NUMBERS */
-input[type=number] {
-    background: #f9fafb !important;
-    color: #1a202c !important;
-    border: 1px solid #d1d5db !important;
-    border-radius: 8px !important;
-}
-
-/* LABELS */
-label span {
-    color: #374151 !important;
-    font-weight: 500 !important;
-    font-size: 0.85em !important;
 }
+.tab-nav button:hover { color: #111827 !important; background: #f9fafb !important; }
+.tab-nav button.selected { color: #c1121f !important; border-bottom: 2px solid #c1121f !important; font-weight: 700 !important; background: transparent !important; }
+.message.user { background: #f3f4f6 !important; color: #1a202c !important; border-radius: 12px !important; }
+.message.bot { background: #ffffff !important; color: #1a202c !important; border-left: 3px solid #c1121f !important; border-radius: 0 12px 12px 12px !important; }
+textarea { background: #ffffff !important; color: #1a202c !important; border: 1px solid #d1d5db !important; border-radius: 12px !important; }
+textarea:focus { border-color: #c1121f !important; outline: none !important; }
+button.primary { background: #c1121f !important; color: white !important; border: none !important; border-radius: 8px !important; font-weight: 600 !important; }
+button.primary:hover { background: #a00e18 !important; }
+button.secondary { background: #f3f4f6 !important; color: #374151 !important; border: 1px solid #d1d5db !important; border-radius: 8px !important; }
+input[type=number] { background: #f9fafb !important; color: #1a202c !important; border: 1px solid #d1d5db !important; border-radius: 8px !important; }
+label span { color: #374151 !important; font-weight: 500 !important; font-size: 0.85em !important; }
+::-webkit-scrollbar { width: 5px; } ::-webkit-scrollbar-thumb { background: #d1d5db; border-radius: 3px; }
+"""
 
-/* FILE UPLOAD */
-.file-preview {
-    background: #f9fafb !important;
-    border: 2px dashed #d1d5db !important;
-    border-radius: 12px !important;
-}
+HEADER_HTML = """
+<div style="
+    background: linear-gradient(135deg, #0a0f2e 0%, #1a0a0a 40%, #0a0f2e 100%);
+    padding: 0;
+    margin: 0;
+    border-bottom: 3px solid #c1121f;
+    position: relative;
+    overflow: hidden;
+">
+    <!-- ECG Background Line -->
+    <svg style="position:absolute;top:0;left:0;width:100%;height:100%;opacity:0.08;" viewBox="0 0 1200 120" preserveAspectRatio="none">
+        <polyline points="0,60 100,60 130,20 150,100 170,10 200,90 220,60 400,60 430,20 450,100 470,10 500,90 520,60 700,60 730,20 750,100 770,10 800,90 820,60 1000,60 1030,20 1050,100 1070,10 1100,90 1120,60 1200,60"
+            fill="none" stroke="#c1121f" stroke-width="3"/>
+    </svg>
+
+    <div style="
+        max-width: 1200px;
+        margin: 0 auto;
+        padding: 18px 24px;
+        display: flex;
+        align-items: center;
+        justify-content: space-between;
+        position: relative;
+        z-index: 1;
+    ">
+        <!-- LEFT: SJSU Spartan Logo SVG -->
+        <div style="display:flex;align-items:center;gap:16px;">
+            <svg width="60" height="60" viewBox="0 0 100 100" xmlns="http://www.w3.org/2000/svg">
+                <!-- Spartan helmet simplified -->
+                <circle cx="50" cy="35" r="28" fill="#0057a8" opacity="0.9"/>
+                <!-- Helmet crest -->
+                <ellipse cx="50" cy="14" rx="22" ry="10" fill="#0057a8"/>
+                <!-- Crest spikes -->
+                <polygon points="30,14 33,4 36,14" fill="#e8a020"/>
+                <polygon points="36,12 39,2 42,12" fill="#e8a020"/>
+                <polygon points="42,11 45,1 48,11" fill="#e8a020"/>
+                <polygon points="48,11 51,1 54,11" fill="#e8a020"/>
+                <polygon points="54,12 57,2 60,12" fill="#e8a020"/>
+                <polygon points="60,14 63,4 66,14" fill="#e8a020"/>
+                <!-- Helmet face -->
+                <rect x="36" y="30" width="28" height="22" rx="4" fill="#0057a8"/>
+                <rect x="40" y="35" width="8" height="12" rx="2" fill="#e8a020"/>
+                <!-- Chin guard -->
+                <rect x="34" y="50" width="32" height="8" rx="4" fill="#0057a8"/>
+                <!-- Helmet shine -->
+                <ellipse cx="42" cy="28" rx="5" ry="3" fill="white" opacity="0.25"/>
+            </svg>
+
+            <div>
+                <div style="color:#9ca3af;font-size:0.7em;font-weight:500;letter-spacing:2px;text-transform:uppercase;">San Jose State University</div>
+                <div style="color:#e8a020;font-size:0.85em;font-weight:700;letter-spacing:1px;">Biomedical Engineering</div>
+            </div>
+        </div>
+
+        <!-- CENTER: CardioLab AI Branding -->
+        <div style="text-align:center;flex:1;padding:0 20px;">
+            <!-- ECG + Heart icon inline -->
+            <div style="display:flex;align-items:center;justify-content:center;gap:12px;margin-bottom:4px;">
+                <svg width="120" height="32" viewBox="0 0 120 32">
+                    <polyline points="0,16 20,16 26,4 30,28 34,2 38,26 44,16 120,16"
+                        fill="none" stroke="#c1121f" stroke-width="2.5" stroke-linecap="round" stroke-linejoin="round"/>
+                    <!-- Heart dot on ECG -->
+                    <circle cx="34" cy="2" r="3" fill="#c1121f"/>
+                </svg>
+                <div style="font-size:2.2em;font-weight:900;letter-spacing:2px;">
+                    <span style="color:#ffffff;">Cardio</span><span style="color:#c1121f;">Lab</span><span style="color:#ffffff;"> AI</span>
+                </div>
+                <svg width="120" height="32" viewBox="0 0 120 32" style="transform:scaleX(-1);">
+                    <polyline points="0,16 20,16 26,4 30,28 34,2 38,26 44,16 120,16"
+                        fill="none" stroke="#c1121f" stroke-width="2.5" stroke-linecap="round" stroke-linejoin="round"/>
+                    <circle cx="34" cy="2" r="3" fill="#c1121f"/>
+                </svg>
+            </div>
+            <div style="color:#9ca3af;font-size:0.72em;letter-spacing:3px;text-transform:uppercase;">
+                AI Research Agent &nbsp;|&nbsp; Built on Biomni Stanford &nbsp;|&nbsp; Llama 3.3 70B
+            </div>
+        </div>
+
+        <!-- RIGHT: Heart + Stats -->
+        <div style="display:flex;align-items:center;gap:16px;">
+            <div style="text-align:right;">
+                <div style="color:#9ca3af;font-size:0.7em;letter-spacing:1px;text-transform:uppercase;">Research Pillars</div>
+                <div style="color:#ffffff;font-size:0.75em;margin-top:4px;">🫀 MHV &nbsp; 🔬 CKD &nbsp; 💻 FSI</div>
+                <div style="color:#9ca3af;font-size:0.65em;margin-top:2px;">MCL · PIV · TGT · uPAD · COMSOL</div>
+            </div>
+            <!-- Heart SVG -->
+            <svg width="50" height="50" viewBox="0 0 100 90" xmlns="http://www.w3.org/2000/svg">
+                <path d="M50 85 C50 85 5 55 5 30 C5 15 18 5 30 5 C38 5 45 9 50 15 C55 9 62 5 70 5 C82 5 95 15 95 30 C95 55 50 85 50 85Z"
+                    fill="#c1121f" opacity="0.9"/>
+                <path d="M50 75 C50 75 12 50 12 30 C12 18 22 12 30 12 C38 12 45 16 50 22"
+                    fill="none" stroke="rgba(255,255,255,0.3)" stroke-width="3"/>
+                <!-- ECG inside heart -->
+                <polyline points="25,45 32,45 35,35 38,55 41,30 44,50 50,45 75,45"
+                    fill="none" stroke="white" stroke-width="2.5" stroke-linecap="round" opacity="0.9"/>
+            </svg>
+        </div>
+    </div>
 
-/* SCROLLBAR */
-::-webkit-scrollbar { width: 6px; height: 6px; }
-::-webkit-scrollbar-track { background: transparent; }
-::-webkit-scrollbar-thumb { background: #d1d5db; border-radius: 3px; }
-::-webkit-scrollbar-thumb:hover { background: #9ca3af; }
+    <!-- Bottom accent bar -->
+    <div style="
+        height: 3px;
+        background: linear-gradient(90deg, #0057a8, #c1121f, #e8a020, #c1121f, #0057a8);
+        margin: 0;
+    "></div>
+</div>
 """
 
 def load_all_sessions():
@@ -235,42 +178,37 @@ def save_all_sessions(sessions):
     try:
         api = HfApi(token=HF_TOKEN)
         api.upload_file(path_or_fileobj=json.dumps(sessions, indent=2).encode(), path_in_repo="chat_history.json",
-            repo_id=HISTORY_REPO, repo_type="dataset", token=HF_TOKEN, commit_message="Update chat history")
+            repo_id=HISTORY_REPO, repo_type="dataset", token=HF_TOKEN, commit_message="Update")
         return True
     except: return False
 
 def get_session_list():
-    sessions = load_all_sessions()
-    if not sessions: return ["No saved sessions"]
-    return list(reversed(list(sessions.keys())))
+    s = load_all_sessions()
+    if not s: return ["No saved sessions"]
+    return list(reversed(list(s.keys())))
 
-def save_session(history, session_name):
+def save_session(history, name):
     if not history: return "Nothing to save", gr.update()
-    if not session_name or not session_name.strip():
-        session_name = "Chat " + datetime.now().strftime("%b %d %H:%M")
+    if not name or not name.strip(): name = "Chat "+datetime.now().strftime("%b %d %H:%M")
     sessions = load_all_sessions()
-    sessions[session_name] = {"messages": history, "saved_at": datetime.now().isoformat()}
+    sessions[name] = {"messages":history,"saved_at":datetime.now().isoformat()}
     ok = save_all_sessions(sessions)
     choices = get_session_list()
-    if ok: return "Saved: "+session_name, gr.update(choices=choices, value=session_name)
-    return "Save failed — check HF_TOKEN", gr.update()
+    return ("Saved: "+name if ok else "Save failed"), gr.update(choices=choices, value=name)
 
-def load_session(session_name):
-    if not session_name or "No saved" in session_name: return [], "Select a session first"
+def load_session(name):
+    if not name or "No saved" in name: return [], "Select a session"
     sessions = load_all_sessions()
-    if session_name in sessions:
-        msgs = sessions[session_name]["messages"]
-        return msgs, "Loaded: "+session_name
-    return [], "Session not found"
+    if name in sessions: return sessions[name]["messages"], "Loaded: "+name
+    return [], "Not found"
 
-def delete_session(session_name):
-    if not session_name or "No saved" in session_name: return "Select a session first", gr.update()
+def delete_session(name):
+    if not name or "No saved" in name: return "Select a session", gr.update()
     sessions = load_all_sessions()
-    if session_name in sessions:
-        del sessions[session_name]
-        save_all_sessions(sessions)
+    if name in sessions:
+        del sessions[name]; save_all_sessions(sessions)
         choices = get_session_list()
-        return "Deleted: "+session_name, gr.update(choices=choices, value=choices[0] if choices else None)
+        return "Deleted: "+name, gr.update(choices=choices, value=choices[0] if choices else None)
     return "Not found", gr.update()
 
 def new_chat(): return [], "", "New chat started"
@@ -302,7 +240,7 @@ def research_chat(message, history):
         return "", history
     try:
         client = Groq(api_key=GROQ_KEY)
-        msgs = [{"role":"system","content":"You are CardioLab AI assistant for SJSU Biomedical Engineering. Expert in MHV MCL PIV TGT uPAD CKD FSI. Remember full conversation. Never invent URLs. "+KNOWHOW}]
+        msgs = [{"role":"system","content":"You are CardioLab AI for SJSU Biomedical Engineering. Expert in MHV MCL PIV TGT uPAD CKD FSI. Remember 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})
@@ -342,72 +280,65 @@ def analyze_upad_photo(image):
     if image is None: return None, "Upload a uPAD photo first."
     try:
         img = Image.fromarray(image) if not isinstance(image, Image.Image) else image
-        arr = np.array(img)
-        h,w = arr.shape[:2]
+        arr = np.array(img); h,w = arr.shape[:2]
         y1,y2,x1,x2 = int(h*0.35),int(h*0.65),int(w*0.35),int(w*0.65)
         zone = arr[y1:y2,x1:x2]
         R,G,B = float(np.mean(zone[:,:,0])),float(np.mean(zone[:,:,1])),float(np.mean(zone[:,:,2]))
-        c = max(0, round(0.018*(R-B)-0.3, 2))
+        c = max(0,round(0.018*(R-B)-0.3,2))
         if c<1.2: s,a="Normal","Monitor annually."
         elif c<1.5: s,a="Borderline","Repeat in 3 months."
         elif c<3.0: s,a="Stage 2 CKD","Consult nephrologist."
         elif c<6.0: s,a="Stage 3-4 CKD","Immediate consultation."
-        else: s,a="Stage 5 CKD","Emergency care needed."
-        ri = img.copy()
-        import PIL.ImageDraw as D
-        D.Draw(ri).rectangle([x1,y1,x2,y2], outline=(0,255,0), width=3)
-        return ri, ("uPAD ANALYSIS"+chr(10)+"━"*22+chr(10)+"R:"+str(round(R,1))+" G:"+str(round(G,1))+" B:"+str(round(B,1))+chr(10)+"Creatinine: "+str(c)+" mg/dL"+chr(10)+"Stage: "+s+chr(10)+"Action: "+a)
-    except Exception as e: return None, "Error: "+str(e)
-
-def mk_chart(fn, title, bg, fg, gc, ac, pb):
-    fig2,ax = plt.subplots(figsize=(8,5))
-    fig2.patch.set_facecolor(bg); ax.set_facecolor(pb)
-    fn(ax)
-    ax.set_title(title, color=fg, fontweight="bold", fontsize=13, pad=8)
-    ax.tick_params(colors=ac, labelsize=10)
-    ax.grid(True, alpha=0.3, color=gc, linestyle="--")
+        else: s,a="Stage 5 CKD","Emergency care."
+        ri=img.copy()
+        import PIL.ImageDraw as D; D.Draw(ri).rectangle([x1,y1,x2,y2],outline=(0,255,0),width=3)
+        return ri,("uPAD ANALYSIS"+chr(10)+"━"*22+chr(10)+"R:"+str(round(R,1))+" G:"+str(round(G,1))+" B:"+str(round(B,1))+chr(10)+"Creatinine: "+str(c)+" mg/dL"+chr(10)+"Stage: "+s+chr(10)+"Action: "+a)
+    except Exception as e: return None,"Error: "+str(e)
+
+def mk_chart(fn,title,bg,fg,gc,ac,pb):
+    fig2,ax=plt.subplots(figsize=(8,5)); fig2.patch.set_facecolor(bg); ax.set_facecolor(pb)
+    fn(ax); ax.set_title(title,color=fg,fontweight="bold",fontsize=13,pad=8)
+    ax.tick_params(colors=ac,labelsize=10); ax.grid(True,alpha=0.3,color=gc,linestyle="--")
     for sp in ["top","right"]: ax.spines[sp].set_visible(False)
     for sp in ["bottom","left"]: ax.spines[sp].set_color(gc)
-    plt.tight_layout()
-    buf=io.BytesIO(); plt.savefig(buf,format="png",facecolor=bg,bbox_inches="tight",dpi=130); buf.seek(0)
+    plt.tight_layout(); buf=io.BytesIO(); plt.savefig(buf,format="png",facecolor=bg,bbox_inches="tight",dpi=130); buf.seek(0)
     res=Image.open(buf).copy(); plt.close(); return res
 
-def analyze_piv_csv(file, theme="White"):
-    if file is None: return None,None,None,None,"Upload a PIV CSV file first."
+def analyze_piv_csv(file,theme="White"):
+    if file is None: return None,None,None,None,"Upload PIV CSV first."
     try:
-        df = pd.read_csv(file.name)
-        cols = [c.lower().strip() for c in df.columns]; df.columns = cols
-        num_cols = df.select_dtypes(include=[np.number]).columns.tolist()
-        if not num_cols: return None,None,None,None,"No numeric columns found."
+        df=pd.read_csv(file.name); cols=[c.lower().strip() for c in df.columns]; df.columns=cols
+        num_cols=df.select_dtypes(include=[np.number]).columns.tolist()
+        if not num_cols: return None,None,None,None,"No numeric columns."
         bg="#fff" if theme=="White" else "#0a1628"; fg="#1a202c" if theme=="White" else "white"
         gc="#e2e8f0" if theme=="White" else "#2d4a8a"; ac="#4a5568" if theme=="White" else "#a8b2d8"
         pb="#f7fafc" if theme=="White" else "#132340"
         x=np.arange(len(df))
         vc=next((c for c in cols if any(k in c for k in ["vel","speed","v_mag"])),num_cols[0] if num_cols else None)
         sc=next((c for c in cols if any(k in c for k in ["shear","stress","tau","wss"])),num_cols[1] if len(num_cols)>1 else None)
-        tc=next((c for c in cols if "time" in c or "frame" in c),None)
-        xv=df[tc] if tc else x
+        tc=next((c for c in cols if "time" in c or "frame" in c),None); xv=df[tc] if tc else x
         def pv(ax):
             if vc:
-                ax.plot(xv,df[vc],color="#e63946",linewidth=2.5,marker="o",markersize=5)
-                ax.fill_between(xv,df[vc],alpha=0.15,color="#e63946")
+                ax.plot(xv,df[vc],color="#c1121f",linewidth=2.5,marker="o",markersize=5)
+                ax.fill_between(xv,df[vc],alpha=0.15,color="#c1121f")
                 ax.axhline(y=2.0,color="#f59e0b",linestyle="--",linewidth=2,label="Risk: 2.0 m/s")
                 ax.set_ylabel("Velocity (m/s)",color=ac,fontsize=11); ax.set_xlabel(tc or "Sample",color=ac,fontsize=11)
                 ax.legend(fontsize=9,labelcolor=fg,facecolor=pb)
         def ps(ax):
             if sc:
                 xp=xv.values if tc else x
-                ax.plot(xp,df[sc],color="#4361ee",linewidth=2.5,marker="s",markersize=5)
-                ax.fill_between(xp,df[sc],alpha=0.15,color="#4361ee")
+                ax.plot(xp,df[sc],color="#0057a8",linewidth=2.5,marker="s",markersize=5)
+                ax.fill_between(xp,df[sc],alpha=0.15,color="#0057a8")
                 ax.axhline(y=5,color="#f59e0b",linestyle="--",linewidth=2,label="Caution: 5 Pa")
-                ax.axhline(y=10,color="#e63946",linestyle="--",linewidth=2,label="High risk: 10 Pa")
-                ax.set_ylabel("Shear Stress (Pa)",color=ac,fontsize=11); ax.set_xlabel(tc or "Sample",color=ac,fontsize=11)
+                ax.axhline(y=10,color="#c1121f",linestyle="--",linewidth=2,label="High risk: 10 Pa")
+                ax.set_ylabel("Shear (Pa)",color=ac,fontsize=11); ax.set_xlabel(tc or "Sample",color=ac,fontsize=11)
                 ax.legend(fontsize=9,labelcolor=fg,facecolor=pb)
         def psc(ax):
             if vc and sc:
                 s2=ax.scatter(df[vc],df[sc],c=x,cmap="RdYlGn_r",s=90,edgecolors=fg,linewidth=0.5,zorder=5)
                 cb=plt.colorbar(s2,ax=ax,label="Time"); cb.ax.yaxis.label.set_color(fg); cb.ax.tick_params(colors=ac)
-                ax.axvline(x=2.0,color="#f59e0b",linestyle="--",linewidth=2,label="Vel risk"); ax.axhline(y=10,color="#e63946",linestyle="--",linewidth=2,label="Shear risk")
+                ax.axvline(x=2.0,color="#f59e0b",linestyle="--",linewidth=2,label="Vel risk")
+                ax.axhline(y=10,color="#c1121f",linestyle="--",linewidth=2,label="Shear risk")
                 ax.set_xlabel("Velocity (m/s)",color=ac,fontsize=11); ax.set_ylabel("Shear (Pa)",color=ac,fontsize=11)
                 ax.legend(fontsize=9,labelcolor=fg,facecolor=pb)
         def psum(ax):
@@ -418,7 +349,7 @@ def analyze_piv_csv(file, theme="White"):
                 st+=col[:14]+":"+chr(10)+"  Mean: "+str(mn)+chr(10)+"  Max:  "+str(mx)+chr(10)+chr(10)
                 if "vel" in col and mx>2.0: risk.append("HIGH VELOCITY")
                 if "shear" in col and mx>10: risk.append("HIGH SHEAR")
-            bc="#e63946" if risk else "#2ecc71"
+            bc="#c1121f" if risk else "#2ecc71"
             st+="━"*20+chr(10)+("OVERALL: HIGH RISK" if risk else "OVERALL: LOW RISK")
             ax.text(0.05,0.97,st,transform=ax.transAxes,color=fg,fontsize=10,va="top",fontfamily="monospace",
                 bbox=dict(boxstyle="round,pad=0.8",facecolor=pb,edgecolor=bc,linewidth=2.5))
@@ -438,12 +369,11 @@ def analyze_piv_csv(file, theme="White"):
         return i1,i2,i3,i4,"PIV: "+str(len(df))+" rows | "+", ".join(df.columns.tolist())+ai
     except Exception as e: return None,None,None,None,"Error: "+str(e)
 
-def analyze_tgt_csv(file, theme="White"):
-    if file is None: return None,None,None,None,"Upload a TGT CSV file first."
+def analyze_tgt_csv(file,theme="White"):
+    if file is None: return None,None,None,None,"Upload TGT CSV first."
     try:
-        df = pd.read_csv(file.name)
-        cols = [c.lower().strip() for c in df.columns]; df.columns = cols
-        num_cols = df.select_dtypes(include=[np.number]).columns.tolist()
+        df=pd.read_csv(file.name); cols=[c.lower().strip() for c in df.columns]; df.columns=cols
+        num_cols=df.select_dtypes(include=[np.number]).columns.tolist()
         bg="#fff" if theme=="White" else "#0a1628"; fg="#1a202c" if theme=="White" else "white"
         gc="#e2e8f0" if theme=="White" else "#2d4a8a"; ac="#4a5568" if theme=="White" else "#a8b2d8"
         pb="#f7fafc" if theme=="White" else "#132340"
@@ -469,16 +399,16 @@ def analyze_tgt_csv(file, theme="White"):
                     mv=round(float(np.max(yp)),2); st="HIGH" if mv>lim else "NORMAL"
                     ax.set_title(title+chr(10)+"Max: "+str(mv)+"  Status: "+st,color=fg,fontweight="bold",fontsize=12)
             return mk_chart(fn,title,bg,fg,gc,ac,pb)
-        i1=mk2(tatc,"#e63946","TAT (ng/mL)",8,"Normal: 8","TAT Thrombin-Antithrombin")
-        i2=mk2(pfc,"#4361ee","PF1.2 (nmol/L)",2.0,"Normal: 2.0","PF1.2 Prothrombin Fragment")
+        i1=mk2(tatc,"#c1121f","TAT (ng/mL)",8,"Normal: 8","TAT Thrombin-Antithrombin")
+        i2=mk2(pfc,"#0057a8","PF1.2 (nmol/L)",2.0,"Normal: 2.0","PF1.2 Prothrombin Fragment")
         i3=mk2(hc,"#2ecc71","Free Hemoglobin (mg/L)",20,"Normal: 20","Free Hemoglobin",bar=True)
-        i4=mk2(plc,"#e67e22","Platelet Count",150,"Normal min: 150","Platelet Count")
+        i4=mk2(plc,"#e8a020","Platelet Count",150,"Normal min: 150","Platelet Count")
         ai=""
         if GROQ_KEY:
             try:
                 client=Groq(api_key=GROQ_KEY)
                 resp=client.chat.completions.create(model="llama-3.3-70b-versatile",
-                    messages=[{"role":"system","content":"Hematology expert SJSU CardioLab. Give thrombogenicity risk LOW MODERATE or HIGH. Normal: TAT<8, PF1.2<2.0, Hemo<20, Plt>150."},
+                    messages=[{"role":"system","content":"Hematology expert SJSU CardioLab. Give thrombogenicity risk LOW MODERATE or HIGH."},
                     {"role":"user","content":"TGT from 27mm SJM Regent:"+chr(10)+df.describe().to_string()[:500]}],max_tokens=250)
                 ai=chr(10)+"━"*20+chr(10)+"AI: "+resp.choices[0].message.content
             except: pass
@@ -520,59 +450,33 @@ def tgt_manual(t,p,h,pl,tm):
     risk=sum([float(t)>15,float(p)>2.0,float(h)>50,float(pl)<150])
     return "TAT:"+str(t)+" PF1.2:"+str(p)+chr(10)+"Hemo:"+str(h)+" Plt:"+str(pl)+chr(10)+"Time:"+str(tm)+" min"+chr(10)+"RESULT: "+("HIGH RISK" if risk>=3 else "MODERATE" if risk>=2 else "LOW RISK")
 
-with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
+with gr.Blocks(title="CardioLab AI — SJSU", css=CSS) as demo:
 
-    gr.HTML("""
-    <div style="background:linear-gradient(135deg,#1a237e 0%,#b71c1c 100%);padding:16px 24px;display:flex;align-items:center;gap:16px;">
-        <div style="font-size:1.8em;font-weight:900;color:#fff;letter-spacing:2px;">❤️ CardioLab AI</div>
-        <div style="color:rgba(255,255,255,0.7);font-size:0.85em;">SJSU Biomedical Engineering</div>
-    </div>
-    """)
+    gr.HTML(HEADER_HTML)
 
     with gr.Tabs():
 
         with gr.Tab("💬 Chat"):
             with gr.Row():
-
-                # LEFT SIDEBAR - ChatGPT style
-                with gr.Column(scale=1, min_width=220):
-                    gr.HTML('<div style="background:#202123;padding:12px;border-radius:8px;margin-bottom:8px;"><div style="color:white;font-weight:700;font-size:0.9em;margin-bottom:8px;">💬 Conversations</div></div>')
-                    new_chat_btn = gr.Button("✏️ New Chat", variant="secondary")
-                    gr.HTML('<div style="color:#9ca3af;font-size:0.75em;padding:8px 0 4px 0;">SAVED SESSIONS</div>')
-                    session_dropdown = gr.Dropdown(
-                        choices=get_session_list(),
-                        label="",
-                        interactive=True,
-                        container=False
-                    )
-                    load_btn = gr.Button("📂 Load", variant="primary")
-                    session_name_box = gr.Textbox(
-                        placeholder="Session name...",
-                        label="",
-                        lines=1,
-                        container=False
-                    )
+                with gr.Column(scale=1, min_width=210):
+                    gr.HTML('''<div style="background:#202123;padding:10px;border-radius:8px;margin-bottom:6px;">
+                        <div style="color:#e8a020;font-weight:700;font-size:0.85em;letter-spacing:1px;">⚔️ SJSU CARDIOLAB</div>
+                        <div style="color:#9ca3af;font-size:0.7em;margin-top:2px;">Conversations</div>
+                    </div>''')
+                    new_chat_btn = gr.Button("✏️  New Chat", variant="secondary")
+                    gr.HTML('''<div style="color:#9ca3af;font-size:0.72em;padding:8px 2px 4px 2px;letter-spacing:1px;">SAVED SESSIONS</div>''')
+                    session_dropdown = gr.Dropdown(choices=get_session_list(), label="", interactive=True, container=False)
+                    load_btn = gr.Button("📂  Load Session", variant="primary")
+                    session_name_box = gr.Textbox(placeholder="Name this session...", label="", lines=1, container=False)
                     with gr.Row():
-                        save_btn = gr.Button("💾 Save", variant="primary", scale=1)
-                        delete_btn = gr.Button("🗑️", variant="secondary", scale=0)
+                        save_btn = gr.Button("💾 Save", variant="primary", scale=2)
+                        delete_btn = gr.Button("🗑️", variant="secondary", scale=1)
                     session_status = gr.Textbox(label="", lines=1, interactive=False, container=False)
 
-                # RIGHT - Main chat area
                 with gr.Column(scale=4):
-                    chatbot = gr.Chatbot(
-                        label="",
-                        height=520,
-                        show_label=False,
-                        container=False
-                    )
+                    chatbot = gr.Chatbot(label="", height=500, show_label=False, container=False)
                     with gr.Row():
-                        msg_box = gr.Textbox(
-                            placeholder="Message CardioLab AI...",
-                            label="",
-                            lines=2,
-                            scale=5,
-                            container=False
-                        )
+                        msg_box = gr.Textbox(placeholder="Message CardioLab AI...", label="", lines=2, scale=5, container=False)
                         with gr.Column(scale=1, min_width=80):
                             send_btn = gr.Button("Send ↑", variant="primary")
                             clear_btn = gr.Button("Clear", variant="secondary")
@@ -586,13 +490,11 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
             delete_btn.click(delete_session, inputs=session_dropdown, outputs=[session_status, session_dropdown])
 
         with gr.Tab("🎙️ Voice"):
+            voice_chatbot = gr.Chatbot(label="", height=380, show_label=False)
+            audio_input = gr.Audio(sources=["microphone"], type="filepath", label="Record your question")
             with gr.Row():
-                with gr.Column():
-                    voice_chatbot = gr.Chatbot(label="", height=400, show_label=False)
-                    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)
 
@@ -600,7 +502,7 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
             with gr.Row():
                 search_input = gr.Textbox(placeholder="e.g. mechanical heart valve thrombogenicity 2024", label="Research Topic", scale=4)
                 search_btn = gr.Button("Search", variant="primary", scale=1)
-            search_output = gr.Textbox(label="Verified Results", lines=18)
+            search_output = gr.Textbox(label="Verified Results — PubMed + Semantic Scholar", lines=18)
             search_btn.click(quick_search, inputs=search_input, outputs=search_output)
             search_input.submit(quick_search, inputs=search_input, outputs=search_output)
 
@@ -612,11 +514,11 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
             piv_btn = gr.Button("Analyze PIV Data", variant="primary")
             piv_result = gr.Textbox(label="AI Analysis", lines=4)
             with gr.Row():
-                piv_c1 = gr.Image(label="Velocity Profile", type="pil")
-                piv_c2 = gr.Image(label="Shear Stress", type="pil")
+                piv_c1=gr.Image(label="Velocity Profile",type="pil")
+                piv_c2=gr.Image(label="Shear Stress",type="pil")
             with gr.Row():
-                piv_c3 = gr.Image(label="Velocity vs Shear", type="pil")
-                piv_c4 = gr.Image(label="Clinical Summary", type="pil")
+                piv_c3=gr.Image(label="Velocity vs Shear",type="pil")
+                piv_c4=gr.Image(label="Clinical Summary",type="pil")
             piv_btn.click(analyze_piv_csv, inputs=[piv_file,piv_theme], outputs=[piv_c1,piv_c2,piv_c3,piv_c4,piv_result])
 
         with gr.Tab("🩸 TGT CSV"):
@@ -627,23 +529,21 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
             tgt_btn = gr.Button("Analyze TGT Data", variant="primary")
             tgt_result = gr.Textbox(label="AI Assessment", lines=4)
             with gr.Row():
-                tgt_c1 = gr.Image(label="TAT", type="pil")
-                tgt_c2 = gr.Image(label="PF1.2", type="pil")
+                tgt_c1=gr.Image(label="TAT",type="pil"); tgt_c2=gr.Image(label="PF1.2",type="pil")
             with gr.Row():
-                tgt_c3 = gr.Image(label="Hemoglobin", type="pil")
-                tgt_c4 = gr.Image(label="Platelets", type="pil")
+                tgt_c3=gr.Image(label="Hemoglobin",type="pil"); tgt_c4=gr.Image(label="Platelets",type="pil")
             tgt_btn.click(analyze_tgt_csv, inputs=[tgt_file,tgt_theme], outputs=[tgt_c1,tgt_c2,tgt_c3,tgt_c4,tgt_result])
 
         with gr.Tab("🧪 uPAD"):
             with gr.Row():
                 with gr.Column():
-                    photo_input = gr.Image(label="Upload uPAD Photo", type="numpy", height=280)
+                    photo_input = gr.Image(label="Upload uPAD Photo", type="numpy", height=260)
                     analyze_btn = gr.Button("Analyze uPAD Photo", variant="primary")
                 with gr.Column():
-                    photo_img = gr.Image(label="Detection Zone (green box)", type="pil", height=280)
-                    photo_text = gr.Textbox(label="CKD Result", lines=10)
+                    photo_img = gr.Image(label="Detection Zone", type="pil", height=260)
+                    photo_text = gr.Textbox(label="CKD Result", lines=8)
             analyze_btn.click(analyze_upad_photo, inputs=photo_input, outputs=[photo_img, photo_text])
-            gr.Markdown("**Manual RGB entry:**")
+            gr.Markdown("**Manual RGB:**")
             with gr.Row():
                 r=gr.Number(label="R",value=210); g=gr.Number(label="G",value=140); b=gr.Number(label="B",value=80)
                 out3=gr.Textbox(label="Result",lines=3)
@@ -658,14 +558,14 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
                     img_btn = gr.Button("Generate Image", variant="primary")
                     img_status = gr.Textbox(label="Status", lines=1)
             img_desc = gr.Textbox(label="AI Description", lines=2, interactive=False)
-            img_output = gr.Image(label="Generated Image", type="pil", height=420)
+            img_output = gr.Image(label="Generated Image", type="pil", height=400)
             img_btn.click(generate_image, inputs=img_prompt, outputs=[img_output,img_status,img_desc])
 
         with gr.Tab("📐 PIV Manual"):
             with gr.Row():
                 with gr.Column():
                     v=gr.Number(label="Max Velocity m/s",value=1.8,info="Normal: 0.5-2.0")
-                    s=gr.Number(label="Wall Shear Stress Pa",value=6.5,info="Normal: <5 Pa")
+                    s=gr.Number(label="Wall Shear Stress Pa",value=6.5,info="Normal: <5")
                     h=gr.Number(label="Heart Rate bpm",value=72,info="Normal: 60-100")
                     piv_out=gr.Textbox(label="Result",lines=4)
                     gr.Button("Analyze PIV",variant="primary").click(piv_manual,inputs=[v,s,h],outputs=piv_out)
@@ -681,4 +581,15 @@ with gr.Blocks(title="CardioLab AI", css=CSS) as demo:
                     out2=gr.Textbox(label="Result",lines=6)
                     gr.Button("Analyze TGT",variant="primary").click(tgt_manual,inputs=[t1,t2,t3,t4,t5],outputs=out2)
 
+    gr.HTML("""
+    <div style="text-align:center;padding:12px;border-top:1px solid #e5e7eb;margin-top:8px;background:#f9fafb;">
+        <span style="color:#9ca3af;font-size:0.75em;">
+            ❤️ CardioLab AI &nbsp;|&nbsp; SJSU Biomedical Engineering &nbsp;|&nbsp;
+            Built on <a href="https://github.com/snap-stanford/Biomni" style="color:#c1121f;">Biomni Stanford</a> &nbsp;|&nbsp;
+            <a href="https://github.com/pranatechsol/Cardio-Lab-Ai" style="color:#0057a8;">GitHub</a> &nbsp;|&nbsp;
+            Apache 2.0 &nbsp;|&nbsp; $0 Cost
+        </span>
+    </div>
+    """)
+
 demo.launch()