routers/summarizer.py

from fastapi import APIRouter, HTTPException
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()

# 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:
        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)
        
        # Gera o PDF
        generate_pdf(file_path)
        
        # 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(
            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"Erro inesperado: {str(e)}")
    
    finally:
        # Remove o arquivo temporário após o envio
        if os.path.exists(file_path):
            os.remove(file_path)