File size: 369 Bytes
0a3aede |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# mcp/nlp.py
import spacy
try:
nlp = spacy.load("en_core_sci_sm")
except Exception:
nlp = spacy.load("en_core_web_sm") # Fallback to general English
def extract_keywords(text: str):
"""Extract biomedical entities and drugs from text."""
doc = nlp(text)
keywords = list(set(ent.text for ent in doc.ents if len(ent.text) > 2))
return keywords
|