Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 import os
 from groq import Groq
 
-def consultar_llm(disciplina, perfil):
+def consultar_llm(estudante, disciplina, perfil):
     # Constrói o prompt personalizado utilizando os valores informados
-    prompt = f"Considere a disciplina '{disciplina}' e o perfil do estudante '{perfil}'. " \
-             "Forneça um plano de estudos baseadas nesses dados."
+    prompt = f"Atue com umo professor. Considere a disciplina '{disciplina}' e o perfil do estudante '{perfil}'. Forneça um plano de estudos baseadas nesses dados. Adicione uma nova linha abaixo do título com o nome do estudante: '{estudante}'."
 
     client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
@@ -24,26 +23,39 @@ def consultar_llm(disciplina, perfil):
 
 # Criação da interface com Gradio utilizando Blocks para organizar os componentes
 with gr.Blocks() as demo:
+    # Adiciona um título
+    gr.Markdown("# Gerador de planos de estudos personalizados")
+    gr.Markdown("## Visão do professor")
+    
     with gr.Row():
         # Coluna da esquerda: inputs
         with gr.Column():
-            disciplina_input = gr.Textbox(label="Disciplina", placeholder="Digite o nome da disciplina", value="Introdução a programação")
+            estudante_combo = gr.Dropdown(
+                choices=["Jose", "Maria", "João"],
+                label="Nome do estudante",
+                value="Jose"  
+            )
+            disciplina_combo = gr.Dropdown(
+                choices=["Introdução a programação", "Estrutura de dados", "Algoritmos avançados"],
+                label="O nome da disciplina",
+                value="Introdução a programação"  
+            )
             perfil_combo = gr.Dropdown(
                 choices=["Iniciante", "Intermediário", "Avançado"],
                 label="Perfil do Estudante",
-                value="Iniciante"  # valor padrão
+                value="Iniciante"  
             )
             consultar_btn = gr.Button("Gerar plano de estudo personalizado")
-        # Coluna da direita: área de saída
+        
         with gr.Column():
             with gr.Row():
-                aceitar_btn = gr.Button("Aceitar") 
-                rejeitar_btn = gr.Button("Rejeitar")
+                aceitar_btn = gr.Button("Aceitar/Enviar") 
+                rejeitar_btn = gr.Button("Rejeitar/Alterar")
                 download_btn = gr.Button("Download PDF")
             resultado = gr.Markdown(label="Plano de estudo")
             
     # Configura o botão para chamar a função 'consultar_llm'
-    consultar_btn.click(fn=consultar_llm, inputs=[disciplina_input, perfil_combo], outputs=resultado)
+    consultar_btn.click(fn=consultar_llm, inputs=[estudante_combo, disciplina_combo, perfil_combo], outputs=resultado)
 
 # Inicia a interface web
 demo.launch()