Update app.py

app.py

@@ -6,6 +6,8 @@ import numpy as np
 from fpdf import FPDF
 import tempfile
 import os
+import matplotlib
+matplotlib.use('Agg')  # Usar backend não-interativo
 
 def converter_nota(valor):
     if pd.isna(valor) or valor == '-' or valor == 'N':
@@ -69,13 +71,14 @@ def plotar_evolucao_bimestres(df_filtrado, temp_dir):
     plt.legend(bbox_to_anchor=(1.05, 1), loc='upper left')
     plt.tight_layout()
     
-    # Salvar o gráfico
     plot_path = os.path.join(temp_dir, 'evolucao_notas.png')
     plt.savefig(plot_path, bbox_inches='tight', dpi=300)
     plt.close()
     return plot_path
 
 def plotar_graficos_destacados(df_boletim_clean, temp_dir):
+    plt.figure(figsize=(12, 6))
+    
     disciplinas = df_boletim_clean['Disciplina'].astype(str)
     
     medias_frequencia = df_boletim_clean[['Freq B1', 'Freq B2', 'Freq B3', 'Freq B4']].apply(pd.to_numeric, errors='coerce').mean(axis=1)
@@ -86,8 +89,6 @@ def plotar_graficos_destacados(df_boletim_clean, temp_dir):
     
     frequencia_global_media = medias_frequencia.mean()
     
-    plt.figure(figsize=(12, 6))
-    
     plt.subplot(1, 2, 1)
     plt.bar(disciplinas, medias_notas, color=cores_notas)
     plt.title('Média de Notas por Disciplina (Vermelho: < 5)')
@@ -107,7 +108,6 @@ def plotar_graficos_destacados(df_boletim_clean, temp_dir):
     
     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()
@@ -117,18 +117,15 @@ def gerar_relatorio_pdf(df, grafico1_path, grafico2_path):
     pdf = FPDF()
     pdf.add_page()
     
-    # Título
     pdf.set_font('Arial', 'B', 16)
     pdf.cell(0, 10, 'Relatório de Desempenho Escolar', 0, 1, 'C')
     pdf.ln(10)
     
-    # Adicionar gráficos
     pdf.image(grafico1_path, x=10, w=190)
     pdf.ln(10)
     pdf.image(grafico2_path, x=10, w=190)
     pdf.ln(10)
     
-    # Adicionar avisos
     pdf.set_font('Arial', 'B', 12)
     pdf.cell(0, 10, 'Avisos Importantes:', 0, 1, 'L')
     pdf.set_font('Arial', '', 10)
@@ -142,7 +139,6 @@ def gerar_relatorio_pdf(df, grafico1_path, grafico2_path):
         if media_freq < 75:
             pdf.cell(0, 10, f'- {disciplina}: Frequência abaixo de 75% ({media_freq:.1f}%)', 0, 1, 'L')
     
-    # Salvar PDF em um arquivo temporário
     temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
     pdf_path = temp_pdf.name
     pdf.output(pdf_path)
@@ -150,54 +146,53 @@ def gerar_relatorio_pdf(df, grafico1_path, grafico2_path):
 
 def processar_boletim(pdf_file):
     try:
-        # Criar diretório temporário
         temp_dir = tempfile.mkdtemp()
         
-        # Salvar o arquivo PDF enviado
+        # Ler o arquivo temporário e salvar seu conteúdo
         temp_pdf = os.path.join(temp_dir, 'boletim.pdf')
         with open(temp_pdf, 'wb') as f:
-            f.write(pdf_file)
+            f.write(pdf_file.read())  # Usar .read() para obter o conteúdo do arquivo
         
-        # Extrair tabelas do PDF
         tables = camelot.read_pdf(temp_pdf, pages='all', flavor='lattice')
         
         if len(tables) == 0:
             return None, "Nenhuma tabela encontrada no PDF."
         
-        # Processar primeira tabela
         df = tables[0].df
         
-        # Renomear colunas
         df.columns = ['Disciplina', 'Nota B1', 'Freq B1', '%Freq B1', 'AC B1',
                      'Nota B2', 'Freq B2', '%Freq B2', 'AC B2',
                      'Nota B3', 'Freq B3', '%Freq B3', 'AC B3',
                      'Nota B4', 'Freq B4', '%Freq B4', 'AC B4',
                      'CF', 'Nota Final', 'Freq Final', 'AC Final']
         
-        # Converter notas
         colunas_notas = ['Nota B1', 'Nota B2', 'Nota B3', 'Nota B4']
         for col in colunas_notas:
             df[col] = df[col].apply(converter_nota)
         
-        # Gerar gráficos
         grafico1_path = plotar_evolucao_bimestres(df, temp_dir)
         grafico2_path = plotar_graficos_destacados(df, temp_dir)
         
-        # Gerar PDF com o relatório
         pdf_path = gerar_relatorio_pdf(df, grafico1_path, grafico2_path)
         
-        return pdf_path, "Relatório gerado com sucesso!"
+        with open(pdf_path, 'rb') as f:
+            pdf_content = f.read()
+            
+        # Limpar arquivos temporários
+        os.remove(pdf_path)
+        for file in os.listdir(temp_dir):
+            os.remove(os.path.join(temp_dir, file))
+        os.rmdir(temp_dir)
+        
+        return pdf_content, "Relatório gerado com sucesso!"
         
     except Exception as e:
-        return None, f"Erro ao processar o boletim: {str(e)}"
-    finally:
-        # Limpar arquivos temporários
         if 'temp_dir' in locals():
             for file in os.listdir(temp_dir):
                 os.remove(os.path.join(temp_dir, file))
             os.rmdir(temp_dir)
+        return None, f"Erro ao processar o boletim: {str(e)}"
 
-# Interface Gradio
 iface = gr.Interface(
     fn=processar_boletim,
     inputs=gr.File(label="Upload do Boletim (PDF)"),