import gradio as gr

# Função para gerar o prompt baseado na área escolhida
def generate_prompt(area, details):
    if area == "Mídia e Conteúdo Digital":
        return f"Estou planejando {details}. Pode sugerir três ângulos criativos ou ideias de subtemas para explorar?"
    elif area == "Escritório e Produtividade":
        return f"Preciso organizar uma reunião sobre {details}. Pode sugerir uma agenda clara com três pontos principais a serem abordados?"
    elif area == "Fotografia e Design":
        return f"Estou organizando uma sessão fotográfica com o tema {details}. Pode sugerir três ideias criativas de poses ou ambientes para explorar?"
    elif area == "Advocacia":
        return f"Estou pesquisando sobre {details}. Pode me ajudar a estruturar a pesquisa com fontes confiáveis e os principais pontos a abordar?"
    elif area == "Finanças e Contabilidade":
        return f"Preciso de ajuda para organizar um plano financeiro para {details}. Pode sugerir três estratégias principais a serem consideradas?"
    elif area == "Educação e Treinamento":
        return f"Estou criando um plano de aula sobre {details}. Pode sugerir uma estrutura de três tópicos principais para abordar e uma atividade prática?"
    else:
        return "Área não reconhecida."

# Interface do Gradio
def interface(area, details):
    prompt = generate_prompt(area, details)
    return prompt

# Definindo as opções de áreas para seleção
areas = [
    "Mídia e Conteúdo Digital",
    "Escritório e Produtividade",
    "Fotografia e Design",
    "Advocacia",
    "Finanças e Contabilidade",
    "Educação e Treinamento"
]

# Configurando o Gradio
with gr.Blocks() as demo:
    gr.Markdown("# Gerador de Prompts Inteligentes")
    area = gr.Dropdown(choices=areas, label="Escolha a área profissional")
    details = gr.Textbox(label="Insira detalhes sobre a área escolhida (ex: tema, situação)")
    output = gr.Textbox(label="Seu Prompt Gerado")
    submit = gr.Button("Gerar Prompt")
    
    submit.click(interface, inputs=[area, details], outputs=output)

# Rodando o app no Hugging Face Spaces
if __name__ == "__main__":
    demo.launch()