Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import pandas as pd
 import re
 import os
+import matplotlib.pyplot as plt
 from datetime import timedelta
+from fpdf import FPDF
 
 def parse_duration(duration_str):
     try:
@@ -46,9 +48,6 @@ def match_alunos(tarefas_csv_path, alunos_csv_path, contador_csv_path):
         print(f"Arquivo {tarefas_csv_path} ou {alunos_csv_path} está vazio. Pulando...")
         return
 
-    print(f"Tarefas DataFrame (antes da normalização):\n{tarefas_df.head()}")
-    print(f"Alunos DataFrame (antes da normalização):\n{alunos_df.head()}")
-
     tarefas_df.columns = tarefas_df.columns.str.strip()
     alunos_df.columns = alunos_df.columns.str.strip()
 
@@ -75,8 +74,6 @@ def match_alunos(tarefas_csv_path, alunos_csv_path, contador_csv_path):
 
     alunos_df['Aluno_Pattern'] = alunos_df.apply(lambda row: generate_aluno_pattern(row['RA'], row['Dig. RA']), axis=1)
 
-    print(f"Alunos DataFrame (com padrão):\n{alunos_df.head()}")
-
     def extract_aluno_pattern(nome):
         if isinstance(nome, str):
             match = re.search(r'\d+.*', nome.lower())
@@ -86,12 +83,8 @@ def match_alunos(tarefas_csv_path, alunos_csv_path, contador_csv_path):
     tarefas_df['Aluno_Pattern'] = tarefas_df['Aluno'].apply(extract_aluno_pattern)
     tarefas_df['Duração'] = tarefas_df['Duração'].apply(parse_duration)
 
-    print(f"Tarefas DataFrame (com padrão):\n{tarefas_df.head()}")
-
     matched_alunos = alunos_df[alunos_df['Aluno_Pattern'].isin(tarefas_df['Aluno_Pattern'])]
 
-    print(f"Matched Alunos DataFrame:\n{matched_alunos.head()}")
-
     result_df = matched_alunos[['Nome do Aluno']].drop_duplicates()
 
     for aluno in result_df['Nome do Aluno']:
@@ -108,8 +101,6 @@ def match_alunos(tarefas_csv_path, alunos_csv_path, contador_csv_path):
         else:
             contador_df = pd.concat([contador_df, pd.DataFrame({'Nome do Aluno': [aluno], 'Tarefas Completadas': [1], 'Acertos Absolutos': [nota_total], 'Total Tempo': [str(tempo_total)]})], ignore_index=True)
 
-    print(f"Contador DataFrame (atualizado):\n{contador_df.head()}")
-
     contador_df.to_csv(contador_csv_path, index=False)
 
     return result_df
@@ -118,9 +109,7 @@ def process_all_tarefas_in_directory(directory, alunos_csv_path, contador_csv_pa
     tarefas_files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith('.csv') and f not in ['alunos_fim.csv', 'contador_tarefas.csv']]
 
     for i, tarefas_file in enumerate(tarefas_files):
-        print(f"Processando arquivo {i+1}/{len(tarefas_files)}: {tarefas_file}")
         match_alunos(tarefas_file, alunos_csv_path, contador_csv_path)
-        print(f"Arquivo {tarefas_file} processado.")
 
     process_relatorios(contador_csv_path, relatorio_csv_path)
 
@@ -134,6 +123,78 @@ def process_relatorios(contador_csv_path, relatorio_csv_path):
     contador_df.to_csv(relatorio_csv_path, index=False)
     return contador_df
 
+def generate_pdf_report(dataframe, output_pdf_path):
+    class PDF(FPDF):
+        def header(self):
+            self.set_font('Arial', 'B', 12)
+            self.cell(0, 10, 'Relatório de Tarefas', 0, 1, 'C')
+
+        def footer(self):
+            self.set_y(-15)
+            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
+            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)
+
+        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
+    top_students = dataframe.nlargest(5, 'Acertos Absolutos')
+    plt.figure(figsize=(10, 6))
+    plt.bar(top_students['Nome do Aluno'], top_students['Acertos Absolutos'], color='blue')
+    plt.xlabel('Nome do Aluno')
+    plt.ylabel('Acertos Absolutos')
+    plt.title('Top 5 Alunos - Acertos Absolutos')
+    plt.xticks(rotation=45, ha='right')
+    plt.tight_layout()
+    graph_path = 'top_5_acertos_absolutos.png'
+    plt.savefig(graph_path)
+    pdf.add_image(graph_path)
+
+    plt.figure(figsize=(10, 6))
+    plt.bar(top_students['Nome do Aluno'], top_students['Média de Acertos'].str.rstrip('%').astype('float'), color='green')
+    plt.xlabel('Nome do Aluno')
+    plt.ylabel('Percentual de Acertos (%)')
+    plt.title('Top 5 Alunos - Percentual de Acertos')
+    plt.xticks(rotation=45, ha='right')
+    plt.tight_layout()
+    graph_path = 'top_5_percentual_acertos.png'
+    plt.savefig(graph_path)
+    pdf.add_image(graph_path)
+
+    plt.figure(figsize=(10, 6))
+    plt.bar(top_students['Nome do Aluno'], top_students['Tarefas Completadas'], color='red')
+    plt.xlabel('Nome do Aluno')
+    plt.ylabel('Tarefas Completadas')
+    plt.title('Top 5 Alunos - Tarefas Completadas')
+    plt.xticks(rotation=45, ha='right')
+    plt.tight_layout()
+    graph_path = 'top_5_tarefas_completadas.png'
+    plt.savefig(graph_path)
+    pdf.add_image(graph_path)
+
+    pdf.output(output_pdf_path)
 
 def processar_relatorio(html_file, tarefa_files):
     input_directory = "temp_files"  # Diretório temporário para os arquivos
@@ -169,11 +230,14 @@ def processar_relatorio(html_file, tarefa_files):
     process_all_tarefas_in_directory(output_directory, alunos_csv_path, contador_csv_path, relatorio_csv_path)
     df = process_relatorios(contador_csv_path, relatorio_csv_path)
 
-    # Salva o relatório em HTML
+    # Salva o relatório em HTML e PDF
     html_output_path = os.path.join(output_directory, "relatorio_final.html")
     df.to_html(html_output_path, index=False)
+    
+    pdf_output_path = os.path.join(output_directory, "relatorio_final.pdf")
+    generate_pdf_report(df, pdf_output_path)
 
-    return df.to_html(index=False), html_output_path
+    return df.to_html(index=False), html_output_path, pdf_output_path
 
 # --- Interface Gradio ---
 with gr.Blocks() as interface:
@@ -182,8 +246,13 @@ with gr.Blocks() as interface:
     excel_files = gr.Files(label="Upload Excel Files (Relatórios de Tarefas)", type="binary", file_count="multiple")
     generate_btn = gr.Button("Generate Report")
     output_html = gr.HTML()
-    download_btn = gr.File(label="Download 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 html_content, html_path, pdf_path
 
-    generate_btn.click(fn=processar_relatorio, inputs=[html_file, excel_files], outputs=[output_html, download_btn])
+    generate_btn.click(fn=wrapper, inputs=[html_file, excel_files], outputs=[output_html, download_html_btn, download_pdf_btn])
 
 interface.launch()