import gradio as gr
import camelot
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from fpdf import FPDF
from fpdf.enums import XPos, YPos
import tempfile
import os
import matplotlib
import shutil
import colorsys
from datetime import datetime
matplotlib.use('Agg')

# Configurações globais
ESCALA_MAXIMA_NOTAS = 12
LIMITE_APROVACAO_NOTA = 5
LIMITE_APROVACAO_FREQ = 75
BIMESTRES = ['1º Bimestre', '2º Bimestre', '3º Bimestre', '4º Bimestre']
CONCEITOS_VALIDOS = ['ES', 'EP', 'ET']

# Definição das disciplinas de formação básica
FORMACAO_BASICA = {
    'fundamental': {
        'LINGUA PORTUGUESA',
        'MATEMATICA',
        'HISTORIA',
        'GEOGRAFIA',
        'CIENCIAS',
        'LINGUA ESTRANGEIRA INGLES',
        'ARTE',
        'EDUCACAO FISICA'
    },
    'medio': {
        'LINGUA PORTUGUESA',
        'MATEMATICA',
        'HISTORIA',
        'GEOGRAFIA',
        'BIOLOGIA',
        'FISICA',
        'QUIMICA',
        'INGLÊS',
        'FILOSOFIA',
        'SOCIOLOGIA',
        'ARTE',
        'EDUCACAO FISICA'
    }
}

def detectar_nivel_ensino(disciplinas):
    """Detecta se é ensino fundamental ou médio baseado nas disciplinas presentes."""
    disciplinas_set = set(disciplinas)
    disciplinas_exclusivas_medio = {'BIOLOGIA', 'FISICA', 'QUIMICA', 'FILOSOFIA', 'SOCIOLOGIA'}
    return 'medio' if any(d in disciplinas_set for d in disciplinas_exclusivas_medio) else 'fundamental'

def separar_disciplinas_por_categoria(disciplinas_dados):
    """Separa as disciplinas em formação básica e diversificada."""
    disciplinas = [d['disciplina'] for d in disciplinas_dados]
    nivel = detectar_nivel_ensino(disciplinas)
    
    formacao_basica = []
    diversificada = []
    
    for disc_data in disciplinas_dados:
        if disc_data['disciplina'] in FORMACAO_BASICA[nivel]:
            formacao_basica.append(disc_data)
        else:
            diversificada.append(disc_data)
    
    return {
        'nivel': nivel,
        'formacao_basica': formacao_basica,
        'diversificada': diversificada
    }

def converter_nota(valor):
    """Converte valor de nota para float, tratando casos especiais e conceitos."""
    if pd.isna(valor) or valor == '-' or valor == 'N' or valor == '' or valor == 'None':
        return None
    
    if isinstance(valor, str):
        valor_limpo = valor.strip().upper()
        if valor_limpo in CONCEITOS_VALIDOS:
            conceitos_map = {'ET': 10, 'ES': 8, 'EP': 6}
            return conceitos_map.get(valor_limpo)
        
        try:
            return float(valor_limpo.replace(',', '.'))
        except:
            return None
    
    if isinstance(valor, (int, float)):
        return float(valor)
    
    return None

def calcular_media_bimestres(notas):
    """Calcula média considerando apenas bimestres com notas válidas."""
    notas_validas = [nota for nota in notas if nota is not None]
    if not notas_validas:
        return 0
    return sum(notas_validas) / len(notas_validas)

def calcular_frequencia_media(frequencias):
    """Calcula média de frequência considerando apenas bimestres cursados."""
    freq_validas = []
    for freq in frequencias:
        try:
            if isinstance(freq, str):
                freq = freq.strip().replace('%', '').replace(',', '.')
            if freq and freq != '-':
                valor = float(freq)
                if valor > 0:
                    freq_validas.append(valor)
        except:
            continue
    
    if not freq_validas:
        return 0
    return sum(freq_validas) / len(freq_validas)

def extrair_tabelas_pdf(pdf_path):
    """Extrai tabelas do PDF usando stream apenas para o nome e lattice para notas."""
    try:
        # Extrair nome do aluno usando stream
        tables_header = camelot.read_pdf(
            pdf_path,
            pages='1',
            flavor='stream',
            edge_tol=500
        )
        
        info_aluno = {}
        
        # Procurar apenas o nome do aluno
        for table in tables_header:
            df = table.df
            for i in range(len(df)):
                for j in range(len(df.columns)):
                    texto = str(df.iloc[i,j]).strip()
                    if 'Nome do Aluno' in texto:
                        try:
                            if j + 1 < len(df.columns):
                                nome = str(df.iloc[i,j+1]).strip()
                            elif i + 1 < len(df):
                                nome = str(df.iloc[i+1,j]).strip()
                            if nome and nome != 'Nome do Aluno:':
                                info_aluno['nome'] = nome
                                break
                        except:
                            continue
        
        # Extrair tabela de notas usando lattice
        tables_notas = camelot.read_pdf(
            pdf_path,
            pages='all',
            flavor='lattice'
        )
        
        # Encontrar tabela de notas (procurar a maior tabela com 'Disciplina')
        df_notas = None
        max_rows = 0
        
        for table in tables_notas:
            df_temp = table.df
            if len(df_temp) > max_rows and 'Disciplina' in str(df_temp.iloc[0,0]):
                max_rows = len(df_temp)
                df_notas = df_temp.copy()
                df_notas = df_notas.rename(columns={
                    0: 'Disciplina',
                    1: 'Nota B1', 2: 'Freq B1', 3: '%Freq B1', 4: 'AC B1',
                    5: 'Nota B2', 6: 'Freq B2', 7: '%Freq B2', 8: 'AC B2',
                    9: 'Nota B3', 10: 'Freq B3', 11: '%Freq B3', 12: 'AC B3',
                    13: 'Nota B4', 14: 'Freq B4', 15: '%Freq B4', 16: 'AC B4',
                    17: 'CF', 18: 'Nota Final', 19: 'Freq Final', 20: 'AC Final'
                })
        
        if df_notas is None:
            raise ValueError("Tabela de notas não encontrada")
        
        # Adicionar apenas o nome ao DataFrame
        df_notas.attrs['nome'] = info_aluno.get('nome', 'Nome não encontrado')
        
        return df_notas
        
    except Exception as e:
        print(f"Erro na extração das tabelas: {str(e)}")
        raise
        
def obter_disciplinas_validas(df):
    """Identifica disciplinas válidas no boletim com seus dados."""
    colunas_notas = ['Nota B1', 'Nota B2', 'Nota B3', 'Nota B4']
    colunas_freq = ['%Freq B1', '%Freq B2', '%Freq B3', '%Freq B4']
    
    disciplinas_dados = []
    
    for _, row in df.iterrows():
        disciplina = row['Disciplina']
        if pd.isna(disciplina) or disciplina == '':
            continue
        
        notas = []
        freqs = []
        bimestres_cursados = []
        
        for i, (col_nota, col_freq) in enumerate(zip(colunas_notas, colunas_freq), 1):
            nota = converter_nota(row[col_nota])
            freq = row[col_freq] if col_freq in row else None
            
            if nota is not None or (freq and freq != '-'):
                bimestres_cursados.append(i)
                notas.append(nota if nota is not None else 0)
                freqs.append(freq)
            else:
                notas.append(None)
                freqs.append(None)
        
        if bimestres_cursados:
            media_notas = calcular_media_bimestres(notas)
            media_freq = calcular_frequencia_media(freqs)
            
            disciplinas_dados.append({
                'disciplina': disciplina,
                'notas': notas,
                'frequencias': freqs,
                'media_notas': media_notas,
                'media_freq': media_freq,
                'bimestres_cursados': bimestres_cursados
            })
    
    return disciplinas_dados

def gerar_paleta_cores(n_cores):
    """Gera uma paleta de cores distintas para o número de disciplinas."""
    cores_base = [
        '#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd',
        '#8c564b', '#e377c2', '#7f7f7f', '#bcbd22', '#17becf',
        '#393b79', '#637939', '#8c6d31', '#843c39', '#7b4173'
    ]
    
    if n_cores > len(cores_base):
        HSV_tuples = [(x/n_cores, 0.7, 0.85) for x in range(n_cores)]
        cores_extras = ['#%02x%02x%02x' % tuple(int(x*255) for x in colorsys.hsv_to_rgb(*hsv)) 
                       for hsv in HSV_tuples]
        return cores_extras
    
    return cores_base[:n_cores]

def plotar_evolucao_bimestres(disciplinas_dados, temp_dir, titulo=None, nome_arquivo=None):
    """Plota gráfico de evolução das notas por bimestre com visualização refinada."""
    n_disciplinas = len(disciplinas_dados)
    
    if n_disciplinas == 0:
        raise ValueError("Nenhuma disciplina válida encontrada para plotar.")
    
    plt.figure(figsize=(11.69, 8.27))
    
    cores = gerar_paleta_cores(n_disciplinas)
    marcadores = ['o', 's', '^', 'D', 'v', '<', '>', 'p', 'h', '*']
    estilos_linha = ['-', '--', '-.', ':', '-', '--', '-.', ':', '-', '--']
    
    plt.grid(True, linestyle='--', alpha=0.3, zorder=0)
    
    # Deslocamento ainda menor e mais refinado
    deslocamentos = np.linspace(-0.03, 0.03, n_disciplinas)
    
    # Estrutura para armazenar as posições das anotações já utilizadas
    anotacoes_usadas = {}  # formato: {bimestre: [(y, texto)]}
    
    # Primeira passagem: coletar todos os valores e determinar grupos
    grupos_notas = {}  # {bimestre: {nota: [índices]}}
    for idx, disc_data in enumerate(disciplinas_dados):
        notas = pd.Series(disc_data['notas'])
        bimestres_cursados = disc_data['bimestres_cursados']
        
        if bimestres_cursados:
            notas_validas = [nota for i, nota in enumerate(notas, 1) if i in bimestres_cursados and nota is not None]
            bimestres = [bim for bim in bimestres_cursados if notas[bim-1] is not None]
            
            for bim, nota in zip(bimestres, notas_validas):
                if nota is not None:
                    if bim not in grupos_notas:
                        grupos_notas[bim] = {}
                    if nota not in grupos_notas[bim]:
                        grupos_notas[bim][nota] = []
                    grupos_notas[bim][nota].append(idx)

    # Segunda passagem: plotar e anotar
    for idx, disc_data in enumerate(disciplinas_dados):
        notas = pd.Series(disc_data['notas'])
        bimestres_cursados = disc_data['bimestres_cursados']
        desloc = deslocamentos[idx]
        
        if bimestres_cursados:
            notas_validas = [nota for i, nota in enumerate(notas, 1) if i in bimestres_cursados and nota is not None]
            bimestres = [bim for bim in bimestres_cursados if notas[bim-1] is not None]
            bimestres_deslocados = [bim + desloc for bim in bimestres]
            
            if notas_validas:
                # Plotar linha e pontos
                plt.plot(bimestres_deslocados, notas_validas,
                        color=cores[idx % len(cores)],
                        marker=marcadores[idx % len(marcadores)],
                        markersize=7,
                        linewidth=1.5,
                        label=disc_data['disciplina'],
                        linestyle=estilos_linha[idx % len(estilos_linha)],
                        alpha=0.8)
                
                # Adicionar anotações com posicionamento otimizado
                for bim_orig, bim_desloc, nota in zip(bimestres, bimestres_deslocados, notas_validas):
                    if nota is not None:
                        # Verificar se é o primeiro índice para esta nota neste bimestre
                        if grupos_notas[bim_orig][nota][0] == idx:
                            # Determinar posição vertical da anotação
                            if bim_orig not in anotacoes_usadas:
                                anotacoes_usadas[bim_orig] = []
                            
                            # Encontrar posição vertical disponível
                            y_base = nota
                            y_offset = 10
                            texto = f"{nota:.1f}"
                            
                            # Verificar sobreposição com anotações existentes
                            while any(abs(y - (y_base + y_offset/20)) < 0.4 for y, _ in anotacoes_usadas.get(bim_orig, [])):
                                y_offset += 5
                            
                            # Adicionar anotação
                            plt.annotate(texto,
                                       (bim_orig, nota),
                                       textcoords="offset points",
                                       xytext=(0, y_offset),
                                       ha='center',
                                       va='bottom',
                                       fontsize=8,
                                       bbox=dict(facecolor='white',
                                               edgecolor='none',
                                               alpha=0.8,
                                               pad=0.5))
                            
                            anotacoes_usadas[bim_orig].append((nota + y_offset/20, texto))
    
    # Usar título personalizado se fornecido
    titulo_grafico = titulo or 'Evolução das Médias por Disciplina ao Longo dos Bimestres'
    plt.title(titulo_grafico, pad=20, fontsize=12, fontweight='bold')
    
    plt.xlabel('Bimestres', fontsize=10)
    plt.ylabel('Notas', fontsize=10)
    plt.xticks([1, 2, 3, 4], ['1º Bim', '2º Bim', '3º Bim', '4º Bim'])
    plt.ylim(0, ESCALA_MAXIMA_NOTAS)
    
    # Adicionar linha de aprovação
    plt.axhline(y=LIMITE_APROVACAO_NOTA, color='r', linestyle='--', alpha=0.3)
    plt.text(0.02, LIMITE_APROVACAO_NOTA + 0.1, 'Média mínima para aprovação',
             transform=plt.gca().get_yaxis_transform(), color='r', alpha=0.5)
    
    # Ajustar legenda
    if n_disciplinas > 8:
        plt.legend(bbox_to_anchor=(1.05, 1), loc='upper left', fontsize=8,
                  ncol=max(1, n_disciplinas // 12))
    else:
        plt.legend(bbox_to_anchor=(1.05, 1), loc='upper left', ncol=1)
    
    plt.tight_layout()
    
    # Usar nome de arquivo personalizado se fornecido
    nome_arquivo = nome_arquivo or 'evolucao_notas.png'
    plot_path = os.path.join(temp_dir, nome_arquivo)
    plt.savefig(plot_path, bbox_inches='tight', dpi=300)
    plt.close()
    return plot_path

def plotar_graficos_destacados(disciplinas_dados, temp_dir):
    """Plota gráficos de médias e frequências com destaques."""
    n_disciplinas = len(disciplinas_dados)
    
    if not n_disciplinas:
        raise ValueError("Nenhuma disciplina válida encontrada no boletim.")
    
    # Criar figura
    plt.figure(figsize=(12, 10))
    
    disciplinas = [d['disciplina'] for d in disciplinas_dados]
    medias_notas = [d['media_notas'] for d in disciplinas_dados]
    medias_freq = [d['media_freq'] for d in disciplinas_dados]
    
    # Criar subplot com mais espaço entre os gráficos
    fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(12, 10), height_ratios=[1, 1])
    plt.subplots_adjust(hspace=0.5)  # Aumentar espaço entre os gráficos
    
    # Definir cores baseadas nos limites de aprovação
    cores_notas = ['red' if media < LIMITE_APROVACAO_NOTA else '#2ecc71' for media in medias_notas]
    cores_freq = ['red' if media < LIMITE_APROVACAO_FREQ else '#2ecc71' for media in medias_freq]
    
    # Calcular médias globais
    media_global = np.mean(medias_notas)
    freq_global = np.mean(medias_freq)
    
    # Gráfico de notas
    barras_notas = ax1.bar(disciplinas, medias_notas, color=cores_notas)
    ax1.set_title('Média de Notas por Disciplina', pad=20, fontsize=12, fontweight='bold')
    ax1.set_ylim(0, ESCALA_MAXIMA_NOTAS)
    ax1.grid(True, axis='y', alpha=0.3, linestyle='--')
    
    # Melhorar a apresentação dos rótulos
    ax1.set_xticklabels(disciplinas, rotation=45, ha='right', va='top')
    ax1.set_ylabel('Notas', fontsize=10, labelpad=10)
    
    # Adicionar linha de média mínima
    ax1.axhline(y=LIMITE_APROVACAO_NOTA, color='r', linestyle='--', alpha=0.3)
    ax1.text(0.02, LIMITE_APROVACAO_NOTA + 0.1, 'Média mínima (5,0)',
             transform=ax1.get_yaxis_transform(), color='r', alpha=0.7)
    
    # Valores nas barras de notas
    for barra in barras_notas:
        altura = barra.get_height()
        ax1.text(barra.get_x() + barra.get_width()/2., altura,
                f'{altura:.1f}',
                ha='center', va='bottom', fontsize=8)
    
    # Gráfico de frequências
    barras_freq = ax2.bar(disciplinas, medias_freq, color=cores_freq)
    ax2.set_title('Frequência Média por Disciplina', pad=20, fontsize=12, fontweight='bold')
    ax2.set_ylim(0, 110)
    ax2.grid(True, axis='y', alpha=0.3, linestyle='--')
    
    # Melhorar a apresentação dos rótulos
    ax2.set_xticklabels(disciplinas, rotation=45, ha='right', va='top')
    ax2.set_ylabel('Frequência (%)', fontsize=10, labelpad=10)
    
    # Adicionar linha de frequência mínima
    ax2.axhline(y=LIMITE_APROVACAO_FREQ, color='r', linestyle='--', alpha=0.3)
    ax2.text(0.02, LIMITE_APROVACAO_FREQ + 1, 'Frequência mínima (75%)',
             transform=ax2.get_yaxis_transform(), color='r', alpha=0.7)
    
    # Valores nas barras de frequência
    for barra in barras_freq:
        altura = barra.get_height()
        ax2.text(barra.get_x() + barra.get_width()/2., altura,
                f'{altura:.1f}%',
                ha='center', va='bottom', fontsize=8)
    
    # Título global com informações de média
    plt.suptitle(
        f'Desempenho Geral\nMédia Global: {media_global:.1f} | Frequência Global: {freq_global:.1f}%',
        y=0.98, fontsize=14, fontweight='bold'
    )
    
    # Aviso de risco de reprovação se necessário
    if freq_global < LIMITE_APROVACAO_FREQ:
        plt.figtext(0.5, 0.02,
                   "Atenção: Risco de Reprovação por Baixa Frequência",
                   ha="center", fontsize=11, color="red", weight='bold')
    
    plt.tight_layout()
    
    # Salvar o gráfico
    plot_path = os.path.join(temp_dir, 'medias_frequencias.png')
    plt.savefig(plot_path, bbox_inches='tight', dpi=300)
    plt.close()
    
    return plot_path

def gerar_relatorio_pdf(df, disciplinas_dados, grafico_basica, grafico_diversificada, grafico_medias):
    """Gera relatório PDF com os gráficos e análises."""
    pdf = FPDF()
    pdf.set_auto_page_break(auto=True, margin=15)
    
    # Primeira página - Informações e Formação Básica
    pdf.add_page()
    pdf.set_font('Helvetica', 'B', 18)
    pdf.cell(0, 10, 'Relatório de Desempenho Escolar', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='C')
    pdf.ln(15)
    
    # Informações do aluno
    pdf.set_font('Helvetica', 'B', 12)
    pdf.cell(0, 10, 'Informações do Aluno', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(5)
    
    # Mostrar apenas o nome
    if hasattr(df, 'attrs') and 'nome' in df.attrs:
        pdf.set_font('Helvetica', 'B', 11)
        pdf.cell(30, 7, 'Nome:', 0, 0)
        pdf.set_font('Helvetica', '', 11)
        pdf.cell(0, 7, df.attrs['nome'], 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT)
    
    pdf.ln(10)
    
    # Data do relatório
    data_atual = datetime.now().strftime('%d/%m/%Y')
    pdf.set_font('Helvetica', 'I', 10)
    pdf.cell(0, 5, f'Data de geração: {data_atual}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='R')
    pdf.ln(15)
    
    # Gráfico de evolução da formação básica
    pdf.set_font('Helvetica', 'B', 14)
    pdf.cell(0, 10, 'Evolução das Notas - Formação Geral Básica', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(10)
    pdf.image(grafico_basica, x=10, w=190)
    
    # Segunda página - Parte Diversificada
    pdf.add_page()
    pdf.set_font('Helvetica', 'B', 14)
    pdf.cell(0, 10, 'Evolução das Notas - Parte Diversificada', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(10)
    pdf.image(grafico_diversificada, x=10, w=190)
    
    # Terceira página - Médias e Frequências
    pdf.add_page()
    pdf.set_font('Helvetica', 'B', 14)
    pdf.cell(0, 10, 'Análise de Médias e Frequências', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(10)
    pdf.image(grafico_medias, x=10, w=190)
    
    # Quarta página - Análise Detalhada
    pdf.add_page()
    pdf.set_font('Helvetica', 'B', 14)
    pdf.cell(0, 10, 'Análise Detalhada', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(10)
    
    # Calcular médias globais
    medias_notas = [d['media_notas'] for d in disciplinas_dados]
    medias_freq = [d['media_freq'] for d in disciplinas_dados]
    media_global = np.mean(medias_notas)
    freq_global = np.mean(medias_freq)
    
    # Resumo geral
    pdf.set_font('Helvetica', 'B', 12)
    pdf.cell(0, 7, 'Resumo Geral:', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.ln(5)
    
    pdf.set_font('Helvetica', '', 11)
    pdf.cell(0, 7, f'Média Global: {media_global:.1f}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.cell(0, 7, f'Frequência Global: {freq_global:.1f}%', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.ln(10)
    
    # Avisos Importantes
    pdf.set_font('Helvetica', 'B', 12)
    pdf.cell(0, 10, 'Pontos de Atenção:', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    pdf.ln(5)
    
    pdf.set_font('Helvetica', '', 10)
    
    # Disciplinas com baixo desempenho
    disciplinas_risco = []
    for disc_data in disciplinas_dados:
        avisos = []
        if disc_data['media_notas'] < LIMITE_APROVACAO_NOTA:
            avisos.append(f"Média de notas abaixo de {LIMITE_APROVACAO_NOTA} ({disc_data['media_notas']:.1f})")
        if disc_data['media_freq'] < LIMITE_APROVACAO_FREQ:
            avisos.append(f"Frequência abaixo de {LIMITE_APROVACAO_FREQ}% ({disc_data['media_freq']:.1f}%)")
        
        if avisos:
            disciplinas_risco.append((disc_data['disciplina'], avisos))
    
    if disciplinas_risco:
        for disc, avisos in disciplinas_risco:
            pdf.set_font('Helvetica', 'B', 10)
            pdf.cell(0, 7, f'- {disc}:', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
            pdf.set_font('Helvetica', '', 10)
            for aviso in avisos:
                pdf.cell(10)  # Indentação
                pdf.cell(0, 7, f'- {aviso}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    else:
        pdf.cell(0, 7, 'Nenhum problema identificado.', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
    
    # Rodapé
    pdf.set_y(-30)
    pdf.line(10, pdf.get_y(), 200, pdf.get_y())
    pdf.ln(5)
    pdf.set_font('Helvetica', 'I', 8)
    pdf.cell(0, 10, 'Este relatório é uma análise automática e deve ser validado junto à secretaria da escola.',
             0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='C')
    
    # Salvar PDF
    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
    pdf_path = temp_pdf.name
    pdf.output(pdf_path)
    return pdf_path

def processar_boletim(file):
    """Função principal que processa o boletim e gera o relatório."""
    temp_dir = None
    try:
        if file is None:
            return None, "Nenhum arquivo foi fornecido."
        
        temp_dir = tempfile.mkdtemp()
        print(f"Diretório temporário criado: {temp_dir}")
        
        # Salvar o arquivo binário como um arquivo PDF temporário
        temp_pdf = os.path.join(temp_dir, 'boletim.pdf')
        with open(temp_pdf, 'wb') as f:
            f.write(file)  # Salva os bytes do arquivo no disco
        print(f"PDF salvo temporariamente em: {temp_pdf}")
        
        if os.path.getsize(temp_pdf) == 0:
            return None, "O arquivo está vazio."
        
        print("Iniciando extração das tabelas...")
        df = extrair_tabelas_pdf(temp_pdf)
        print("Tabelas extraídas com sucesso")
        
        if df is None or df.empty:
            return None, "Não foi possível extrair dados do PDF."
        
        try:
            # Processar disciplinas
            disciplinas_dados = obter_disciplinas_validas(df)
            if not disciplinas_dados:
                return None, "Nenhuma disciplina válida encontrada no boletim."
            
            # Separar disciplinas por categoria
            categorias = separar_disciplinas_por_categoria(disciplinas_dados)
            nivel = categorias['nivel']
            nivel_texto = "Ensino Médio" if nivel == "medio" else "Ensino Fundamental"
            
            # Gerar gráficos
            print("Gerando gráficos...")
            grafico_basica = plotar_evolucao_bimestres(
                categorias['formacao_basica'],
                temp_dir,
                titulo=f"Evolução das Médias - Formação Geral Básica ({nivel_texto})",
                nome_arquivo='evolucao_basica.png'
            )
            
            grafico_diversificada = plotar_evolucao_bimestres(
                categorias['diversificada'],
                temp_dir,
                titulo=f"Evolução das Médias - Parte Diversificada ({nivel_texto})",
                nome_arquivo='evolucao_diversificada.png'
            )
            
            grafico_medias = plotar_graficos_destacados(disciplinas_dados, temp_dir)
            print("Gráficos gerados")
            
            # Gerar PDF
            print("Gerando relatório PDF...")
            pdf_path = gerar_relatorio_pdf(df, disciplinas_dados, grafico_basica, grafico_diversificada, grafico_medias)
            print("Relatório PDF gerado")
            
            # Criar arquivo de retorno
            output_file = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
            output_path = output_file.name
            shutil.copy2(pdf_path, output_path)
                
            return output_path, "Relatório gerado com sucesso!"
            
        except Exception as e:
            return None, f"Erro ao processar os dados: {str(e)}"
        
    except Exception as e:
        print(f"Erro durante o processamento: {str(e)}")
        return None, f"Erro ao processar o boletim: {str(e)}"
        
    finally:
        if temp_dir and os.path.exists(temp_dir):
            try:
                shutil.rmtree(temp_dir)
                print("Arquivos temporários limpos")
            except Exception as e:
                print(f"Erro ao limpar arquivos temporários: {str(e)}")

# Interface Gradio
iface = gr.Interface(
    fn=processar_boletim,
    inputs=gr.File(
        label="Upload do Boletim (PDF)",
        type="binary",
        file_types=[".pdf"]
    ),
    outputs=[
        gr.File(label="Relatório (PDF)"),
        gr.Textbox(label="Status")
    ],
    title="Análise de Boletim Escolar",
    description="Faça upload do boletim em PDF para gerar um relatório com análises e visualizações.",
    allow_flagging="never"
)

if __name__ == "__main__":
    iface.launch(
        server_name="0.0.0.0",
        share=True
    )