Update app.py

app.py

@@ -2,7 +2,13 @@ import os
 import gradio as gr
 from crewai import Task, Agent, LLM
 from youtube import pesquisar_videos_youtube
-import time  # Para simular a progressão
+from wikipedia import pesquisar_wikipedia
+import time
+import requests
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.styles import getSampleStyleSheet
+import io
 
 GROQ_API_KEY_01 = os.getenv("GROQ_API_KEY_01")
 groqllm = LLM(
@@ -20,14 +26,17 @@ def processar_topicos(topicos_str):
     return [t.strip() for t in topicos_str.split(',') if t.strip()]
 
 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"
 
+    # Busca no YouTube
     yield "Buscando vídeos no YouTube...", gr.update(value=10)
     entradaYoutube = pesquisar_videos_youtube(solicitacao)
 
+    # Busca na Wikipedia
+    yield "Buscando artigos na Wikipedia...", gr.update(value=20)
+    entradaWikipedia = pesquisar_wikipedia(topicos)
+
     # Motivação
     yield "Criando mensagem motivacional...", gr.update(value=30)
     agentMotivador = Agent(
@@ -38,9 +47,8 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         verbose=True
     )
     taskMotivador = Task(
-        description = (
+        description=(
             "Escreva uma mensagem motivacional para o estudante, formatada em Markdown.\n\n"
-            "A mensagem deve conter:\n"
             "## Mensagem Motivacional 🎯\n\n"
             "- Um parágrafo inicial incentivando o estudante a seguir seus estudos.\n"
             "- Frases inspiradoras para manter o foco e a disciplina.\n"
@@ -62,7 +70,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         verbose=True
     )
     taskGuia = Task(
-        description = (
+        description=(
             f"Crie um Guia de Estudos para {disciplina}, abordando {assunto} e os tópicos {topicos}. "
             "O guia deve seguir a seguinte estrutura e formatação:\n\n"
             "## Guia de Estudos: {disciplina}\n\n"
@@ -97,7 +105,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         verbose=True
     )
     taskPlano = Task(
-        description = (
+        description=(
             f"Crie um Plano de Estudos para {disciplina}, cobrindo {assunto} e os tópicos {topicos}. "
             f"O aluno tem {horas} horas por dia e {dias} dias para estudar.\n\n"
             "O plano deve seguir esta estrutura e formatação:\n\n"
@@ -141,7 +149,7 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
         verbose=True
     )
     taskYoutube = Task(
-        description = (
+        description=(
             f"Lista do Youtube: {entradaYoutube} "
             "Você receberá uma lista de vídeos extraída da API do YouTube. Sua tarefa é classificar e organizar os vídeos "
             "por categorias, formatando-os em Markdown. As categorias devem ser baseadas no título do vídeo.\n\n"
@@ -161,26 +169,54 @@ def executar_equipe_interface(disciplina, assunto, topicos_str, horas, dias):
     )
     saidaYoutube = agentYoutube.execute_task(taskYoutube)
 
+    # Curadoria de Artigos da Wikipedia
+    yield "Organizando artigos da Wikipedia...", gr.update(value=95)
+    agentWikipedia = Agent(
+        role='Especialista em Curadoria de Artigos Educacionais',
+        goal='Organizar e formatar artigos educacionais encontrados na Wikipedia para aprendizado eficiente.',
+        backstory='Você é um especialista em curadoria de materiais educacionais, com experiência na seleção de artigos para ensino.',
+        llm=groqllm2,
+        verbose=True
+    )
+    taskWikipedia = Task(
+        description=(
+            f"Lista de artigos da Wikipedia: {entradaWikipedia} "
+            "Você receberá uma lista de artigos extraída da API da Wikipedia. Sua tarefa é organizar os artigos "
+            "em Markdown, formatando-os de forma clara e educacional.\n\n"
+            f"## Artigos sobre {assunto}\n\n"
+            "### Formato de saída\n"
+            "- Para cada artigo, a saída deve seguir o formato abaixo:\n"
+            "  **[Título](URL)**\n\n  _Trecho_\n\n"
+            "- Se um artigo não tiver trecho, substituir por '(Sem trecho disponível)'.\n"
+            "- Se houver mais de um artigo, repetir a estrutura para cada um.\n"
+            "- Certifique-se de que a formatação Markdown esteja correta e bem organizada."            
+            "Caso não tenha encontrado artigos relevantes, a saída deverá ser:\n\n"
+            "## Não foram encontrados artigos sobre o assunto\n\n"
+            "Não busque que não tenham sido enviado na lista de artigos."
+
+        ),
+        agent=agentWikipedia,
+        expected_output="Lista de artigos organizados em Markdown."
+    )
+    saidaWikipedia = agentWikipedia.execute_task(taskWikipedia)
+
     yield "Processo concluído!", gr.update(value=100)
 
     saidaCompleta = f"""
 # 🎯 Motivação
 {saidaMotivador}
-
 ---
-
 # 📖 Guia de Estudos
 {saidaGuia}
-
 ---
-
 # 📅 Plano de Estudos
 {saidaPlano}
-
 ---
-
 # 🎥 Vídeos Educacionais
 {saidaYoutube}
+---
+# 📜 Artigos da Wikipedia
+{saidaWikipedia}
 """
 
     yield saidaCompleta, gr.update(value=100)