Update app.py

app.py

@@ -157,82 +157,86 @@ class ReportGenerator:
         return basic_stats
 
     def generate_graphs(self) -> List[plt.Figure]:
-        plt.style.use('seaborn')
-        graphs = []
-        
-        # 1. Distribuição por nível
-        plt.figure(figsize=(12, 6))
-        nivel_counts = self.data['Nível'].value_counts()
-        colors = {'Avançado': '#2ecc71', 'Intermediário': '#f1c40f', 'Necessita Atenção': '#e74c3c'}
-        
-        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', pad=20, fontsize=12)
-        plt.ylabel('Número de Alunos', labelpad=10)
-        plt.grid(True, alpha=0.3)
-        graphs.append(plt.gcf())
-        plt.close()
-
-        # 2. Top 10 alunos
-        plt.figure(figsize=(12, 6))
-        top_10 = self.data.head(10)
-        bars = plt.barh(top_10['Nome do Aluno'], top_10['Acertos Absolutos'], 
-                       color='#3498db')
-        
-        plt.title('Top 10 Alunos - Acertos Absolutos', pad=20, fontsize=12)
-        plt.xlabel('Número de Acertos', labelpad=10)
-        
-        for i, bar in enumerate(bars):
-            plt.text(bar.get_width(), bar.get_y() + bar.get_height()/2,
-                    f' {bar.get_width():.0f}', 
-                    va='center')
-        
-        plt.grid(True, alpha=0.3)
-        plt.tight_layout()
-        graphs.append(plt.gcf())
-        plt.close()
-
-        # 3. Relação tempo x acertos
-        plt.figure(figsize=(12, 6))
-        for nivel in colors:
-            mask = self.data['Nível'] == nivel
-            tempo = pd.to_timedelta(self.data[mask]['Total Tempo']).dt.total_seconds() / 60
-            plt.scatter(tempo, self.data[mask]['Acertos Absolutos'],
-                       c=colors[nivel], label=nivel, alpha=0.6, s=100)
-        
-        plt.title('Relação entre Tempo e Acertos por Nível', pad=20, fontsize=12)
-        plt.xlabel('Tempo Total (minutos)', labelpad=10)
-        plt.ylabel('Número de Acertos', labelpad=10)
-        plt.legend()
-        plt.grid(True, alpha=0.3)
-        graphs.append(plt.gcf())
-        plt.close()
-
-        # 4. Relação Tarefas x Acertos com linha de tendência
-        plt.figure(figsize=(12, 6))
-        plt.scatter(self.data['Tarefas Completadas'], self.data['Acertos Absolutos'],
-                   color='#3498db', alpha=0.6, s=100)
-        
-        z = np.polyfit(self.data['Tarefas Completadas'], self.data['Acertos Absolutos'], 1)
-        p = np.poly1d(z)
-        x_range = np.linspace(self.data['Tarefas Completadas'].min(), 
-                            self.data['Tarefas Completadas'].max(), 100)
-        plt.plot(x_range, p(x_range), "r--", alpha=0.8, label='Tendência')
-        
-        plt.title('Relação entre Tarefas Completadas e Acertos', pad=20, fontsize=12)
-        plt.xlabel('Número de Tarefas Completadas', labelpad=10)
-        plt.ylabel('Número de Acertos', labelpad=10)
-        plt.legend()
-        plt.grid(True, alpha=0.3)
-        graphs.append(plt.gcf())
-        plt.close()
-
-        return graphs
+    graphs = []
+    
+    # Configurações globais de estilo
+    plt.rcParams['figure.figsize'] = [12, 6]
+    plt.rcParams['axes.grid'] = True
+    plt.rcParams['grid.alpha'] = 0.3
+    plt.rcParams['font.size'] = 10
+    plt.rcParams['axes.titlesize'] = 12
+    plt.rcParams['axes.titlepad'] = 20
+    plt.rcParams['axes.labelpad'] = 10
+    
+    # 1. Distribuição por nível
+    plt.figure()
+    nivel_counts = self.data['Nível'].value_counts()
+    colors = {'Avançado': '#2ecc71', 'Intermediário': '#f1c40f', 'Necessita Atenção': '#e74c3c'}
+    
+    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')
+    graphs.append(plt.gcf())
+    plt.close()
+
+    # 2. Top 10 alunos
+    plt.figure()
+    top_10 = self.data.head(10)
+    bars = plt.barh(top_10['Nome do Aluno'], top_10['Acertos Absolutos'], 
+                   color='#3498db')
+    
+    plt.title('Top 10 Alunos - Acertos Absolutos')
+    plt.xlabel('Número de Acertos')
+    
+    for i, bar in enumerate(bars):
+        plt.text(bar.get_width(), bar.get_y() + bar.get_height()/2,
+                f' {bar.get_width():.0f}', 
+                va='center')
+    
+    plt.tight_layout()
+    graphs.append(plt.gcf())
+    plt.close()
+
+    # 3. Relação tempo x acertos
+    plt.figure()
+    for nivel in colors:
+        mask = self.data['Nível'] == nivel
+        tempo = pd.to_timedelta(self.data[mask]['Total Tempo']).dt.total_seconds() / 60
+        plt.scatter(tempo, self.data[mask]['Acertos Absolutos'],
+                   c=colors[nivel], label=nivel, alpha=0.6, s=100)
+    
+    plt.title('Relação entre Tempo e Acertos por Nível')
+    plt.xlabel('Tempo Total (minutos)')
+    plt.ylabel('Número de Acertos')
+    plt.legend()
+    graphs.append(plt.gcf())
+    plt.close()
+
+    # 4. Relação Tarefas x Acertos com linha de tendência
+    plt.figure()
+    plt.scatter(self.data['Tarefas Completadas'], self.data['Acertos Absolutos'],
+                color='#3498db', alpha=0.6, s=100)
+    
+    z = np.polyfit(self.data['Tarefas Completadas'], self.data['Acertos Absolutos'], 1)
+    p = np.poly1d(z)
+    x_range = np.linspace(self.data['Tarefas Completadas'].min(), 
+                         self.data['Tarefas Completadas'].max(), 100)
+    plt.plot(x_range, p(x_range), "r--", alpha=0.8, label='Tendência')
+    
+    plt.title('Relação entre Tarefas Completadas e Acertos')
+    plt.xlabel('Número de Tarefas Completadas')
+    plt.ylabel('Número de Acertos')
+    plt.legend()
+    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 com análise detalhada."""