from equipe import formarEquipe
import gradio as gr

def executar_equipe(solicitacao):
    equipe = formarEquipe(solicitacao) 
    resultado = equipe.kickoff()
    return resultado

def gradio_interface(disciplina, assunto, topicos):
    solicitacao = {
        "disciplina": disciplina,
        "assunto": assunto,
        "topicos": topicos.split(", ")
    }
    return executar_equipe(solicitacao)

if __name__ == "__main__":
    iface = gr.Interface(
        fn=gradio_interface,
        inputs=[
            gr.Textbox(label="Disciplina"),
            gr.Textbox(label="Assunto"),
            gr.Textbox(label="Tópicos (separados por vírgula)")
        ],
        outputs=gr.Textbox(label="Plano de estudo"),
        title="Plano de estudos com vídeo e motivacional",
        description="Criador de plano de estudos com Mensagem motivacional e Lista de vídeos com três agentes (falta adicionar o recurso dos vídeos).",
        examples=[
            ["Matemática", "Funções", "Função quadrática, Função exponencial, Função logarítmica"],
            ["História", "Revolução Francesa", "Causas, Eventos principais, Consequências"],
            ["Biologia", "Genética", "DNA, RNA, Hereditariedade"]
        ],
        flagging_options=[], 
        css="footer{display:none !important}",
        theme="soft"  # Adicionando um tema
    )
    iface.launch()