Update app.py

app.py

@@ -682,27 +682,108 @@ def processar_boletim(file):
             )
             
             grafico_medias = plotar_graficos_destacados(disciplinas_dados, temp_dir)
-            print("Gráficos gerados com sucesso")
+def processar_boletim(file):
+    """Função principal que processa o boletim e gera o relatório."""
+    temp_dir = None
+    try:
+        print("Iniciando processamento do arquivo...")
+        
+        if file is None:
+            return None, "Nenhum arquivo foi fornecido."
+        
+        # Criar diretório temporário
+        temp_dir = tempfile.mkdtemp()
+        print(f"Diretório temporário criado: {temp_dir}")
+        
+        # Salvar o conteúdo bytes em um arquivo temporário
+        temp_pdf = os.path.join(temp_dir, 'boletim.pdf')
+        try:
+            with open(temp_pdf, 'wb') as f:
+                # Se o arquivo já é bytes, escrever diretamente
+                if isinstance(file, bytes):
+                    f.write(file)
+                # Se é um arquivo do Gradio, ler o conteúdo primeiro
+                else:
+                    f.write(file.read())
+            print(f"Arquivo temporário criado: {temp_pdf}")
+        except Exception as e:
+            print(f"Erro ao salvar arquivo temporário: {str(e)}")
+            return None, f"Erro ao processar arquivo: {str(e)}"
+        
+        # Verificar se o arquivo foi criado corretamente
+        if not os.path.exists(temp_pdf) or os.path.getsize(temp_pdf) == 0:
+            return None, "Erro ao salvar o arquivo."
+        
+        # Validar o PDF
+        try:
+            import PyPDF2
+            with open(temp_pdf, 'rb') as pdf_file:
+                pdf_reader = PyPDF2.PdfReader(pdf_file)
+                if len(pdf_reader.pages) == 0:
+                    return None, "O PDF está vazio ou corrompido."
+                print(f"PDF válido com {len(pdf_reader.pages)} páginas")
+        except Exception as e:
+            print(f"Erro ao validar PDF: {str(e)}")
+            return None, "O arquivo não é um PDF válido."
+        
+        print("Iniciando extração das tabelas...")
+        try:
+            df = extrair_tabelas_pdf(temp_pdf)
+            print("Tabelas extraídas com sucesso")
+        except Exception as e:
+            print(f"Erro na extração das tabelas: {str(e)}")
+            return None, f"Erro ao extrair dados do PDF: {str(e)}"
+        
+        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 com sucesso")
+            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)
-            print("Arquivo de saída criado com sucesso")
                 
             return output_path, "Relatório gerado com sucesso!"
             
         except Exception as e:
-            print(f"Erro durante o processamento: {str(e)}")
             return None, f"Erro ao processar os dados: {str(e)}"
         
     except Exception as e:
-        print(f"Erro geral durante o processamento: {str(e)}")
+        print(f"Erro durante o processamento: {str(e)}")
         return None, f"Erro ao processar o boletim: {str(e)}"
         
     finally: