Update app.py

app.py

@@ -151,27 +151,27 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, output_pdf_path):
 
         def add_table(self, dataframe):
             self.set_font('Arial', 'B', 10)
-            col_width = self.w / len(dataframe.columns)
-            row_height = self.font_size
+            col_widths = [30, 35, 40, 30, 40, 40]
+            row_height = self.font_size * 2
 
             # Adiciona os cabeçalhos
-            for col in dataframe.columns:
-                self.cell(col_width, row_height * 2, col, border=1)
-            self.ln(row_height * 2)
+            for i, col in enumerate(dataframe.columns):
+                self.cell(col_widths[i], row_height, col, border=1, align='C')
+            self.ln(row_height)
 
             # Adiciona os dados com quebras de página
             self.set_font('Arial', '', 10)
             for row in dataframe.itertuples(index=False):
-                for item in row:
-                    self.cell(col_width, row_height * 2, str(item), border=1)
-                self.ln(row_height * 2)
-                if self.get_y() > self.page_break_trigger - 2 * row_height:
+                if self.get_y() > self.page_break_trigger - row_height:
                     self.add_page()
                     self.set_font('Arial', 'B', 10)
-                    for col in dataframe.columns:
-                        self.cell(col_width, row_height * 2, col, border=1)
-                    self.ln(row_height * 2)
+                    for i, col in enumerate(dataframe.columns):
+                        self.cell(col_widths[i], row_height, col, border=1, align='C')
+                    self.ln(row_height)
                     self.set_font('Arial', '', 10)
+                for i, item in enumerate(row):
+                    self.cell(col_widths[i], row_height, str(item), border=1, align='C')
+                self.ln(row_height)
 
         def add_image(self, image_path):
             self.add_page()