Update app.py

app.py

@@ -7,6 +7,7 @@ from fpdf import FPDF
 import tempfile
 import os
 import matplotlib
+import shutil
 matplotlib.use('Agg')
 
 def extrair_tabelas_pdf(pdf_path):
@@ -193,16 +194,26 @@ def processar_boletim(file):
         # Verificar se o arquivo é válido
         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 PDF enviado
+        # Verificar se o arquivo tem conteúdo
+        if not hasattr(file, 'name') or not os.path.exists(file.name):
+            return None, "Arquivo inválido ou corrompido."
+            
+        if os.path.getsize(file.name) == 0:
+            return None, "O arquivo está vazio."
+        
+        # Copiar o arquivo para o diretório temporário
         temp_pdf = os.path.join(temp_dir, 'boletim.pdf')
-        with open(temp_pdf, 'wb') as f:
-            f.write(file.read())
-        print(f"PDF salvo em: {temp_pdf}")
+        shutil.copy2(file.name, temp_pdf)
+        print(f"PDF copiado para: {temp_pdf}")
+        
+        # Verificar se a cópia foi bem sucedida
+        if not os.path.exists(temp_pdf) or os.path.getsize(temp_pdf) == 0:
+            return None, "Erro ao copiar o arquivo."
         
         # Extrair tabelas do PDF
         print("Iniciando extração das tabelas...")
@@ -211,7 +222,17 @@ def processar_boletim(file):
         
         # Verificar se o DataFrame foi criado corretamente
         if df is None or df.empty:
-            raise ValueError("Não foi possível extrair dados do PDF.")
+            return None, "Não foi possível extrair dados do PDF."
+        
+        try:
+            # 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']
+        except:
+            return None, "O formato do PDF não corresponde ao esperado."
         
         # Processar notas
         colunas_notas = ['Nota B1', 'Nota B2', 'Nota B3', 'Nota B4']
@@ -231,9 +252,16 @@ def processar_boletim(file):
         pdf_path = gerar_relatorio_pdf(df, grafico1_path, grafico2_path)
         print("Relatório PDF gerado")
         
+        # Verificar se o PDF foi gerado
+        if not os.path.exists(pdf_path) or os.path.getsize(pdf_path) == 0:
+            return None, "Erro ao gerar o relatório PDF."
+        
         # Ler PDF gerado
         with open(pdf_path, 'rb') as f:
             pdf_content = f.read()
+        
+        if not pdf_content:
+            return None, "O relatório PDF gerado está vazio."
             
         return pdf_content, "Relatório gerado com sucesso!"
         
@@ -245,9 +273,7 @@ def processar_boletim(file):
         # Limpar arquivos temporários
         if temp_dir and os.path.exists(temp_dir):
             try:
-                for file in os.listdir(temp_dir):
-                    os.remove(os.path.join(temp_dir, file))
-                os.rmdir(temp_dir)
+                shutil.rmtree(temp_dir)
                 print("Arquivos temporários limpos")
             except Exception as e:
                 print(f"Erro ao limpar arquivos temporários: {str(e)}")