Update app.py

app.py

@@ -54,7 +54,7 @@ def match_alunos(tarefas_csv_path, alunos_csv_path, contador_csv_path):
     tarefas_df.columns = tarefas_df.columns.str.strip()
     alunos_df.columns = alunos_df.columns.str.strip()
 
-    if 'Aluno' not in tarefas_df.columns or 'Nota' not in tarefas_df.columns or 'Duração' not in tarefas_df.columns:
+    if 'Aluno' not in tarefas_df.columns ou 'Nota' not in tarefas_df.columns ou 'Duração' not in tarefas_df.columns:
         print(f"Colunas 'Aluno', 'Nota' ou 'Duração' não encontradas no arquivo {tarefas_csv_path}. Pulando este arquivo.")
         return
 
@@ -136,10 +136,14 @@ def process_relatorios(contador_csv_path, relatorio_csv_path):
     media_tempo_medio_turma = tempo_medio_por_tarefa['Tempo Médio por Tarefa'].mean()
     media_tempo_medio_turma = format_timedelta(media_tempo_medio_turma)
 
+    # Calcula médias gerais da turma
+    media_acertos_turma = (contador_df['Acertos Absolutos'] / (contador_df['Tarefas Completadas'] * 2)).mean() * 100
+    media_tarefas_turma = contador_df['Tarefas Completadas'].mean()
+
     contador_df.to_csv(relatorio_csv_path, index=False, encoding='utf-8-sig')
-    return contador_df, media_tempo_medio_turma
+    return contador_df, media_tempo_medio_turma, media_acertos_turma, media_tarefas_turma
 
-def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pdf_path):
+def generate_pdf_report(dataframe, media_tempo_medio_turma, media_acertos_turma, media_tarefas_turma, html_path, output_pdf_path):
     class PDF(FPDF):
         def header(self):
             self.set_font('Arial', 'B', 12)
@@ -172,7 +176,7 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pd
     bars = plt.bar(top_students['Nome do Aluno'], top_students['Acertos Absolutos'], color='blue')
     plt.xlabel('Nome do Aluno')
     plt.ylabel('Acertos Absolutos')
-    plt.title(f'Top 5 Alunos - Acertos Absolutos (Tempo Médio da Turma: {media_tempo_medio_turma})')
+    plt.title('Top 5 Alunos - Acertos Absolutos')
     plt.xticks(rotation=45, ha='right')
     add_bar_labels(bars, top_students['Acertos Absolutos'])
     plt.tight_layout()
@@ -184,7 +188,7 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pd
     bars = 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(f'Top 5 Alunos - Percentual de Acertos (Tempo Médio da Turma: {media_tempo_medio_turma})')
+    plt.title('Top 5 Alunos - Percentual de Acertos')
     plt.xticks(rotation=45, ha='right')
     add_bar_labels(bars, top_students['Média de Acertos'].str.rstrip('%').astype('float'))
     plt.tight_layout()
@@ -196,7 +200,7 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pd
     bars = plt.bar(top_students['Nome do Aluno'], top_students['Tarefas Completadas'], color='red')
     plt.xlabel('Nome do Aluno')
     plt.ylabel('Tarefas Completadas')
-    plt.title(f'Top 5 Alunos - Tarefas Completadas (Tempo Médio da Turma: {media_tempo_medio_turma})')
+    plt.title('Top 5 Alunos - Tarefas Completadas')
     plt.xticks(rotation=45, ha='right')
     add_bar_labels(bars, top_students['Tarefas Completadas'])
     plt.tight_layout()
@@ -211,7 +215,7 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pd
     bars = plt.bar(top_time_students['Nome do Aluno'], top_time_students['Total Tempo'].dt.total_seconds(), color='purple')
     plt.xlabel('Nome do Aluno')
     plt.ylabel('Tempo Total (hh:mm:ss)')
-    plt.title(f'Top 5 Alunos - Tempo Total (Tempo Médio da Turma: {media_tempo_medio_turma})')
+    plt.title('Top 5 Alunos - Tempo Total')
     plt.xticks(rotation=45, ha='right')
     add_bar_labels(bars, top_time_students['Total Tempo'].apply(format_timedelta))
     plt.tight_layout()
@@ -219,6 +223,27 @@ def generate_pdf_report(dataframe, media_tempo_medio_turma, html_path, output_pd
     plt.savefig(graph_path)
     pdf.add_image(graph_path)
 
+    # Adiciona gráfico de resumo da turma
+    metrics = ['Tempo Médio (hh:mm:ss)', 'Média de Acertos (%)', 'Média de Tarefas']
+    values = [media_tempo_medio_turma, media_acertos_turma, media_tarefas_turma]
+
+    plt.figure(figsize=(10, 6))
+    bars = plt.bar(metrics, values, color=['blue', 'green', 'red'])
+    plt.xlabel('Métricas')
+    plt.ylabel('Valores')
+    plt.title('Resumo da Turma')
+    plt.xticks(rotation=45, ha='right')
+    for bar, value in zip(bars, values):
+        plt.annotate(f'{value:.2f}' if isinstance(value, float) else value, 
+                     xy=(bar.get_x() + bar.get_width() / 2, bar.get_height()), 
+                     xytext=(0, 3),  # 3 points vertical offset
+                     textcoords="offset points",
+                     ha='center', va='bottom')
+    plt.tight_layout()
+    graph_path = 'resumo_turma.png'
+    plt.savefig(graph_path)
+    pdf.add_image(graph_path)
+
     # Salvar o PDF com os gráficos
     temp_graphics_pdf = 'temp_graphics.pdf'
     pdf.output(temp_graphics_pdf)
@@ -295,14 +320,14 @@ def processar_relatorio(html_file, tarefa_files):
     contador_csv_path = os.path.join(output_directory, "contador_tarefas.csv")
     relatorio_csv_path = os.path.join(output_directory, "relatorio_final.csv")
     process_all_tarefas_in_directory(output_directory, alunos_csv_path, contador_csv_path, relatorio_csv_path)
-    df, media_tempo_medio_turma = process_relatorios(contador_csv_path, relatorio_csv_path)
+    df, media_tempo_medio_turma, media_acertos_turma, media_tarefas_turma = process_relatorios(contador_csv_path, relatorio_csv_path)
 
     # 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, encoding='utf-8-sig')
     
     pdf_output_path = os.path.join(output_directory, "relatorio_final.pdf")
-    generate_pdf_report(df, media_tempo_medio_turma, html_output_path, pdf_output_path)
+    generate_pdf_report(df, media_tempo_medio_turma, media_acertos_turma, media_tarefas_turma, html_output_path, pdf_output_path)
 
     return df.to_html(index=False), html_output_path, pdf_output_path