Update app.py

app.py

@@ -64,7 +64,7 @@ def pesquisar_videos_youtube(consulta, num_resultados=5):
 
 ######### YOUTUBE FIM #########
 
-######### CREWA AI INICIO #########
+######### CREWAAI INICIO #########
 
 deepseek_api_key = os.getenv("DEEPSEEK_API_KEY")
 
@@ -82,14 +82,9 @@ def processar_topicos(topicos_str):
 def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
     
     topicos = processar_topicos(topicos_str)
-
     solicitacao = f"Disciplina: {disciplina}\nAssunto: {assunto}\nTópicos: {topicos}\n"
-
-    yield "Buscando vídeos no YouTube...", gr.update(value=10)
     entradaYoutube = pesquisar_videos_youtube(solicitacao)
-
-    # Motivação
-    yield "Criando mensagem motivacional...", gr.update(value=30)
+    
     agentMotivador = Agent(
         role='Motivador',
         goal='Escrever uma mensagem motivacional para o estudante.',
@@ -97,6 +92,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         llm=modelo,
         verbose=False
     )
+
     taskMotivador = Task(
         description = (
             "Escreva uma mensagem motivacional para o estudante, formatada em Markdown.\n\n"
@@ -110,10 +106,9 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         agent=agentMotivador,
         expected_output='Mensagem motivacional em markdown.'
     )
+
     saidaMotivador = agentMotivador.execute_task(taskMotivador)
 
-    # Guia de Estudos
-    yield "Gerando guia de estudos...", gr.update(value=50)
     agentGuia = Agent(
         role="Especialista em Guia de Estudos",
         goal="Criar um guia de estudos estruturado, explicativo e didático sobre um determinado assunto.",
@@ -121,6 +116,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         llm=modelo,
         verbose=False
     )
+
     taskGuia = Task(
         description = (
             f"Crie um Guia de Estudos para {disciplina}, abordando {assunto} e os tópicos {topicos}. "
@@ -145,10 +141,9 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         agent=agentGuia,
         expected_output='Guia de estudos em markdown'
     )
+
     saidaGuia = agentGuia.execute_task(taskGuia)
 
-    # Plano de Estudos
-    yield "Criando plano de estudos...", gr.update(value=70)
     agentPlano = Agent(
         role="Especialista em Plano de Estudos",
         goal="Criar um plano de estudos eficiente para que o aluno aprenda de maneira organizada.",
@@ -189,10 +184,9 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         agent=agentPlano,
         expected_output='Plano de estudos estruturado em markdown'
     )
+
     saidaPlano = agentPlano.execute_task(taskPlano)
 
-    # Curadoria de Vídeos
-    yield "Organizando vídeos do YouTube...", gr.update(value=90)
     agentYoutube = Agent(
         role='Especialista em Curadoria de Vídeos Educacionais',
         goal='Organizar e formatar vídeos educacionais encontrados no YouTube para aprendizado eficiente.',
@@ -200,6 +194,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         llm=modelo,
         verbose=False
     )
+
     taskYoutube = Task(
         description = (
             f"Lista do Youtube: {entradaYoutube} "
@@ -219,37 +214,42 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         agent=agentYoutube,
         expected_output="Lista de vídeos organizados em Markdown."
     )
+
     saidaYoutube = agentYoutube.execute_task(taskYoutube)
 
-    yield "Processo concluído!", gr.update(value=100)
+    saidaCompleta = """
+# Material de estudo    
+
+## 🎯 Motivação
 
-    saidaCompleta = f"""
-# 🎯 Motivação
-{saidaMotivador}
+{}
 
 ---
 
-# 📖 Guia de Estudos
-{saidaGuia}
+## 📖 Guia de Estudos
+
+{}
 
 ---
 
-# 📅 Plano de Estudos
-{saidaPlano}
+## 📅 Plano de Estudos
+
+{}
 
 ---
 
-# 🎥 Vídeos Educacionais
-{saidaYoutube}
-"""
+## 🎥 Vídeos Educacionais
+
+{}
+""".format(saidaMotivador, saidaGuia, saidaPlano, saidaYoutube)
 
-    yield saidaCompleta, gr.update(value=100)
+    return saidaCompleta
 
-######### CREWA AI FIM #########
+######### CREWAAI FIM #########
 
 with gr.Blocks() as demo:
     gr.Markdown("# 📚 Gerador de Material de Estudos")
-    gr.Markdown("## Ajuste do original do [Kleber Galvão](https://huggingface.co/spaces/GalvaoFilho/KMData) ")
+    gr.Markdown("## Pequeno ajuste no original do [Kleber Galvão](https://huggingface.co/spaces/GalvaoFilho/KMData) ")
     with gr.Row():
         with gr.Column():
             disciplina = gr.Textbox(label="Disciplina", value="Matemática")
@@ -258,12 +258,11 @@ with gr.Blocks() as demo:
             horas = gr.Textbox(label="Tempo diário", value="2 horas")
             dias = gr.Textbox(label="Quantos dias", value="5 dias")
             gerar_button = gr.Button("Gerar Material")
-            progress = gr.Slider(minimum=0, maximum=100, step=1, value=0, label="Progresso", interactive=False)
         with gr.Column():
             resultado = gr.Markdown(label="Material Completo (Markdown)")
 
     gerar_button.click(fn=executar_equipe_interface,
                        inputs=[disciplina, assunto, topicos_str, horas, dias],
-                       outputs=[resultado, progress])
+                       outputs=[resultado])
 
 demo.launch()