Update routers/summarizer.py

routers/summarizer.py

@@ -1,36 +1,59 @@
 from fastapi import APIRouter, HTTPException
-from fpdf import FPDF
 from fastapi.responses import FileResponse
 import os
+from datetime import datetime
+from fpdf import FPDF  # Certifique-se de que essa biblioteca está instalada
 
 router = APIRouter()
 
-# Rota para gerar um PDF a partir de texto
-@router.post("/pdf/generate/")
-async def generate_pdf(content: str):
-    """
-    Gera um PDF a partir do texto fornecido.
-    """
+# Caminho absoluto para o diretório de arquivos temporários
+TEMP_DIR = os.path.abspath("temp_files")
+os.makedirs(TEMP_DIR, exist_ok=True)  # Garante que o diretório exista
+
+
+# Função para gerar um PDF
+def generate_pdf(file_path: str):
     try:
-        # Cria um novo PDF
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", size=12)
+        pdf.cell(200, 10, txt="Exemplo de PDF Gerado com FastAPI", ln=True, align='C')
+        pdf.cell(200, 10, txt=f"Data e Hora: {datetime.now()}", ln=True, align='C')
+        
+        pdf.output(file_path)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erro ao gerar PDF: {str(e)}")
+
+
+# Rota para baixar o PDF
+@router.get("/download-pdf")
+async def download_pdf():
+    try:
+        # Caminho absoluto do arquivo PDF
+        file_name = "generated_pdf.pdf"
+        file_path = os.path.join(TEMP_DIR, file_name)
         
-        # Adiciona conteúdo ao PDF
-        pdf.multi_cell(0, 10, content)
+        # Gera o PDF
+        generate_pdf(file_path)
         
-        # Salva o PDF temporariamente
-        pdf_file = "generated_pdf.pdf"
-        pdf.output(pdf_file)
+        # Verifica se o arquivo existe
+        if not os.path.exists(file_path):
+            raise HTTPException(status_code=500, detail="O arquivo PDF não foi gerado corretamente.")
         
+        # Retorna o arquivo PDF
         return FileResponse(
-            pdf_file,
-            headers={"Content-Disposition": f"attachment; filename={pdf_file}"}
+            path=file_path,
+            filename="download.pdf",
+            media_type="application/pdf"
         )
+    
+    except HTTPException as http_err:
+        raise http_err
+    
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error generating PDF: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Erro inesperado: {str(e)}")
+    
     finally:
-        # Remove o arquivo temporário após envio
-        if os.path.exists(pdf_file):
-            os.remove(pdf_file)
+        # Remove o arquivo temporário após o envio
+        if os.path.exists(file_path):
+            os.remove(file_path)