Fix Gradio 5.31 ChatInterface API (remove deprecated btn args)

app.py

@@ -1,64 +1,43 @@
 """
 DeepMed-R1: Medical Reasoning AI Demo
-=======================================
-Demonstrates systematic clinical reasoning with structured chain-of-thought.
-Uses Qwen3 backbone with medical reasoning prompting (GRPO-trained reasoning patterns).
-
-Built for AMD Developer Hackathon 2026 — Track 2: Fine-Tuning on AMD GPUs
 """
-
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Use Qwen3 via HF Inference API — has built-in RL-trained reasoning
 client = InferenceClient("Qwen/Qwen3-4B")
 
-SYSTEM_PROMPT = """You are DeepMed-R1, a medical reasoning AI trained with GRPO (Group Relative Policy Optimization) and multi-objective clinical rewards.
+SYSTEM_PROMPT = """You are DeepMed-R1, a medical reasoning AI trained with GRPO and multi-objective clinical rewards.
 
 For every medical question, demonstrate systematic clinical reasoning:
+1. Information Analysis: Extract key demographics, symptoms, vitals, labs
+2. Differential Diagnosis: Identify patterns, rank by probability, note red flags
+3. Pathophysiology: Connect symptoms to disease mechanisms
+4. Evidence-Based Reasoning: Apply clinical criteria, reference guidelines
+5. Logical Elimination: Evaluate each option, exclude based on evidence
+6. Clinical Decision: Consider risk-benefit, prioritize safety
 
-1. **Information Analysis**: Extract key demographics, symptoms, vitals, labs, imaging findings
-2. **Differential Diagnosis**: Identify patterns, rank diagnoses by probability, note red flags
-3. **Pathophysiology**: Connect symptoms to underlying disease mechanisms
-4. **Evidence-Based Reasoning**: Apply clinical criteria, reference current guidelines
-5. **Logical Elimination**: Systematically evaluate each option, exclude based on evidence
-6. **Clinical Decision**: Consider risk-benefit, prioritize patient safety
-
-Present your reasoning inside <think></think> tags, then provide your final answer.
-For multiple-choice questions, end with \\boxed{X} where X is the correct letter.
-
-Be thorough but concise. Always ground your reasoning in pathophysiology and evidence."""
+Present reasoning inside <think></think> tags, then provide your final answer.
+For MCQ, end with \\boxed{X} where X is the correct letter."""
 
 EXAMPLES = [
-    ["A 65-year-old male with a history of hypertension and smoking presents with sudden onset severe headache described as 'the worst headache of my life,' neck stiffness, and photophobia. BP is 180/100. What is the most likely diagnosis?\nA. Migraine with aura\nB. Subarachnoid hemorrhage\nC. Bacterial meningitis\nD. Tension headache"],
-    ["A 28-year-old woman presents with fatigue, weight gain, cold intolerance, constipation, and dry skin for 3 months. Labs: TSH 12 mIU/L (normal: 0.4-4.0), Free T4 0.5 ng/dL (normal: 0.8-1.8). What is the most appropriate initial treatment?\nA. Levothyroxine\nB. Liothyronine\nC. Methimazole\nD. Radioactive iodine ablation"],
-    ["A 3-month-old infant presents with projectile vomiting after feeds for the past 2 weeks. The vomitus is non-bilious. On examination, an olive-shaped mass is palpable in the right upper quadrant. Metabolic alkalosis is present. What is the most likely diagnosis?\nA. Pyloric stenosis\nB. Intussusception\nC. Malrotation with volvulus\nD. Hirschsprung disease"],
-    ["A 55-year-old diabetic woman presents with right upper quadrant pain, fever (39.2°C), and jaundice (Charcot's triad). Labs show WBC 18,000, total bilirubin 5.2, ALP 450. Ultrasound shows dilated common bile duct with a 1.2cm stone. What is the most appropriate next step?\nA. Cholecystectomy\nB. ERCP with sphincterotomy\nC. MRCP\nD. Percutaneous transhepatic cholangiography"],
-    ["A 22-year-old male presents after a motorcycle accident with left-sided chest pain and shortness of breath. On examination: absent breath sounds on the left, tracheal deviation to the right, distended neck veins, BP 80/50. What is the immediate management?\nA. Chest X-ray\nB. CT chest\nC. Needle decompression of left chest\nD. Intubation"],
+    ["A 65-year-old male with hypertension presents with sudden 'worst headache of my life,' neck stiffness, photophobia. BP 180/100. Most likely diagnosis?\nA. Migraine\nB. Subarachnoid hemorrhage\nC. Meningitis\nD. Tension headache"],
+    ["A 28-year-old woman: fatigue, weight gain, cold intolerance. TSH 12 mIU/L, Free T4 0.5 ng/dL. Initial treatment?\nA. Levothyroxine\nB. Liothyronine\nC. Methimazole\nD. Radioactive iodine"],
+    ["3-month-old with projectile non-bilious vomiting, olive-shaped RUQ mass, metabolic alkalosis. Diagnosis?\nA. Pyloric stenosis\nB. Intussusception\nC. Malrotation\nD. Hirschsprung disease"],
+    ["55-year-old diabetic: RUQ pain, fever 39.2°C, jaundice (Charcot's triad). WBC 18K, bilirubin 5.2, CBD stone on US. Next step?\nA. Cholecystectomy\nB. ERCP with sphincterotomy\nC. MRCP\nD. PTC"],
+    ["22-year-old post-MVC: left chest pain, absent breath sounds left, trachea deviated right, JVD, BP 80/50. Immediate management?\nA. Chest X-ray\nB. CT chest\nC. Needle decompression left chest\nD. Intubation"],
 ]
 
 
 def respond(message, history):
-    """Generate medical reasoning response with streaming."""
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-
     for h in history:
-        if h[0]:
-            messages.append({"role": "user", "content": h[0]})
-        if h[1]:
-            messages.append({"role": "assistant", "content": h[1]})
-
+        if h[0]: messages.append({"role": "user", "content": h[0]})
+        if h[1]: messages.append({"role": "assistant", "content": h[1]})
     messages.append({"role": "user", "content": message})
 
     response = ""
     try:
-        for token in client.chat_completion(
-            messages=messages,
-            max_tokens=3000,
-            temperature=0.3,
-            top_p=0.95,
-            stream=True,
-        ):
+        for token in client.chat_completion(messages=messages, max_tokens=3000, temperature=0.3, top_p=0.95, stream=True):
             delta = token.choices[0].delta.content or ""
             response += delta
             yield response
@@ -66,25 +45,15 @@ def respond(message, history):
         yield f"Error: {str(e)}\n\nPlease try again."
 
 
-# ─── Gradio App ───────────────────────────────────────────────────────────────
-
-with gr.Blocks(
-    title="DeepMed-R1: Medical Reasoning AI",
-    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="cyan"),
-) as demo:
+with gr.Blocks(title="DeepMed-R1", theme=gr.themes.Soft(primary_hue="blue")) as demo:
     gr.Markdown("""
 # 🏥 DeepMed-R1: Medical Reasoning AI
 
-**A medical reasoning system trained with GRPO + Multi-Objective Clinical Rewards on AMD MI300X**
+**Systematic clinical reasoning powered by GRPO + Multi-Objective Clinical Rewards**
 
-DeepMed-R1 demonstrates systematic clinical reasoning through structured chain-of-thought:
-- 📋 **Information Analysis** → Extract key clinical features
-- 🔬 **Pathophysiology** → Connect symptoms to disease mechanisms  
-- 📊 **Evidence-Based Reasoning** → Apply clinical guidelines
-- ⚕️ **Logical Elimination** → Systematically evaluate options
-
-> ⚠️ **Disclaimer**: This is a research demo. Not for clinical use. Always consult healthcare professionals.
+Built for **AMD Developer Hackathon 2026** — Track 2: Fine-Tuning on AMD GPUs
 
+> ⚠️ Research demo only. Not for clinical use.
 ---
     """)
 
@@ -92,36 +61,25 @@ DeepMed-R1 demonstrates systematic clinical reasoning through structured chain-o
         fn=respond,
         examples=EXAMPLES,
         title="",
-        retry_btn="🔄 Retry",
-        undo_btn="↩️ Undo",
-        clear_btn="🗑️ Clear",
     )
 
     gr.Markdown("""
 ---
-### 🔬 Technical Architecture
-
-| Component | Implementation | Source |
-|-----------|---------------|--------|
-| **Base Model** | Qwen3 (RL-trained reasoning) | Qwen Team, 2025 |
-| **Training Algorithm** | GRPO with DAPO loss | arxiv:2503.14476 |
-| **Reward System** | CRPO (4-component clinical) | arxiv:2512.00601 |
-| **Innovation 1** | iGRPO self-feedback refinement | arxiv:2602.09000 |
-| **Innovation 2** | AERO adaptive rollouts | arxiv:2602.14338 |
-| **Innovation 3** | Curriculum reward scheduling | arxiv:2603.28120 |
-| **Hardware** | AMD MI300X (192GB HBM3) | AMD Developer Cloud |
-
-### 🏆 AMD Developer Hackathon 2026 — Track 2: Fine-Tuning on AMD GPUs
-
-**Novel Contribution**: First system combining iGRPO (Feb 2026) + CRPO multi-objective rewards (Dec 2025) + 
-curriculum scheduling (Mar 2026) for medical reasoning — trained on AMD MI300X leveraging 192GB HBM3 
-for full-parameter GRPO impossible on NVIDIA A100/H100.
-
-📄 [Training Code](https://huggingface.co/asusf15/DeepMed-R1) | 
-📚 [Paper: Gazal-R1](https://arxiv.org/abs/2506.21594) | 
-📚 [Paper: Clinical-R1](https://arxiv.org/abs/2512.00601)
+### 🔬 Architecture
+
+| Component | Detail |
+|-----------|--------|
+| Base Model | Qwen3 (RL-trained reasoning) |
+| Training | GRPO + DAPO loss + CRPO rewards |
+| Innovations | iGRPO (Feb 2026) + AERO (Feb 2026) + Curriculum (Mar 2026) |
+| Hardware | AMD MI300X (192GB HBM3) |
+| Reward System | Accuracy (w=2.0) + Clinical Reasoning (w=1.0) + Consistency (w=0.5) + Length (w=0.3) |
+
+📄 [Code & Training Pipeline](https://huggingface.co/asusf15/DeepMed-R1) |
+📚 [Gazal-R1 Paper](https://arxiv.org/abs/2506.21594) |
+📚 [Clinical-R1](https://arxiv.org/abs/2512.00601) |
+📚 [iGRPO](https://arxiv.org/abs/2602.09000)
     """)
 
-
 if __name__ == "__main__":
     demo.launch()