Update app.py

app.py

@@ -137,7 +137,12 @@ def process_relatorios(contador_csv_path, relatorio_csv_path):
     contador_df.to_csv(relatorio_csv_path, index=False)
     return contador_df, media_tempo_medio_turma
 
-def generate_pdf_report(dataframe, media_tempo_medio_turma, output_pdf_path):
+import pdfkit
+from fpdf import FPDF
+import matplotlib.pyplot as plt
+import os
+
+def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pdf_path):
     class PDF(FPDF):
         def header(self):
             self.set_font('Arial', 'B', 12)
@@ -149,37 +154,12 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, output_pdf_path):
             self.set_font('Arial', 'I', 8)
             self.cell(0, 10, f'Page {self.page_no()}', 0, 0, 'C')
 
-        def add_table(self, dataframe):
-            self.set_font('Arial', 'B', 10)
-            col_width = self.w / len(dataframe.columns)
-            row_height = self.font_size
-
-            # Adiciona os cabeçalhos
-            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):
-                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()
             self.image(image_path, x=10, y=10, w=270)
 
     pdf = PDF(orientation='L', unit='mm', format='A4')
     pdf.add_page()
-    pdf.add_table(dataframe)
 
     # Gerar gráficos e adicionar ao PDF
     def add_bar_labels(bars):
@@ -228,7 +208,26 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, output_pdf_path):
     plt.savefig(graph_path)
     pdf.add_image(graph_path)
 
-    pdf.output(output_pdf_path)
+    # Salvar o PDF com os gráficos
+    temp_graphics_pdf = 'temp_graphics.pdf'
+    pdf.output(temp_graphics_pdf)
+
+    # Converter o HTML para PDF
+    temp_html_pdf = 'temp_html.pdf'
+    pdfkit.from_file(html_path, temp_html_pdf)
+
+    # Combinar os PDFs
+    from PyPDF2 import PdfMerger
+
+    merger = PdfMerger()
+    merger.append(temp_html_pdf)
+    merger.append(temp_graphics_pdf)
+    merger.write(output_pdf_path)
+    merger.close()
+
+    # Remover arquivos temporários
+    os.remove(temp_graphics_pdf)
+    os.remove(temp_html_pdf)
 
 def processar_relatorio(html_file, tarefa_files):
     input_directory = "temp_files"  # Diretório temporário para os arquivos
@@ -269,10 +268,11 @@ def processar_relatorio(html_file, tarefa_files):
     df.to_html(html_output_path, index=False)
     
     pdf_output_path = os.path.join(output_directory, "relatorio_final.pdf")
-    generate_pdf_report(df, media_tempo_medio_turma, pdf_output_path)
+    generate_pdf_report(df, media_tempo_medio_turma, html_output_path, pdf_output_path)
 
     return df.to_html(index=False), html_output_path, pdf_output_path
 
+
 # Tema personalizado
 theme = gr.themes.Default(
     primary_hue="blue",  # Cor principal (tons de azul)