Update routers/textclas.py

routers/textclas.py

@@ -1,36 +1,44 @@
 from fastapi import APIRouter, Query, HTTPException
-from langdetect import detect, DetectorFactory
-from langdetect.lang_detect_exception import LangDetectException
+from keybert import KeyBERT
 
-# Garantir consistência nos resultados
-DetectorFactory.seed = 42
+# Inicializa o modelo KeyBERT
+kw_model = KeyBERT()
 
 router = APIRouter()
 
-@router.get("/language/is_ptbr/")
-def is_portuguese_brazilian(
-    text: str = Query(..., description="Texto para verificação de idioma")
+@router.get("/keywords/extract/")
+def extract_keywords(
+    text: str = Query(..., description="Texto para extração de palavras-chave"),
+    num_keywords: int = Query(5, description="Número de palavras-chave a serem retornadas", ge=1, le=20)
 ):
     """
-    Verifica se o texto está em Português do Brasil (pt-BR).
-    Retorna true se for pt-BR, false caso contrário.
+    Extrai palavras-chave relevantes de um texto.
     """
     try:
-        # Detecta o idioma do texto
-        detected_language = detect(text)
+        # Extrai palavras-chave
+        keywords = kw_model.extract_keywords(
+            text,
+            keyphrase_ngram_range=(1, 2),
+            stop_words='english',
+            top_n=num_keywords
+        )
         
-        # Verifica se o idioma detectado é pt (português)
-        is_ptbr = detected_language == "pt"
+        # Formata o retorno
+        keyword_list = [kw[0] for kw in keywords]
         
-        return {"is_ptbr": is_ptbr}
-
-    except LangDetectException:
+        return {
+            "text": text,
+            "num_keywords": num_keywords,
+            "keywords": keyword_list
+        }
+    
+    except ValueError as ve:
         raise HTTPException(
             status_code=400,
-            detail="Unable to detect the language. Please provide more text."
+            detail=f"Invalid input: {str(ve)}"
         )
     except Exception as e:
         raise HTTPException(
             status_code=500,
-            detail=f"An error occurred during language detection: {str(e)}"
+            detail=f"An error occurred during keyword extraction: {str(e)}"
         )