Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 import torch
 from transformers import pipeline
 from datetime import datetime
+import nltk
+from nltk.tokenize import word_tokenize
+from nltk.corpus import stopwords
+from collections import Counter
 
 class GeradorTrilhaAprendizado:
     def __init__(self):
@@ -13,6 +17,10 @@ class GeradorTrilhaAprendizado:
                                 model="gpt2-large",
                                 device=self.device)
         self.historico = []
+        # Inicializa NLTK
+        nltk.download('punkt')
+        nltk.download('stopwords')
+        self.stop_words = set(stopwords.words('portuguese'))
 
     def processar_audio(self, 
                        audio_path: str, 
@@ -23,34 +31,12 @@ class GeradorTrilhaAprendizado:
                        incluir_recursos: bool = True) -> tuple:
         try:
             transcricao = self.transcriber(audio_path)["text"]
-            
-            prompt = f"""
-            Crie uma trilha de aprendizado estruturada para nível {nivel} na área de {area} 
-            com duração de {duracao} com base no seguinte objetivo:
-            {transcricao}
-
-            Inclua:
-            1. Objetivos de aprendizagem específicos
-            2. Pré-requisitos necessários
-            3. Módulos de estudo sequenciais
-            4. Projetos práticos
-            5. Marcos de avaliação
-            6. Cronograma sugerido
-            
-            Trilha de aprendizado para {nome_trilha}:
-            """
-            
-            analise = self.generator(prompt, 
-                                   max_length=500,
-                                   num_return_sequences=1,
-                                   temperature=0.7,
-                                   top_p=0.95)[0]["generated_text"]
+            analise = self._gerar_trilha_personalizada(transcricao, nivel, area, duracao)
             
             if incluir_recursos:
                 recursos = self._gerar_recursos(nivel, area, transcricao)
                 analise += "\n\n" + recursos
 
-            # Salvar no histórico
             self.historico.append({
                 "data": datetime.now().strftime("%d/%m/%Y %H:%M"),
                 "nome": nome_trilha,
@@ -75,12 +61,219 @@ class GeradorTrilhaAprendizado:
                 "❌ Ocorreu um erro"
             )
 
+    def _gerar_trilha_personalizada(self, transcricao: str, nivel: str, area: str, duracao: str) -> str:
+        palavras_chave = self._extrair_palavras_chave(transcricao)
+        
+        modulos_por_area = {
+            "programação": {
+                "iniciante": [
+                    "Lógica de Programação Básica",
+                    "Introdução a Algoritmos",
+                    "Fundamentos de HTML/CSS",
+                    "JavaScript Básico"
+                ],
+                "intermediário": [
+                    "Estruturas de Dados",
+                    "Programação Orientada a Objetos",
+                    "Frameworks Front-end",
+                    "Banco de Dados"
+                ],
+                "avançado": [
+                    "Arquitetura de Software",
+                    "DevOps e CI/CD",
+                    "Microsserviços",
+                    "Segurança e Performance"
+                ]
+            },
+            "data science": {
+                "iniciante": [
+                    "Estatística Básica",
+                    "Python para Análise de Dados",
+                    "SQL Fundamental",
+                    "Visualização de Dados"
+                ],
+                "intermediário": [
+                    "Machine Learning Básico",
+                    "Deep Learning Fundamentos",
+                    "Big Data Analytics",
+                    "Feature Engineering"
+                ],
+                "avançado": [
+                    "MLOps",
+                    "Análise Avançada",
+                    "IA Generativa",
+                    "Pesquisa Aplicada"
+                ]
+            },
+            "design": {
+                "iniciante": [
+                    "Teoria das Cores",
+                    "Tipografia",
+                    "UI Básica",
+                    "Ferramentas de Design"
+                ],
+                "intermediário": [
+                    "Design Systems",
+                    "UX Research",
+                    "Prototipagem",
+                    "Design Visual"
+                ],
+                "avançado": [
+                    "Design Leadership",
+                    "Design Estratégico",
+                    "Design para Produto",
+                    "Design Thinking Avançado"
+                ]
+            }
+        }
+
+        # Usa módulos padrão se a área não estiver mapeada
+        if area not in modulos_por_area:
+            modulos_por_area[area] = {
+                "iniciante": [
+                    "Fundamentos Básicos",
+                    "Introdução à Área",
+                    "Ferramentas Essenciais",
+                    "Práticas Iniciais"
+                ],
+                "intermediário": [
+                    "Conceitos Avançados",
+                    "Técnicas Especializadas",
+                    "Projetos Práticos",
+                    "Metodologias"
+                ],
+                "avançado": [
+                    "Especialização Profissional",
+                    "Gestão de Projetos",
+                    "Inovação na Área",
+                    "Liderança Técnica"
+                ]
+            }
+
+        duracao_meses = int(duracao.split()[0])
+        tempo_por_modulo = duracao_meses / 4
+
+        template = f"""
+        🎯 Objetivos Identificados:
+        {self._formatar_objetivos(palavras_chave)}
+        
+        📚 Trilha Personalizada para {area.title()} - Nível {nivel}:
+        {self._formatar_modulos(modulos_por_area[area][nivel], tempo_por_modulo)}
+        
+        🚀 Projetos Práticos Sugeridos:
+        {self._gerar_projetos(area, nivel, palavras_chave)}
+        
+        📅 Cronograma ({duracao}):
+        {self._gerar_cronograma(duracao_meses, modulos_por_area[area][nivel])}
+        
+        🎖️ Marcos de Avaliação:
+        {self._gerar_marcos_avaliacao(nivel)}
+        """
+        
+        return template
+
+    def _extrair_palavras_chave(self, texto: str) -> list:
+        tokens = word_tokenize(texto.lower())
+        palavras = [palavra for palavra in tokens 
+                   if palavra.isalnum() and 
+                   palavra not in self.stop_words]
+        return [palavra[0] for palavra in Counter(palavras).most_common(5)]
+
+    def _formatar_objetivos(self, palavras_chave: list) -> str:
+        objetivos = {
+            "programação": [
+                "Desenvolver habilidades técnicas em {}",
+                "Criar projetos práticos usando {}",
+                "Dominar conceitos de {}"
+            ],
+            "data science": [
+                "Analisar dados usando {}",
+                "Construir modelos de {}",
+                "Implementar soluções com {}"
+            ],
+            "design": [
+                "Criar interfaces usando {}",
+                "Desenvolver projetos de {}",
+                "Aplicar princípios de {}"
+            ]
+        }
+        
+        return "\n".join([
+            f"• {objetivo.format(palavra)}"
+            for palavra, objetivo in zip(palavras_chave, objetivos.get("geral", objetivos["programação"]))
+        ])
+
+    def _formatar_modulos(self, modulos: list, tempo: float) -> str:
+        return "\n".join([
+            f"Módulo {i+1}: {modulo} ({tempo:.1f} meses)"
+            for i, modulo in enumerate(modulos)
+        ])
+
+    def _gerar_projetos(self, area: str, nivel: str, palavras_chave: list) -> str:
+        projetos = {
+            "iniciante": [
+                "Projeto Tutorial Guiado",
+                "Mini-Projeto Prático",
+                "Exercícios Fundamentais"
+            ],
+            "intermediário": [
+                "Projeto Individual",
+                "Projeto em Equipe",
+                "Case Study Prático"
+            ],
+            "avançado": [
+                "Projeto Complexo",
+                "Contribuição Open Source",
+                "Projeto de Pesquisa"
+            ]
+        }
+        
+        return "\n".join([
+            f"• {projeto}" for projeto in projetos[nivel]
+        ])
+
+    def _gerar_cronograma(self, duracao: int, modulos: list) -> str:
+        meses_por_modulo = duracao / len(modulos)
+        cronograma = []
+        
+        for i, modulo in enumerate(modulos):
+            inicio = i * meses_por_modulo + 1
+            fim = (i + 1) * meses_por_modulo
+            cronograma.append(
+                f"Mês {inicio:.1f}-{fim:.1f}: {modulo}"
+            )
+        
+        return "\n".join(cronograma)
+
+    def _gerar_marcos_avaliacao(self, nivel: str) -> str:
+        marcos = {
+            "iniciante": [
+                "Quiz de Conceitos Básicos",
+                "Exercícios Práticos",
+                "Projeto Final Básico"
+            ],
+            "intermediário": [
+                "Avaliação Técnica",
+                "Projeto Individual",
+                "Apresentação de Resultados"
+            ],
+            "avançado": [
+                "Defesa de Projeto",
+                "Contribuição Técnica",
+                "Artigo/Publicação"
+            ]
+        }
+        
+        return "\n".join([
+            f"• {marco}" for marco in marcos[nivel]
+        ])
+
     def _formatar_historico(self) -> str:
         if not self.historico:
             return "Nenhuma trilha gerada ainda"
         
         historico_fmt = "📋 Histórico de Trilhas:\n\n"
-        for h in self.historico[-5:]:  # Últimas 5 trilhas
+        for h in self.historico[-5:]:
             historico_fmt += f"• {h['data']} - {h['nome']} ({h['nivel']}, {h['area']})\n"
         return historico_fmt
 
@@ -179,7 +372,6 @@ def criar_interface():
                 analise = gr.Textbox(label="Sua Trilha de Aprendizado", lines=10)
                 historico = gr.Markdown(label="Histórico")
         
-        # Botão de ajuda
         with gr.Accordion("ℹ️ Como usar"):
             gr.Markdown("""
             1. Grave um áudio descrevendo seus objetivos de aprendizado