Update app.py

app.py

@@ -98,8 +98,19 @@ def extrair_tabelas_pdf(pdf_path):
         # Encontrar a tabela de notas
         for i, table in enumerate(tables):
             df_temp = table.df
-            if 'Disciplina' in str(df_temp.iloc[0,0]) or 'Bimestre' in str(df_temp.iloc[0,1]):
+            # Verificar se é a tabela de notas
+            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]):
                 df = df_temp
+                # Renomear as colunas corretamente
+                df = df.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'
+                })
                 break
         
         if df.empty:
@@ -114,6 +125,16 @@ def extrair_tabelas_pdf(pdf_path):
     except Exception as e:
         print(f"Erro na extração das tabelas: {str(e)}")
         raise
+        
+        # Adicionar informações do aluno ao DataFrame
+        for key, value in info_aluno.items():
+            df.attrs[key] = value
+        
+        return df
+        
+    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."""