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_widths = [30, 40, 40, 30, 30, 40]
-            row_height = self.font_size * 2
+            col_width = self.w / len(dataframe.columns)
+            row_height = self.font_size
 
             # Adiciona os cabeçalhos
-            for i, col in enumerate(dataframe.columns):
-                self.multi_cell(col_widths[i], row_height, col, border=1, align='C')
-            self.ln(row_height)
+            for col in dataframe.columns:
+                self.cell(col_width, row_height * 2, col, border=1)
+            self.ln(row_height * 2)
 
             # Adiciona os dados com quebras de página
             self.set_font('Arial', '', 10)
             for row in dataframe.itertuples(index=False):
-                if self.get_y() > self.page_break_trigger - row_height:
+                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:
                     self.add_page()
                     self.set_font('Arial', 'B', 10)
-                    for i, col in enumerate(dataframe.columns):
-                        self.multi_cell(col_widths[i], row_height, col, border=1, align='C')
-                    self.ln(row_height)
+                    for col in dataframe.columns:
+                        self.cell(col_width, row_height * 2, col, border=1)
+                    self.ln(row_height * 2)
                     self.set_font('Arial', '', 10)
-                for i, item in enumerate(row):
-                    self.multi_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()
@@ -294,21 +294,13 @@ with gr.Blocks(theme=theme) as interface:
 
     generate_btn = gr.Button("Gerar Relatório", variant="primary")  # Destaque no botão
     output_html = gr.HTML()
-    download_html = gr.File(label="Download HTML Report")
-    download_pdf = gr.File(label="Download PDF Report")
-
-    with gr.Row():  # Crie uma nova linha para os botões
-        download_html_btn = gr.Button("Baixar Relatório HTML")
-        download_pdf_btn = gr.Button("Baixar Relatório PDF")
+    download_html_btn = gr.File(label="Download HTML Report")
+    download_pdf_btn = gr.File(label="Download PDF Report")
 
     def wrapper(html_file, excel_files):
         html_content, html_path, pdf_path = processar_relatorio(html_file, excel_files)
         return html_content, html_path, pdf_path
 
-    generate_btn.click(
-        fn=wrapper,
-        inputs=[html_file, excel_files],
-        outputs=[output_html, download_html, download_pdf]
-    )
+    generate_btn.click(fn=wrapper, inputs=[html_file, excel_files], outputs=[output_html, download_html_btn, download_pdf_btn])
 
 interface.launch()