import os import gradio as gr from crewai import Task, Agent, LLM from youtube import pesquisar_videos_youtube from wikipedia import pesquisar_wikipedia import time import requests import io import tempfile import re from reportlab.lib.pagesizes import letter from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer from reportlab.lib.styles import getSampleStyleSheet from reportlab.lib.units import inch from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase.ttfonts import TTFont GROQ_API_KEY_01 = os.getenv("GROQ_API_KEY_01") groqllm = LLM( model="groq/llama-3.3-70b-versatile", api_key=GROQ_API_KEY_01 ) GROQ_API_KEY_02 = os.getenv("GROQ_API_KEY_02") groqllm2 = LLM( model="groq/llama-3.3-70b-versatile", api_key=GROQ_API_KEY_02 ) def processar_topicos(topicos_str): """ Converte uma string de tópicos separada por vírgulas em uma lista tratada. """ return [t.strip() for t in topicos_str.split(',') if t.strip()] def generate_pdf(markdown_content): """Converte conteúdo Markdown para PDF usando reportlab, suportando caracteres UTF-8.""" pdf_buffer = io.BytesIO() doc = SimpleDocTemplate(pdf_buffer, pagesize=letter, rightMargin=inch, leftMargin=inch, topMargin=inch, bottomMargin=inch) styles = getSampleStyleSheet() # Ajustar estilos para suportar UTF-8 h1_style = styles['Heading1'] h2_style = styles['Heading2'] h3_style = styles['Heading3'] body_style = styles['BodyText'] bullet_style = styles['BodyText'] bullet_style.leftIndent = 20 # Função para sanitizar texto, preservando acentos e cedilhas def sanitize_text(text): # Substituir links Markdown [texto](url) por texto (url) text = re.sub(r'\[([^\]]+)\]\(([^\)]+)\)', r'\1 (\2)', text) # Remover tags HTML text = re.sub(r'<[^>]+>', '', text) # Escapar caracteres que podem parecer tags XML text = text.replace('<', '<').replace('>', '>') # Substituir emojis por texto descritivo emoji_map = { '🎯': '[Alvo] ', '📖': '[Livro] ', '📅': '[Calendário] ', '🎥': '[Vídeo] ', '📜': '[Documento] ' } for emoji, replacement in emoji_map.items(): text = text.replace(emoji, replacement) return text story = [] lines = markdown_content.split('\n') for line in lines: line = line.strip() if not line: story.append(Spacer(1, 0.2 * inch)) continue if line.startswith('# '): text = sanitize_text(line[2:]) story.append(Paragraph(text, h1_style)) elif line.startswith('## '): text = sanitize_text(line[3:]) story.append(Paragraph(text, h2_style)) elif line.startswith('### '): text = sanitize_text(line[4:]) story.append(Paragraph(text, h3_style)) elif line.startswith('- '): text = sanitize_text(line[2:]) story.append(Paragraph(f"• {text}", bullet_style)) elif line.startswith('---'): story.append(Spacer(1, 0.3 * inch)) else: text = sanitize_text(line) story.append(Paragraph(text, body_style)) doc.build(story) pdf_buffer.seek(0) with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_pdf: temp_pdf.write(pdf_buffer.read()) temp_pdf_path = temp_pdf.name return temp_pdf_path def generate_markdown_file(markdown_content): """Gera um arquivo Markdown a partir do conteúdo.""" with tempfile.NamedTemporaryFile(delete=False, suffix='.md', mode='w', encoding='utf-8') as temp_md: temp_md.write(markdown_content) temp_md_path = temp_md.name return temp_md_path 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), None, None entradaYoutube = pesquisar_videos_youtube(solicitacao) # Busca na Wikipedia yield "Buscando artigos na Wikipedia...", gr.update(value=20), None, None entradaWikipedia = pesquisar_wikipedia(topicos) # Motivação yield "Criando mensagem motivacional...", gr.update(value=30), None, None agentMotivador = Agent( role='Motivador', goal='Escrever uma mensagem motivacional para o estudante.', backstory='Você é um coach motivacional com experiência em ajudar estudantes a manterem o foco.', llm=groqllm, verbose=True ) taskMotivador = Task( 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" "- Um fechamento encorajador, reforçando a importância do esforço e da dedicação.\n\n" "O texto deve ser positivo, motivador e adequado para estudantes de ensino médio." ), 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), None, None agentGuia = Agent( role="Especialista em Guia de Estudos", goal="Criar um guia de estudos estruturado, explicativo e didático sobre um determinado assunto.", backstory="Você é um especialista em educação, com experiência na criação de guias de estudo detalhados.", llm=groqllm, verbose=True ) taskGuia = Task( 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" "### Introdução\n" "- Texto introdutório justificado sobre o tema, destacando sua importância e contexto.\n\n" "### Conceitos Fundamentais\n" "- Explicação detalhada dos principais conceitos abordados, com exemplos práticos.\n" "- Utilize listas desordenadas para estruturar os conceitos.\n\n" "### Aplicações Práticas\n" "- Explique como o tema se aplica no mundo real.\n" "- Utilize exemplos concretos e listas para organizar as aplicações.\n\n" "### Técnicas de Aprendizado e Dificuldades Comuns\n" "- Apresente métodos eficazes para aprender o tema.\n" "- Liste dificuldades comuns dos alunos e estratégias para superá-las.\n\n" "### Indicação de Materiais Gratuitos\n" "- Forneça sugestões de livros, vídeos e artigos gratuitos sobre o tema.\n" "- Apresente os materiais em formato de lista com títulos e links quando possível.\n\n" "O conteúdo deve ser didático, acessível para alunos do ensino médio e utilizar texto justificado sempre que possível." ), 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), None, None agentPlano = Agent( role="Especialista em Plano de Estudos", goal="Criar um plano de estudos eficiente para que o aluno aprenda de maneira organizada.", backstory="Você é um planejador educacional especialista em cronogramas de estudo eficientes.", llm=groqllm2, verbose=True ) taskPlano = Task( 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" "## Plano de Estudos: {disciplina}\n\n" "### Introdução\n" "- Apresentação do objetivo do plano de estudos.\n" "- Importância da organização para otimizar o aprendizado.\n\n" "### Distribuição Equilibrada dos Tópicos\n" "- Divisão dos conteúdos de forma proporcional ao tempo disponível.\n" "- Sugerir uma agenda diária/semanal equilibrada.\n\n" "### Técnicas Ativas de Aprendizado\n" "- Explicação de estratégias eficazes para o estudo, incluindo:\n" " - Resumos\n" " - Flashcards\n" " - Mapas mentais\n" " - Resolução de exercícios\n\n" "### Revisões Programadas\n" "- Definir períodos estratégicos para revisão de conteúdos.\n" "- Sugestão de técnicas como repetição espaçada e autoavaliação.\n\n" "### Monitoramento do Progresso\n" "- Métodos para acompanhar a evolução do estudo.\n" "- Uso de checklists ou aplicativos para organização.\n\n" "### Sugestões para Pausas e Evitar Sobrecarga Mental\n" "- Importância das pausas regulares para manter a produtividade.\n" "- Sugestão de técnicas como a Técnica Pomodoro.\n" "- Dicas para manter o bem-estar mental durante os estudos.\n\n" "O plano deve ser didático, bem estruturado e adaptável para alunos do ensino médio." ), 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), None, None agentYoutube = Agent( role='Especialista em Curadoria de Vídeos Educacionais', goal='Organizar e formatar vídeos educacionais encontrados no YouTube para aprendizado eficiente.', backstory='Você é um especialista em curadoria de materiais educacionais, com experiência na seleção de vídeos para ensino.', llm=groqllm2, verbose=True ) taskYoutube = Task( 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" f"## Vídeos sobre {assunto}\n\n" "### Formato de saída\n" "- Para cada vídeo, a saída deve seguir o formato abaixo:\n" " **[Título](URL)**\n\n _Descrição_\n\n" "- Se um vídeo não tiver descrição, substituir por '(Sem descrição disponível)'.\n" "- Se houver mais de um vídeo, repetir a estrutura para cada um.\n" "- Certifique-se de que a formatação Markdown esteja correta e bem organizada." "Caso um dos videos seja esse: https://www.youtube.com/watch?v=dQw4w9WgXcQ" "a saida deverá ser:" "##Não foram encontrados vídeos sobre o assunto" ), agent=agentYoutube, expected_output="Lista de vídeos organizados em Markdown." ) saidaYoutube = agentYoutube.execute_task(taskYoutube) # Curadoria de Artigos da Wikipedia yield "Organizando artigos da Wikipedia...", gr.update(value=95), None, None 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" "- For each article, the output should follow the format below:\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), None, None saidaCompleta = f""" # 🎯 Motivação {saidaMotivador} --- # 📖 Guia de Estudos {saidaGuia} --- # 📅 Plano de Estudos {saidaPlano} --- # 🎥 Vídeos Educacionais {saidaYoutube} --- # 📜 Artigos da Wikipedia {saidaWikipedia} """ # Gerar arquivos para download pdf_file = generate_pdf(saidaCompleta) markdown_file = generate_markdown_file(saidaCompleta) yield saidaCompleta, gr.update(value=100), pdf_file, markdown_file # Interface Gradio with gr.Blocks() as demo: gr.Markdown("# 📚 Gerador de Material de Estudos") with gr.Row(): with gr.Column(): disciplina = gr.Textbox(label="Disciplina", value="Matemática") assunto = gr.Textbox(label="Assunto", value="Funções") topicos_str = gr.Textbox(label="Tópicos", value="Função quadrática, Função exponencial, Função logarítmica") 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)") with gr.Row(): download_pdf = gr.File(label="Baixar PDF", visible=False) download_markdown = gr.File(label="Baixar Markdown", visible=False) def update_download_buttons(pdf_file, markdown_file): return ( gr.update(value=pdf_file, visible=True, label="Baixar PDF", file_types=[".pdf"]), gr.update(value=markdown_file, visible=True, label="Baixar Markdown", file_types=[".md"]) ) gerar_button.click( fn=executar_equipe_interface, inputs=[disciplina, assunto, topicos_str, horas, dias], outputs=[resultado, progress, download_pdf, download_markdown] ).then( fn=update_download_buttons, inputs=[download_pdf, download_markdown], outputs=[download_pdf, download_markdown] ) demo.launch()