Upload app.py with huggingface_hub

app.py

@@ -14,21 +14,15 @@ KNOWHOW = ('SJSU CardioLab: '
 
 CSS = (
     'body, .gradio-container { background: #0a0f1e !important; }'
-    '.tabs { background: #0d1b3e !important; border-bottom: 2px solid #e63946 !important; }'
-    '.tab-nav button { color: #a8b2d8 !important; background: transparent !important; font-size: 1em !important; font-weight: 600 !important; padding: 10px 20px !important; border: none !important; border-bottom: 3px solid transparent !important; }'
-    '.tab-nav button.selected { color: #e63946 !important; border-bottom: 3px solid #e63946 !important; background: transparent !important; }'
+    '.tab-nav button { color: #a8b2d8 !important; background: transparent !important; font-weight: 600 !important; padding: 10px 20px !important; border: none !important; border-bottom: 3px solid transparent !important; }'
+    '.tab-nav button.selected { color: #e63946 !important; border-bottom: 3px solid #e63946 !important; }'
     '.tab-nav button:hover { color: #ffffff !important; background: #1a2744 !important; }'
-    'h1, h2, h3, h4, label, p, span { color: #e2e8f0 !important; }'
-    'h1 { color: #e63946 !important; font-size: 2.8em !important; font-weight: 900 !important; text-align: center !important; padding: 25px 0 !important; letter-spacing: 2px !important; }'
+    'h1 { color: #e63946 !important; font-size: 2.8em !important; font-weight: 900 !important; text-align: center !important; padding: 25px 0 !important; }'
+    'h3, label, p, span { color: #e2e8f0 !important; }'
     'textarea, input[type=number], input[type=text] { background: #1a2744 !important; color: #e2e8f0 !important; border: 1px solid #4361ee !important; border-radius: 8px !important; }'
-    'textarea::placeholder { color: #6b7db3 !important; }'
-    '.gr-button { border-radius: 8px !important; font-weight: 700 !important; padding: 10px 20px !important; }'
-    'button.primary { background: linear-gradient(135deg, #e63946 0%, #c1121f 100%) !important; color: white !important; border: none !important; }'
-    'button.secondary { background: #1d3461 !important; color: #a8b2d8 !important; border: 1px solid #4361ee !important; }'
-    '.message.bot { background: #1a2744 !important; border: 1px solid #2d4a8a !important; color: #e2e8f0 !important; border-radius: 12px !important; }'
-    '.message.user { background: linear-gradient(135deg, #e63946, #c1121f) !important; color: white !important; border-radius: 12px !important; }'
+    'button.primary { background: linear-gradient(135deg, #e63946 0%, #c1121f 100%) !important; color: white !important; border: none !important; border-radius: 8px !important; }'
+    'button.secondary { background: #1d3461 !important; color: #a8b2d8 !important; border: 1px solid #4361ee !important; border-radius: 8px !important; }'
     '.block { background: #0d1b3e !important; border: 1px solid #1e3a6e !important; border-radius: 12px !important; }'
-    '.chatbot { background: #0d1b3e !important; border: 1px solid #2d4a8a !important; border-radius: 12px !important; }'
 )
 
 def get_pubmed(query, n=5):
@@ -91,6 +85,44 @@ def research_chat(message, history):
         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.'})
+        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
+        system_msg = 'You are CardioLab AI assistant. Expert in MHV MCL PIV TGT uPAD CKD FSI. ' + KNOWHOW
+        msgs = [{'role':'system','content':system_msg}]
+        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})
+        return history
+    except Exception as e:
+        history.append({'role':'assistant','content':'Voice error: '+str(e)})
+        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 diagram expert for SJSU CardioLab. Generate a detailed ASCII diagram or technical schematic based on the topic. 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)
+
 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' if float(shear)>5 else 'NORMAL'
@@ -120,8 +152,8 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
     gr.Markdown('# CardioLab AI')
 
     with gr.Tabs():
-        with gr.Tab('💬 Chat'):
-            chatbot = gr.Chatbot(label='', height=500)
+        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)
                 with gr.Column(scale=1):
@@ -131,14 +163,32 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
             msg_box.submit(research_chat, inputs=[msg_box, chatbot], outputs=[msg_box, chatbot])
             clear_btn.click(lambda: ([], ''), outputs=[chatbot, msg_box])
 
-        with gr.Tab('🔍 Paper Search'):
+        with gr.Tab('Voice'):
+            gr.Markdown('### Speak your question - powered by Groq Whisper')
+            voice_chatbot = gr.Chatbot(label='Voice Chat', height=350)
+            audio_input = gr.Audio(sources=['microphone'], type='filepath', label='Record your question')
+            with gr.Row():
+                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('Paper Search'):
             search_input = gr.Textbox(placeholder='e.g. mechanical heart valve thrombogenicity', label='Research topic')
             search_btn = gr.Button('Search Papers', variant='primary')
             search_output = gr.Textbox(label='Results', lines=20)
             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('📊 PIV Analysis'):
+        with gr.Tab('Diagram Generator'):
+            gr.Markdown('### Generate technical diagrams and schematics for CardioLab')
+            gr.Markdown('Examples: MCL setup | TGT circuit | uPAD fabrication flow | PIV system | valve design')
+            diagram_input = gr.Textbox(placeholder='e.g. Mock Circulatory Loop setup with PIV system', label='What to diagram')
+            diagram_btn = gr.Button('Generate Diagram', variant='primary')
+            diagram_output = gr.Textbox(label='Diagram', lines=20)
+            diagram_btn.click(generate_diagram, inputs=diagram_input, outputs=diagram_output)
+
+        with gr.Tab('PIV Analysis'):
             with gr.Row():
                 with gr.Column():
                     v=gr.Number(label='Max Velocity m/s', value=1.8)
@@ -147,7 +197,7 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
                     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 Results'):
+        with gr.Tab('TGT Results'):
             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)
@@ -156,7 +206,7 @@ with gr.Blocks(title='CardioLab AI', css=CSS) as demo:
             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 CKD'):
+        with gr.Tab('uPAD CKD'):
             r=gr.Number(label='R value', value=210)
             g=gr.Number(label='G value', value=140)
             b=gr.Number(label='B value', value=80)