Upload 6 files

app.py

@@ -158,18 +158,30 @@ def load_reranker():
 
 # Cache for spaCy models
 _spacy_models = {}
+_spacy_available = True
 
 def load_spacy_model(lang: str):
     """Carrega modelo spaCy com cache."""
+    global _spacy_available
+    
+    if not _spacy_available:
+        return None
+        
     if lang not in _spacy_models:
         model_name = "pt_core_news_sm" if lang == "pt" else "en_core_web_sm"
         try:
             _spacy_models[lang] = spacy.load(model_name)
             logging.info(f"Modelo spaCy '{model_name}' carregado.")
         except OSError:
-            logging.warning(f"Modelo {model_name} não encontrado. Baixando...")
-            spacy.cli.download(model_name.replace("_", "-").replace("-sm", ""))
-            _spacy_models[lang] = spacy.load(model_name)
+            logging.warning(f"Modelo {model_name} não encontrado. Tentando baixar...")
+            try:
+                import subprocess
+                subprocess.run(["python", "-m", "spacy", "download", model_name], check=True)
+                _spacy_models[lang] = spacy.load(model_name)
+            except Exception as e:
+                logging.error(f"Falha ao baixar modelo spaCy: {e}")
+                _spacy_available = False
+                return None
     return _spacy_models[lang]
 
 def detect_language(texts: List[str]) -> str:
@@ -186,6 +198,11 @@ def extract_entities(textos: List[str]) -> List[List[Tuple[str, str]]]:
     lang = detect_language(textos)
     nlp = load_spacy_model(lang)
     
+    # Fallback se spaCy não estiver disponível
+    if nlp is None:
+        logging.warning("spaCy não disponível. Retornando entidades vazias.")
+        return [[] for _ in textos]
+    
     entities_by_doc = []
     for text in textos:
         # Limitar tamanho do texto para performance

requirements.txt

@@ -18,6 +18,8 @@ hdbscan
 nltk
 spacy
 langdetect
+https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1-py3-none-any.whl
+https://github.com/explosion/spacy-models/releases/download/pt_core_news_sm-3.7.0/pt_core_news_sm-3.7.0-py3-none-any.whl
 
 # --- TORCH CPU ---
 torch