Update app.py

app.py

@@ -131,10 +131,21 @@ class ReportGenerator:
     def __init__(self, data: pd.DataFrame):
         self.data = data
         self.stats = self.calculate_statistics()
+        # Classificar alunos por níveis
+        self.data['Nível'] = self.data['Média de Acertos'].str.rstrip('%').astype(float).apply(self.classify_performance)
+        self.data = self.data.sort_values('Média de Acertos'.str.rstrip('%').astype(float), ascending=False)
+
+    def classify_performance(self, score):
+        if score >= 70:
+            return 'Avançado'
+        elif score >= 40:
+            return 'Intermediário'
+        else:
+            return 'Necessita Atenção'
 
     def calculate_statistics(self) -> Dict:
-        """Calcula estatísticas gerais da turma."""
-        return {
+        """Calcula estatísticas gerais e por grupo."""
+        basic_stats = {
             'media_acertos': float(self.data['Média de Acertos'].str.rstrip('%').astype(float).mean()),
             'desvio_padrao': float(self.data['Média de Acertos'].str.rstrip('%').astype(float).std()),
             'mediana_acertos': float(self.data['Média de Acertos'].str.rstrip('%').astype(float).median()),
@@ -143,38 +154,68 @@ class ReportGenerator:
             'media_tempo': str(pd.to_timedelta(self.data['Total Tempo']).mean())
         }
 
+        # Top performers
+        top_students = self.data.nlargest(3, 'Média de Acertos')
+        basic_stats['top_performers'] = top_students[['Nome do Aluno', 'Média de Acertos']].values.tolist()
+
+        # Mais eficientes (melhor relação acerto/tempo)
+        efficient_students = self.data.nlargest(3, 'Eficiência')
+        basic_stats['most_efficient'] = efficient_students[['Nome do Aluno', 'Eficiência', 'Média de Acertos']].values.tolist()
+
+        return basic_stats
+
     def generate_graphs(self) -> List[plt.Figure]:
         """Gera gráficos para o relatório."""
         graphs = []
         
-        # Distribuição de notas
-        plt.figure(figsize=(10, 6))
-        acertos_data = pd.Series(self.data['Média de Acertos']).str.rstrip('%').astype(float)
-        sns.histplot(data=acertos_data, bins=10)
-        plt.axvline(self.stats['media_acertos'], color='r', linestyle='--', 
-                   label=f'Média ({self.stats["media_acertos"]:.1f}%)')
-        plt.title('Distribuição das Notas')
-        plt.xlabel('Percentual de Acertos')
+        # 1. Distribuição de notas por nível
+        plt.figure(figsize=(12, 6))
+        nivel_counts = self.data['Nível'].value_counts()
+        colors = {'Avançado': 'green', 'Intermediário': 'yellow', 'Necessita Atenção': 'red'}
+        bars = plt.bar(nivel_counts.index, nivel_counts.values)
+        for i, bar in enumerate(bars):
+            bar.set_color(colors[nivel_counts.index[i]])
+            plt.text(bar.get_x() + bar.get_width()/2, bar.get_height(), 
+                    str(nivel_counts.values[i]), 
+                    ha='center', va='bottom')
+        plt.title('Distribuição dos Alunos por Nível de Desempenho')
         plt.ylabel('Número de Alunos')
-        plt.legend()
         graphs.append(plt.gcf())
         plt.close()
 
-        # Relação tempo x desempenho
+        # 2. Top 10 alunos
+        plt.figure(figsize=(12, 6))
+        top_10 = self.data.head(10)
+        acertos = top_10['Média de Acertos'].str.rstrip('%').astype(float)
+        plt.barh(top_10['Nome do Aluno'], acertos)
+        plt.title('Top 10 Alunos por Desempenho')
+        plt.xlabel('Percentual de Acertos')
+        for i, v in enumerate(acertos):
+            plt.text(v, i, f'{v:.1f}%', va='center')
+        plt.tight_layout()
+        graphs.append(plt.gcf())
+        plt.close()
+
+        # 3. Relação tempo x desempenho com níveis
         plt.figure(figsize=(10, 6))
-        tempo_segundos = pd.to_timedelta(self.data['Total Tempo']).dt.total_seconds()
-        acertos = pd.Series(self.data['Média de Acertos']).str.rstrip('%').astype(float)
-        plt.scatter(tempo_segundos / 60, acertos)
-        plt.title('Tempo x Desempenho')
+        colors = {'Avançado': 'green', 'Intermediário': 'yellow', 'Necessita Atenção': 'red'}
+        for nivel in colors:
+            mask = self.data['Nível'] == nivel
+            tempo = pd.to_timedelta(self.data[mask]['Total Tempo']).dt.total_seconds() / 60
+            acertos = self.data[mask]['Média de Acertos'].str.rstrip('%').astype(float)
+            plt.scatter(tempo, acertos, c=colors[nivel], label=nivel, alpha=0.6)
+        plt.title('Relação Tempo x Desempenho por Nível')
         plt.xlabel('Tempo Total (minutos)')
         plt.ylabel('Percentual de Acertos')
+        plt.legend()
+        plt.grid(True, alpha=0.3)
         graphs.append(plt.gcf())
         plt.close()
 
         return graphs
 
     def generate_pdf(self, output_path: str, graphs: List[plt.Figure]) -> None:
-        """Gera relatório em PDF."""
+        """Gera relatório em PDF com análise detalhada."""
         class PDF(FPDF):
             def header(self):
                 self.set_font('Arial', 'B', 15)
@@ -190,17 +231,27 @@ class ReportGenerator:
         pdf.set_font('Arial', '', 10)
         
         summary_text = f"""
-        Análise da Turma:
+        Visão Geral da Turma:
         - Média de Acertos: {self.stats['media_acertos']:.1f}%
         - Desvio Padrão: {self.stats['desvio_padrao']:.1f}%
         - Mediana: {self.stats['mediana_acertos']:.1f}%
-        - Número de Alunos: {self.stats['total_alunos']}
-        - Média de Tarefas por Aluno: {self.stats['media_tarefas']:.1f}
-        - Tempo Médio Total: {self.stats['media_tempo']}
+        - Total de Alunos: {self.stats['total_alunos']}
+        
+        Destaques:
+        Top 3 Melhores Desempenhos:
         """
         pdf.multi_cell(0, 10, summary_text)
 
-        # Gráficos
+        # Adicionar top performers
+        for aluno, nota in self.stats['top_performers']:
+            pdf.cell(0, 10, f"- {aluno}: {nota}", 0, 1)
+
+        pdf.ln()
+        pdf.cell(0, 10, "Alunos Mais Eficientes:", 0, 1)
+        for aluno, eficiencia, nota in self.stats['most_efficient']:
+            pdf.cell(0, 10, f"- {aluno}: Eficiência {eficiencia:.1f} (Acertos: {nota})", 0, 1)
+
+        # Gráficos e análises
         for i, graph in enumerate(graphs):
             pdf.add_page()
             graph_path = f'temp_graph_{i}.png'
@@ -208,28 +259,31 @@ class ReportGenerator:
             pdf.image(graph_path, x=10, y=30, w=270)
             os.remove(graph_path)
 
-        # Tabela de alunos
-        pdf.add_page()
-        pdf.set_font('Arial', 'B', 12)
-        pdf.cell(0, 10, 'Desempenho Individual', 0, 1)
-        
-        # Cabeçalhos
-        columns = ['Nome do Aluno', 'Média de Acertos', 'Tarefas', 'Tempo Total', 'Eficiência']
-        widths = [80, 30, 30, 30, 30]
-        pdf.set_font('Arial', 'B', 8)
-        for i, col in enumerate(columns):
-            pdf.cell(widths[i], 7, col, 1)
-        pdf.ln()
-
-        # Dados
-        pdf.set_font('Arial', '', 8)
-        for _, row in self.data.iterrows():
-            pdf.cell(widths[0], 6, str(row['Nome do Aluno'])[:40], 1)
-            pdf.cell(widths[1], 6, str(row['Média de Acertos']), 1)
-            pdf.cell(widths[2], 6, str(row['Tarefas Completadas']), 1)
-            pdf.cell(widths[3], 6, str(row['Total Tempo']), 1)
-            pdf.cell(widths[4], 6, str(row['Eficiência']), 1)
-            pdf.ln()
+        # Tabela detalhada por nível
+        for nivel in ['Avançado', 'Intermediário', 'Necessita Atenção']:
+            alunos_nivel = self.data[self.data['Nível'] == nivel]
+            if not alunos_nivel.empty:
+                pdf.add_page()
+                pdf.set_font('Arial', 'B', 12)
+                pdf.cell(0, 10, f'Alunos - Nível {nivel}', 0, 1)
+                
+                # Cabeçalhos
+                columns = ['Nome do Aluno', 'Média de Acertos', 'Tarefas', 'Tempo Total', 'Eficiência']
+                widths = [80, 30, 30, 30, 30]
+                pdf.set_font('Arial', 'B', 8)
+                for i, col in enumerate(columns):
+                    pdf.cell(widths[i], 7, col, 1)
+                pdf.ln()
+
+                # Dados
+                pdf.set_font('Arial', '', 8)
+                for _, row in alunos_nivel.iterrows():
+                    pdf.cell(widths[0], 6, str(row['Nome do Aluno'])[:40], 1)
+                    pdf.cell(widths[1], 6, str(row['Média de Acertos']), 1)
+                    pdf.cell(widths[2], 6, str(row['Tarefas Completadas']), 1)
+                    pdf.cell(widths[3], 6, str(row['Total Tempo']), 1)
+                    pdf.cell(widths[4], 6, str(row['Eficiência']), 1)
+                    pdf.ln()
 
         pdf.output(output_path)