# 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