Update app.py

app.py

@@ -5,12 +5,11 @@ from transformers import pipeline
 class GeradorTrilhaAprendizado:
     def __init__(self):
         self.device = 0 if torch.cuda.is_available() else -1
-        # Inicializa modelos
         self.transcriber = pipeline("automatic-speech-recognition", 
                                   model="openai/whisper-base",
                                   device=self.device)
         self.generator = pipeline("text-generation", 
-                                model="gpt2",
+                                model="gpt2-large",  # Modelo maior
                                 device=self.device)
 
     def processar_audio(self, 
@@ -19,54 +18,76 @@ class GeradorTrilhaAprendizado:
                        nivel: str = "intermediário",
                        incluir_recursos: bool = True) -> tuple:
         try:
-            # Transcrição do áudio
             transcricao = self.transcriber(audio_path)["text"]
             
-            # Geração da trilha
             prompt = f"""
-            Com base no seguinte texto, crie uma trilha de aprendizado detalhada 
-            para nível {nivel}:
+            Crie uma trilha de aprendizado estruturada para nível {nivel} 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
             
-            Trilha de aprendizado:
+            Trilha de aprendizado para {nome_trilha}:
             """
             
             analise = self.generator(prompt, 
-                                   max_length=300, 
-                                   num_return_sequences=1)[0]["generated_text"]
+                                   max_length=500,  # Aumentado
+                                   num_return_sequences=1,
+                                   temperature=0.7,  # Adicionado
+                                   top_p=0.95)[0]["generated_text"]  # Adicionado
             
             if incluir_recursos:
-                recursos = self._gerar_recursos()
+                recursos = self._gerar_recursos(nivel, transcricao)
                 analise += "\n\n" + recursos
 
-            return (
-                transcricao,
-                analise,
-                "✅ Trilha gerada com sucesso!"
-            )
+            return (transcricao, analise, "✅ Trilha gerada com sucesso!")
         except Exception as e:
-            return (
-                f"Erro: {str(e)}",
-                "Não foi possível gerar a análise.",
-                "❌ Ocorreu um erro"
-            )
+            return (f"Erro: {str(e)}", "Não foi possível gerar a análise.", "❌ Ocorreu um erro")
+
+    def _gerar_recursos(self, nivel: str, objetivo: str) -> str:
+        recursos_por_nivel = {
+            "iniciante": {
+                "cursos": ["Fundamentos Básicos", "Introdução Prática"],
+                "livros": ["Guia do Iniciante", "Primeiros Passos"],
+                "projetos": ["Projeto inicial guiado", "Mini-projetos práticos"]
+            },
+            "intermediário": {
+                "cursos": ["Especialização Prática", "Técnicas Avançadas"],
+                "livros": ["Guia Completo", "Estudos de Caso"],
+                "projetos": ["Projetos de média complexidade", "Desafios práticos"]
+            },
+            "avançado": {
+                "cursos": ["Masterclass Avançada", "Especialização Profissional"],
+                "livros": ["Técnicas Avançadas", "Estado da Arte"],
+                "projetos": ["Projetos complexos", "Contribuições open-source"]
+            }
+        }
 
-    def _gerar_recursos(self) -> str:
-        return """
-        📚 Recursos Recomendados:
+        recursos = recursos_por_nivel[nivel]
+        return f"""
+        📚 Recursos Recomendados para Nível {nivel}:
         
-        1. Livros:
-           - "Guia Essencial"
-           - "Técnicas Avançadas"
+        1. Cursos Recomendados:
+           - {recursos['cursos'][0]}
+           - {recursos['cursos'][1]}
+           
+        2. Material de Estudo:
+           - {recursos['livros'][0]}
+           - {recursos['livros'][1]}
            
-        2. Cursos Online:
-           - Coursera: "Especialização no Tema"
-           - edX: "Curso Avançado"
+        3. Projetos Práticos:
+           - {recursos['projetos'][0]}
+           - {recursos['projetos'][1]}
            
-        3. Recursos Práticos:
-           - Tutoriais interativos
-           - Exercícios práticos
-           - Projetos do mundo real
+        4. Recursos Adicionais:
+           - Comunidades de prática
+           - Mentoria entre pares
+           - Workshops práticos
+           - Avaliações periódicas
         """
 
 def criar_interface():