Update app.py

app.py

@@ -129,20 +129,32 @@ def extrair_tabelas_pdf(pdf_path):
         if len(tables) == 0:
             raise ValueError("Nenhuma tabela foi extraída do PDF.")
         
-        df = tables[0].df
-        
+        # Processar todas as tabelas procurando informações do aluno
         info_aluno = {}
-        for i, row in df.iterrows():
-            if 'Nome do Aluno' in str(row[0]):
-                info_aluno['nome'] = row[1].strip() if len(row) > 1 else ''
-            elif 'RA' in str(row[0]):
-                info_aluno['ra'] = row[1].strip() if len(row) > 1 else ''
-            elif 'Escola' in str(row[0]):
-                info_aluno['escola'] = row[1].strip() if len(row) > 1 else ''
-            elif 'Turma' in str(row[0]):
-                info_aluno['turma'] = row[1].strip() if len(row) > 1 else ''
-        
-        for i, table in enumerate(tables):
+        for table in tables:
+            df_temp = table.df
+            for i, row in df_temp.iterrows():
+                # Procurar por variações dos campos
+                if any(campo in str(row[0]).upper() for campo in ['NOME DO ALUNO:', 'NOME:', 'ALUNO:']):
+                    nome = row[1].strip() if len(row) > 1 else ''
+                    if nome and 'nome' not in info_aluno:  # Evitar sobrescrever se já encontrou
+                        info_aluno['nome'] = nome
+                elif any(campo in str(row[0]).upper() for campo in ['RA:', 'R.A.:', 'REGISTRO ACADÊMICO:']):
+                    ra = row[1].strip() if len(row) > 1 else ''
+                    if ra and 'ra' not in info_aluno:  # Evitar sobrescrever se já encontrou
+                        info_aluno['ra'] = ra
+                elif any(campo in str(row[0]).upper() for campo in ['ESCOLA:', 'UNIDADE:', 'UNIDADE ESCOLAR:']):
+                    escola = row[1].strip() if len(row) > 1 else ''
+                    if escola and 'escola' not in info_aluno:
+                        info_aluno['escola'] = escola
+                elif any(campo in str(row[0]).upper() for campo in ['TURMA:', 'SÉRIE:', 'ANO/SÉRIE:']):
+                    turma = row[1].strip() if len(row) > 1 else ''
+                    if turma and 'turma' not in info_aluno:
+                        info_aluno['turma'] = turma
+
+        # Encontrar a tabela com as notas
+        df = None
+        for table in tables:
             df_temp = table.df
             if any('Disciplina' in str(col) for col in df_temp.iloc[0]) or \
                any('Bimestre' in str(col) for col in df_temp.iloc[0]):
@@ -157,12 +169,17 @@ def extrair_tabelas_pdf(pdf_path):
                 })
                 break
         
-        if df.empty:
-            raise ValueError("A tabela extraída está vazia.")
+        if df is None or df.empty:
+            raise ValueError("A tabela de notas não foi encontrada.")
         
-        for key, value in info_aluno.items():
-            df.attrs[key] = value
+        # Adicionar informações do aluno ao DataFrame
+        # Garantir valores padrão se não encontrados
+        df.attrs['nome'] = info_aluno.get('nome', 'Nome não encontrado')
+        df.attrs['ra'] = info_aluno.get('ra', 'RA não encontrado')
+        df.attrs['escola'] = info_aluno.get('escola', 'Escola não encontrada')
+        df.attrs['turma'] = info_aluno.get('turma', 'Turma não encontrada')
         
+        print("Informações do aluno encontradas:", info_aluno)
         return df
         
     except Exception as e:
@@ -462,14 +479,31 @@ def gerar_relatorio_pdf(df, disciplinas_dados, grafico_basica, grafico_diversifi
     
     pdf.set_font('Helvetica', '', 11)
     if hasattr(df, 'attrs'):
-        if 'nome' in df.attrs:
-            pdf.cell(0, 7, f'Nome: {df.attrs["nome"]}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
-        if 'ra' in df.attrs:
-            pdf.cell(0, 7, f'RA: {df.attrs["ra"]}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
-        if 'escola' in df.attrs:
-            pdf.cell(0, 7, f'Escola: {df.attrs["escola"]}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
-        if 'turma' in df.attrs:
-            pdf.cell(0, 7, f'Turma: {df.attrs["turma"]}', 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT, align='L')
+        nome = df.attrs.get('nome', 'Nome não encontrado')
+        ra = df.attrs.get('ra', 'RA não encontrado')
+        escola = df.attrs.get('escola', 'Escola não encontrada')
+        turma = df.attrs.get('turma', 'Turma não encontrada')
+        
+        # Adicionar rótulos em negrito
+        pdf.set_font('Helvetica', 'B', 11)
+        pdf.cell(30, 7, 'Nome:', 0, 0)
+        pdf.set_font('Helvetica', '', 11)
+        pdf.cell(0, 7, nome, 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT)
+        
+        pdf.set_font('Helvetica', 'B', 11)
+        pdf.cell(30, 7, 'RA:', 0, 0)
+        pdf.set_font('Helvetica', '', 11)
+        pdf.cell(0, 7, ra, 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT)
+        
+        pdf.set_font('Helvetica', 'B', 11)
+        pdf.cell(30, 7, 'Escola:', 0, 0)
+        pdf.set_font('Helvetica', '', 11)
+        pdf.cell(0, 7, escola, 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT)
+        
+        pdf.set_font('Helvetica', 'B', 11)
+        pdf.cell(30, 7, 'Turma:', 0, 0)
+        pdf.set_font('Helvetica', '', 11)
+        pdf.cell(0, 7, turma, 0, new_x=XPos.LMARGIN, new_y=YPos.NEXT)
     
     pdf.ln(10)