Update app.py

app.py

@@ -4,6 +4,7 @@ from refined.inference.processor import Refined
 import requests
 import json
 import spacy
+import spacy.cli
 
 # Page config
 st.set_page_config(
@@ -63,32 +64,42 @@ else:
 def load_model(selected_language, model_name=None, entity_set=None):
     try:
         if selected_language == "German":
-            # Load the German-specific model
-            nlp_model_de = spacy.load("de_core_news_lg")
+            # Download and load the German-specific model
+            try:
+                nlp_model_de = spacy.load("de_core_news_lg")
+            except OSError:
+                st.info("Downloading German language model... This may take a moment.")
+                spacy.cli.download("de_core_news_lg")
+                nlp_model_de = spacy.load("de_core_news_lg")
             
             # Check if entityfishing component is available
             if "entityfishing" not in nlp_model_de.pipe_names:
                 try:
                     nlp_model_de.add_pipe("entityfishing")
                 except Exception as e:
-                    st.error(f"Error adding entityfishing component: {e}")
-                    st.error("Please make sure entity-fishing is properly installed and configured.")
-                    return None
+                    st.warning(f"Entity-fishing not available, using basic NER only: {e}")
+                    # Return model without entityfishing for basic NER
+                    return nlp_model_de
             
             return nlp_model_de
             
         elif selected_language == "English - spaCy":
-            # Load English-specific model
-            nlp_model_en = spacy.load("en_core_web_sm")
+            # Download and load English-specific model
+            try:
+                nlp_model_en = spacy.load("en_core_web_sm")
+            except OSError:
+                st.info("Downloading English language model... This may take a moment.")
+                spacy.cli.download("en_core_web_sm")
+                nlp_model_en = spacy.load("en_core_web_sm")
             
             # Check if entityfishing component is available
             if "entityfishing" not in nlp_model_en.pipe_names:
                 try:
                     nlp_model_en.add_pipe("entityfishing")
                 except Exception as e:
-                    st.error(f"Error adding entityfishing component: {e}")
-                    st.error("Please make sure entity-fishing is properly installed and configured.")
-                    return None
+                    st.warning(f"Entity-fishing not available, using basic NER only: {e}")
+                    # Return model without entityfishing for basic NER
+                    return nlp_model_en
 
             return nlp_model_en    
         else:
@@ -247,5 +258,4 @@ if submit_button and entities_map:
     with st.expander("Here is the final JSON-LD"):
         st.json(json_ld_data)  # Output JSON-LD
 elif submit_button and not entities_map:
-    st.warning("No entities found in the text. Please try with different text or check if the model is working correctly.")
-    
+    st.warning("No entities found in the text. Please try with different text or check if the model is working correctly.")