Update app.py

app.py

@@ -23,8 +23,27 @@ warnings.filterwarnings('ignore', category=FutureWarning, module='huggingface_hu
 nltk.download('vader_lexicon', quiet=True)
 nltk.download('punkt', quiet=True)
 
-# Load spaCy model
-nlp = spacy.load("en_core_web_sm")
+ef download_spacy_model():
+    subprocess.check_call([sys.executable, "-m", "spacy", "download", "en_core_web_sm"])
+
+try:
+    import spacy
+    try:
+        nlp = spacy.load("en_core_web_sm")
+    except OSError:
+        print("Downloading spaCy model...")
+        download_spacy_model()
+        nlp = spacy.load("en_core_web_sm")
+except ImportError:
+    print("Error: Unable to import spaCy. NLP features will be disabled.")
+    nlp = None
+
+def extract_entities(text):
+    if nlp is None:
+        return []
+    doc = nlp(text)
+    entities = [(ent.text, ent.label_) for ent in doc.ents]
+    return entities
 
 # Initialize Example Dataset (For Emotion Prediction)
 data = {