Update routers/summarizer.py

routers/summarizer.py

@@ -1,13 +1,14 @@
-import nltk
+
+from fastapi import APIRouter, Query, HTTPException
 from textblob import TextBlob
+from langdetect import detect
+import nltk
 
 # Baixa os corpora necessários para o TextBlob
 nltk.download('punkt')
 nltk.download('averaged_perceptron_tagger')
 
-# Agora a API pode usar o TextBlob normalmente
-from fastapi import APIRouter, Query, HTTPException
-
+# Cria um roteador para a API de correção gramatical
 router = APIRouter()
 
 @router.get("/grammar/correct/")
@@ -16,19 +17,48 @@ def correct_grammar(
 ):
     """
     Corrige erros gramaticais e ortográficos no texto e retorna sugestões.
-    Suporta correção em inglês.
+    Suporta correção em inglês e detecta o idioma do texto.
     """
     try:
+        # Detecta o idioma do texto
+        detected_language = detect(text)
+
+        # Se o texto não estiver em inglês, retorna uma mensagem indicando que não há suporte para correção
+        if detected_language != 'en':
+            return {
+                "original_text": text,
+                "corrected_text": text,
+                "detected_language": detected_language,
+                "message": "Grammatical correction is only supported for English text."
+            }
+
         # Cria um objeto TextBlob para o texto fornecido
         blob = TextBlob(text)
 
         # Corrige o texto automaticamente
         corrected_text = blob.correct()
 
-        # Retorna o texto corrigido
+        # Conta o número de palavras antes e depois da correção
+        original_word_count = len(text.split())
+        corrected_word_count = len(str(corrected_text).split())
+
+        # Adiciona detalhes sobre as correções, como palavras corrigidas
+        corrections = []
+        for word, corrected_word in zip(text.split(), str(corrected_text).split()):
+            if word != corrected_word:
+                corrections.append({
+                    "original_word": word,
+                    "corrected_word": corrected_word
+                })
+
+        # Retorna o texto corrigido, idioma detectado e detalhes adicionais
         return {
             "original_text": text,
-            "corrected_text": str(corrected_text)
+            "corrected_text": str(corrected_text),
+            "detected_language": detected_language,
+            "original_word_count": original_word_count,
+            "corrected_word_count": corrected_word_count,
+            "corrections": corrections
         }
     
     except Exception as e:
@@ -36,4 +66,4 @@ def correct_grammar(
         raise HTTPException(
             status_code=500,
             detail=f"An error occurred during grammar correction: {str(e)}"
-        )
+        )