Update mcp/nlp.py

mcp/nlp.py

@@ -1,14 +1,21 @@
 # mcp/nlp.py
 
 import spacy
+import subprocess
 
-try:
-    nlp = spacy.load("en_core_sci_sm")
-except Exception:
-    nlp = spacy.load("en_core_web_sm")  # Fallback to general English
+def load_model():
+    try:
+        return spacy.load("en_core_sci_sm")
+    except OSError:
+        try:
+            subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"], check=True)
+            return spacy.load("en_core_web_sm")
+        except Exception as e:
+            raise RuntimeError("Failed to download spaCy model. Details: " + str(e))
+
+nlp = load_model()
 
 def extract_keywords(text: str):
-    """Extract biomedical entities and drugs from text."""
+    """Extract biomedical or general entities from text."""
     doc = nlp(text)
-    keywords = list(set(ent.text for ent in doc.ents if len(ent.text) > 2))
-    return keywords
+    return list(set(ent.text for ent in doc.ents if len(ent.text.strip()) > 2))