Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,76 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
pipeline_tag: text-generation
|
| 4 |
+
base_model: mistralai/Mistral-7B-v0.1
|
| 5 |
+
tags:
|
| 6 |
+
- medical-ai
|
| 7 |
+
- llm
|
| 8 |
+
- clinical-decision-support
|
| 9 |
+
- langgraph
|
| 10 |
+
- pubmed
|
| 11 |
+
- groq
|
| 12 |
+
- llama
|
| 13 |
+
- agent
|
| 14 |
---
|
| 15 |
+
# 🏥 Clinical Decision Support Agent
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
|
| 21 |
+
## 🎯 Live Demo
|
| 22 |
+
👉 [Try it on Hugging Face Spaces](https://huggingface.co/spaces/mou11/clinical-decision-support-agent)
|
| 23 |
+
|
| 24 |
+
## 🔍 Overview
|
| 25 |
+
A production-grade Medical AI Agent that takes patient symptoms as natural language input, searches real PubMed medical literature automatically, and generates structured clinical recommendations with citations. Built with LangGraph for agent orchestration and Groq (LLaMA 3.3 70B) for clinical reasoning.
|
| 26 |
+
|
| 27 |
+
## 📊 Results
|
| 28 |
+
| Metric | Score |
|
| 29 |
+
|--------|-------|
|
| 30 |
+
| BERTScore Precision | 0.8449 |
|
| 31 |
+
| BERTScore Recall | 0.8887 |
|
| 32 |
+
| BERTScore F1 | 0.8663 |
|
| 33 |
+
|
| 34 |
+
## 🧪 Test Cases
|
| 35 |
+
| Case | Symptoms | Urgency Level |
|
| 36 |
+
|------|----------|--------------|
|
| 37 |
+
| Case 1 | Fever, cough, chest pain | High ✅ |
|
| 38 |
+
| Case 2 | Stiff neck, fever, confusion | Emergency ✅ |
|
| 39 |
+
| Case 3 | Chest pain radiating to left arm, diabetes | Emergency ✅ |
|
| 40 |
+
|
| 41 |
+
## ✨ Features
|
| 42 |
+
- ✅ Real-time PubMed literature search via NCBI API
|
| 43 |
+
- ✅ LLM-powered clinical reasoning (LLaMA 3.3 70B via Groq)
|
| 44 |
+
- ✅ Structured output: conditions, tests, treatments, urgency
|
| 45 |
+
- ✅ Multi-turn conversation memory
|
| 46 |
+
- ✅ BERTScore quality evaluation
|
| 47 |
+
- ✅ Interactive Gradio web interface
|
| 48 |
+
|
| 49 |
+
## 🏗️ Architecture
|
| 50 |
+
Patient Symptoms → Extract Search Query (LLM) → Search PubMed (NCBI API) → Generate Recommendation (LLaMA 3.3 70B) → Format Report → Gradio UI
|
| 51 |
+
|
| 52 |
+
## 🛠️ Tech Stack
|
| 53 |
+
| Component | Tool |
|
| 54 |
+
|-----------|------|
|
| 55 |
+
| Agent Orchestration | LangGraph |
|
| 56 |
+
| LLM | Groq — LLaMA 3.3 70B |
|
| 57 |
+
| Medical Literature | PubMed NCBI API |
|
| 58 |
+
| Evaluation | BERTScore |
|
| 59 |
+
| UI | Gradio |
|
| 60 |
+
| Platform | Google Colab (CPU) |
|
| 61 |
+
|
| 62 |
+
## 🚀 How to Run
|
| 63 |
+
1. Get a free Groq API key at [console.groq.com](https://console.groq.com)
|
| 64 |
+
2. Open `app.py` in Google Colab
|
| 65 |
+
3. Replace `your-groq-api-key-here` with your actual key
|
| 66 |
+
4. Run all cells in order
|
| 67 |
+
|
| 68 |
+
## ⚠️ Medical Disclaimer
|
| 69 |
+
This system is for educational and research purposes only. It does not provide medical advice. Always consult a qualified healthcare professional for medical decisions.
|
| 70 |
+
|
| 71 |
+
## 📌 Project Status
|
| 72 |
+
✅ PubMed search tool complete
|
| 73 |
+
✅ LangGraph agent built
|
| 74 |
+
✅ Multi-turn memory working
|
| 75 |
+
✅ BERTScore evaluation done
|
| 76 |
+
✅ Gradio demo deployed on Hugging Face Spaces
|