Upload app.py with huggingface_hub

app.py

@@ -1,20 +1,17 @@
 import gradio as gr
 import os, requests, io
-import matplotlib
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
-import matplotlib.patches as mpatches
-import numpy as np
 from groq import Groq
 from PIL import Image
 
 GROQ_KEY = os.environ.get('GROQ_API_KEY', '')
+HF_TOKEN = os.environ.get('HF_TOKEN', '')
+
 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.'
 
 CSS = '''
-body, .gradio-container { background: #f0f4f8 !important; color: #1a202c !important; }
+body, .gradio-container { background: #f0f4f8 !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; }
+.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-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; }
@@ -96,86 +93,36 @@ def voice_chat(audio, history):
         history.append({'role':'assistant','content':'Voice error: '+str(e)})
         return history
 
-def generate_diagram(topic):
-    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,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 - 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)
-    buf.seek(0)
-    img = Image.open(buf)
-    plt.close()
-    return img
+def generate_image(prompt):
+    if not prompt.strip(): return None, 'Please enter a description.'
+    if not HF_TOKEN: return None, 'Error: Add HF_TOKEN to Space Settings Secrets.'
+    try:
+        # Add biomedical context to prompt
+        full_prompt = 'Highly detailed scientific biomedical illustration of: ' + prompt + ', professional medical diagram, high quality, detailed, photorealistic'
+        headers = {'Authorization': 'Bearer ' + HF_TOKEN}
+        payload = {'inputs': full_prompt}
+        # Use FLUX.1-schnell - best free model on HuggingFace
+        r = requests.post(
+            'https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell',
+            headers=headers, json=payload, timeout=60
+        )
+        if r.status_code == 200:
+            img = Image.open(io.BytesIO(r.content))
+            return img, 'Image generated successfully!'
+        elif r.status_code == 503:
+            # Model loading - try stable diffusion
+            r2 = requests.post(
+                'https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0',
+                headers=headers, json=payload, timeout=60
+            )
+            if r2.status_code == 200:
+                img = Image.open(io.BytesIO(r2.content))
+                return img, 'Image generated!'
+            return None, 'Model loading, please wait 30 seconds and try again.'
+        else:
+            return None, 'Error: '+str(r.status_code)+' '+r.text[:200]
+    except Exception as e:
+        return None, 'Error: '+str(e)
 
 def piv_tool(velocity, shear, hr):
     v = 'HIGH - stenosis risk' if float(velocity)>2.0 else 'NORMAL'
@@ -194,9 +141,9 @@ def upad_tool(r,g,b):
     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>')
+    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'):
+        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)
@@ -206,7 +153,7 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
             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'):
+        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')
@@ -215,22 +162,30 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
                 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 research papers - verified links only')
+        with gr.Tab('Papers'):
             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='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)
-        with gr.Tab('🎨 Diagrams'):
-            gr.Markdown('### Real visual diagrams - try: TGT setup | MCL PIV | uPAD CKD | MHV valve | CardioLab overview')
+        with gr.Tab('AI Image Generator'):
+            gr.Markdown('### Real AI Image Generation using FLUX.1 - Free HuggingFace Model')
+            gr.Markdown('**Describe any biomedical image and AI will generate it**')
             with gr.Row():
-                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)
-        with gr.Tab('📊 PIV'):
+                img_prompt = gr.Textbox(
+                    placeholder='e.g. mechanical heart valve bileaflet design | uPAD microfluidic device | blood flow through valve | Arduino circuit for TGT',
+                    label='Describe the image you want',
+                    lines=3,
+                    scale=4
+                )
+                with gr.Column(scale=1):
+                    img_btn = gr.Button('Generate Image', variant='primary')
+                    img_status = gr.Textbox(label='Status', lines=2)
+            img_output = gr.Image(label='Generated Image', type='pil', height=500)
+            img_btn.click(generate_image, inputs=img_prompt, outputs=[img_output, img_status])
+            gr.Markdown('**Try:** `27mm bileaflet mechanical heart valve` | `microfluidic paper device for CKD testing` | `blood flow visualization PIV` | `Arduino circuit with stepper motor`')
+        with gr.Tab('PIV'):
             with gr.Row():
                 with gr.Column():
                     v=gr.Number(label='Max Velocity m/s', value=1.8)
@@ -238,7 +193,7 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
                     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.Tab('TGT'):
             with gr.Row():
                 with gr.Column():
                     t1=gr.Number(label='TAT ng/mL', value=18)
@@ -248,7 +203,7 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
                     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.Tab('uPAD'):
             with gr.Row():
                 with gr.Column():
                     r=gr.Number(label='R value', value=210)