Update app.py

app.py

@@ -6,7 +6,6 @@ import matplotlib.pyplot as plt
 from datetime import timedelta
 from fpdf import FPDF
 import numpy as np
-import tabula
 
 def parse_duration(duration_str):
     try:
@@ -151,30 +150,28 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, output_pdf_path):
             self.cell(0, 10, f'Page {self.page_no()}', 0, 0, 'C')
 
         def add_table(self, dataframe):
-            self.set_font("Arial", "B", 10)
-
-            # Calcular larguras dinâmicas das colunas
-            col_widths = []
-            for col in dataframe.columns:
-                max_width = max(self.get_string_width(str(cell)) for cell in dataframe[col])
-                col_widths.append(max_width + 4)  # Adicionar um pequeno espaçamento
-
-            row_height = self.font_size * 2
+            self.set_font('Arial', 'B', 10)
+            col_width = self.w / len(dataframe.columns)
+            row_height = self.font_size
 
             # Adiciona os cabeçalhos
-            for i, col in enumerate(dataframe.columns):
-                self.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 quebra de linha na coluna "Nome do Aluno"
-            self.set_font("Arial", "", 10)
+            # Adiciona os dados com quebras de página
+            self.set_font('Arial', '', 10)
             for row in dataframe.itertuples(index=False):
-                for i, item in enumerate(row):
-                    if i == 0:  # Coluna "Nome do Aluno"
-                        self.multi_cell(col_widths[i], row_height, str(item), border=1, align="C")
-                    else:
-                        self.cell(col_widths[i], row_height, str(item), border=1, align="C")
-                self.ln(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 col in dataframe.columns:
+                        self.cell(col_width, row_height * 2, col, border=1)
+                    self.ln(row_height * 2)
+                    self.set_font('Arial', '', 10)
 
         def add_image(self, image_path):
             self.add_page()
@@ -295,18 +292,15 @@ with gr.Blocks(theme=theme) as interface:
             gr.Markdown("## Arquivos Excel (Relatórios de Tarefas)")
             excel_files = gr.Files(label="Arraste os arquivos .xlsx aqui", type="binary", file_count="multiple")
 
-    generate_btn = gr.Button("Gerar Relatório", variant="primary")
+    generate_btn = gr.Button("Gerar Relatório", variant="primary")  # Destaque no botão
     output_html = gr.HTML()
-    pdf_output = gr.File(label="Download PDF Report")
+    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 {output_html: html_content, pdf_output: pdf_path}
-
-    generate_btn.click(
-        fn=wrapper,
-        inputs=[html_file, excel_files],
-        outputs=[output_html, pdf_output]
-    )
-   
-interface.launch()
+        return html_content, html_path, pdf_path
+
+    generate_btn.click(fn=wrapper, inputs=[html_file, excel_files], outputs=[output_html, download_html_btn, download_pdf_btn])
+
+interface.launch()