Create summarizer.py

routers/summarizer.py

@@ -0,0 +1,37 @@
+from fastapi import APIRouter, Query, HTTPException
+from transformers import BartForConditionalGeneration, BartTokenizer
+
+# Cria um roteador para a API de resumo de texto
+router = APIRouter()
+
+# Carregar o modelo e o tokenizer do BART
+model_name = "facebook/bart-large-cnn"
+model = BartForConditionalGeneration.from_pretrained(model_name)
+tokenizer = BartTokenizer.from_pretrained(model_name)
+
+@router.get("/summarize/")
+def summarize_text(text: str = Query(..., description="Texto a ser resumido")):
+    """
+    Gera um resumo do texto fornecido utilizando o modelo BART.
+    """
+    try:
+        # Tokenizar o texto e gerar um resumo
+        inputs = tokenizer([text], max_length=1024, return_tensors="pt", truncation=True)
+        summary_ids = model.generate(inputs["input_ids"], max_length=150, min_length=30, length_penalty=2.0, num_beams=4, early_stopping=True)
+
+        # Decodificar o resumo gerado
+        summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+
+        return {
+            "original_text": text,
+            "summary": summary,
+            "original_length": len(text.split()),
+            "summary_length": len(summary.split())
+        }
+
+    except Exception as e:
+        # Em caso de erro, retorna uma mensagem de erro
+        raise HTTPException(
+            status_code=500,
+            detail=f"An error occurred during text summarization: {str(e)}"
+        )