|
|
|
|
|
|
|
|
import pandas as pd |
|
|
import re |
|
|
import os |
|
|
import requests |
|
|
import matplotlib.font_manager as fm |
|
|
from pydantic import BaseModel, Field, ConfigDict |
|
|
from typing import List, Tuple, Optional |
|
|
from fpdf import FPDF |
|
|
from datetime import datetime |
|
|
|
|
|
|
|
|
class Historico(BaseModel): |
|
|
entradas: List[Tuple[str, str]] = [] |
|
|
|
|
|
def adicionar(self, pergunta: str, resposta: str): |
|
|
if pergunta and resposta: |
|
|
self.entradas.append((pergunta, resposta)) |
|
|
|
|
|
|
|
|
def descricao_colunas(df: pd.DataFrame) -> str: |
|
|
descricao = "\n".join([f"`{col}`: {str(df[col].dtype)}" for col in df.columns]) |
|
|
return "Colunas do DataFrame:\n" + descricao |
|
|
|
|
|
def limpar_codigo_pandas(codigo: str) -> str: |
|
|
codigo = re.sub(r'```(?:python)?\n?', '', codigo) |
|
|
codigo = re.sub(r'```', '', codigo) |
|
|
linhas = [linha.strip() for linha in codigo.split('\n') if linha.strip()] |
|
|
codigo_filtrado = [l for l in linhas if not (l.startswith('#') or 'resposta:' in l.lower())] |
|
|
return codigo_filtrado[-1] if codigo_filtrado else "" |
|
|
|
|
|
|
|
|
FONT_PATH = "" |
|
|
def setup_font(): |
|
|
global FONT_PATH |
|
|
FONT_NAME_TO_FIND = "DejaVu Sans" |
|
|
try: |
|
|
FONT_PATH = fm.findfont(FONT_NAME_TO_FIND, fallback_to_default=True) |
|
|
if not os.path.exists(FONT_PATH): raise FileNotFoundError |
|
|
print(f"✅ Fonte encontrada localmente: {FONT_PATH}") |
|
|
except Exception: |
|
|
print(f"⚠️ Fonte '{FONT_NAME_TO_FIND}' não encontrada. Baixando alternativa (Roboto)...") |
|
|
FONT_FILE_NAME = "Roboto-Regular.ttf" |
|
|
FONT_URL = "https://github.com/google/fonts/raw/main/ofl/roboto/Roboto-Regular.ttf" |
|
|
if not os.path.exists(FONT_FILE_NAME): |
|
|
try: |
|
|
response = requests.get(FONT_URL) |
|
|
response.raise_for_status() |
|
|
with open(FONT_FILE_NAME, "wb") as f: f.write(response.content) |
|
|
FONT_PATH = FONT_FILE_NAME |
|
|
print(f"✅ Fonte alternativa baixada!") |
|
|
except Exception as e: |
|
|
print(f"❌ FALHA CRÍTICA ao baixar fonte: {e}") |
|
|
FONT_PATH = "" |
|
|
else: |
|
|
FONT_PATH = FONT_FILE_NAME |
|
|
print(f"✅ Fonte alternativa já existe.") |
|
|
return FONT_PATH |
|
|
|
|
|
|
|
|
def gerar_pdf_report(historico: Historico, font_path: str): |
|
|
if not historico or not historico.entradas: |
|
|
print("⚠️ Tentativa de gerar PDF com histórico vazio.") |
|
|
return None |
|
|
if not font_path or not os.path.exists(font_path): |
|
|
print(f"❌ Erro ao gerar PDF: Arquivo da fonte não encontrado em '{font_path}'.") |
|
|
return None |
|
|
|
|
|
try: |
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") |
|
|
caminho_pdf = f"relatorio_consultas_{timestamp}.pdf" |
|
|
pdf = FPDF() |
|
|
pdf.add_page() |
|
|
pdf.add_font('CustomFont', '', font_path, uni=True) |
|
|
pdf.set_font('CustomFont', '', 16) |
|
|
pdf.cell(0, 10, "Relatório de Consultas ao CSV", ln=True, align="C") |
|
|
pdf.ln(10) |
|
|
|
|
|
for i, (pergunta, resposta) in enumerate(historico.entradas, 1): |
|
|
pdf.set_font('CustomFont', '', 12) |
|
|
pdf.multi_cell(0, 8, f"Pergunta {i}: {pergunta}") |
|
|
pdf.ln(2) |
|
|
pdf.set_font('CustomFont', '', 10) |
|
|
pdf.multi_cell(0, 6, f"Resposta: {resposta}") |
|
|
pdf.ln(8) |
|
|
pdf.cell(0, 0, '', 'T'); pdf.ln(8) |
|
|
|
|
|
pdf.output(caminho_pdf) |
|
|
print(f"✅ PDF gerado: {caminho_pdf}") |
|
|
return caminho_pdf |
|
|
except Exception as e: |
|
|
print(f"❌ Erro crítico ao gerar PDF: {e}") |
|
|
return None |
