Update README.md

README.md

@@ -4,26 +4,62 @@ emoji: 🏥
 colorFrom: purple
 colorTo: pink
 sdk: gradio
-sdk_version: 5.44.0
+sdk_version: 5.49.1
 app_file: app.py
 pinned: false
 ---
 
 # 🏥 AI Diagnostics Agent: Early Cancer Discovery (Demo)
 
-**⚠️ Research demo only — not for clinical use**
+[![Made with Python](https://img.shields.io/badge/Made%20with-Python-3776AB?logo=python&logoColor=white)](https://www.python.org/)
+[![Powered by Moonshot.ai](https://img.shields.io/badge/Powered%20by-Moonshot.ai-blue)](https://moonshot.ai)
+[![Gradio App](https://img.shields.io/badge/Gradio-5.44.0-orange)](https://gradio.app/)
+
+⚠️ **Research Demo Only — Not for Clinical Use**
 
 ---
 
-### 🚀 Overview
-AI Diagnostics Agent is a **proof-of-concept early cancer screening tool** combining:
-- **Imaging Agent** — analyzes chest X-rays using TorchXRayVision to detect lung anomalies (mass, nodule, opacity) and generate heatmaps.
-- **Lab Agent** — parses tumor marker results (PSA, CA125, AFP) for abnormal readings.
-- **AI Coordinator (Moonshot.ai)** — uses a large language model to summarize overall cancer risk, highlight abnormalities, and suggest next steps in human-readable form.
+## 🌍 Overview
+
+**AI Diagnostics (POC)** is a research project designed to demonstrate early cancer risk screening using **AI-driven imaging + lab report synthesis**.
+
+It integrates:
+- 🖼️ **Imaging Agent** — Detects lung abnormalities (mass, opacity, nodules) from chest X-rays.  
+- 🧪 **Lab Agent** — Parses tumor markers (PSA, CA-125, AFP) and highlights abnormal results.  
+- 🧠 **AI Coordinator** — Generates diagnostic summaries powered by **Moonshot.ai** with a local **T5 fallback** for offline use.  
+- 🔥 **Heatmap Overlay** — Highlights model-identified lung regions for interpretability.  
+
+---
+
+## ⚙️ Features
+
+| Agent | Description |
+|--------|--------------|
+| **Imaging Agent** | TorchXRayVision + PyTorch for lung-related cancer indicators |
+| **Lab Agent** | Extracts structured tumor marker insights |
+| **Coordinator** | Combines both into concise AI summaries (Moonshot.ai LLM) |
+| **Fallback AI** | Local text generation using T5-small if network/API fails |
+| **Heatmap Overlay** | Visual interpretability of X-ray activations |
+| **Gradio UI** | Easy drag-and-drop web interface for demos |
 
 ---
 
-### 🧠 Powered by Moonshot.ai
-The AI Coordinator uses [Moonshot.ai](https://www.moonshot.cn) LLM API to generate intelligent summaries.
+## 🧠 How It Works
+
+1. Upload or select a sample chest X-ray.  
+2. Paste lab or MRI/CT report text.  
+3. The AI analyzes the image + text.  
+4. Results are combined into a natural-language diagnostic summary.
+
+---
+
+## 🚀 Run Locally
+
+```bash
+git clone https://github.com/<your-org>/ai_diagnostics.git
+cd ai_diagnostics
+pip install -r requirements.txt
+python app.py
+