import gradio as gr
from transformers import pipeline

# Usa o modelo Falcon 1B mais leve, já disponível publicamente
generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")

def gerar_prompt(nicho, sensacao, cor, estilo, beneficio, promessa, publico, provas, cta):
    prompt_base = f"""
Quero criar um site para o nicho: {nicho}.
A sensação que o visual deve transmitir é: {sensacao}.
A cor principal do site deve ser: {cor}.
O estilo visual preferido é: {estilo}.
O maior benefício do produto ou serviço é: {beneficio}.
A promessa principal é: {promessa}.
O público-alvo do site são: {publico}.
Desejo incluir provas sociais ou autoridade? {provas}.
A chamada para ação (CTA) é: {cta}.

Agora, gere o prompt perfeito para o DeepSite criar esse site automaticamente, incluindo instruções de identidade visual e cópia para cada seção.
"""
    result = generator(prompt_base, max_length=400, do_sample=True, temperature=0.7)
    return result[0]["generated_text"]

with gr.Blocks() as demo:
    gr.Markdown("### ✨ Agente Raypages: Gere o prompt ideal para seu site IA")

    nicho = gr.Textbox(label="Qual o nicho do seu site?")
    sensacao = gr.Textbox(label="Que sensação o site deve transmitir?")
    cor = gr.Textbox(label="Cor principal do site")
    estilo = gr.Textbox(label="Estilo visual preferido")
    beneficio = gr.Textbox(label="Maior benefício do produto ou serviço")
    promessa = gr.Textbox(label="Promessa principal")
    publico = gr.Textbox(label="Público-alvo do site")
    provas = gr.Textbox(label="Deseja adicionar provas sociais ou autoridade?")
    cta = gr.Textbox(label="Chamada para ação (CTA)")

    btn = gr.Button("Gerar Prompt")
    output = gr.Textbox(label="Prompt Gerado")

    btn.click(gerar_prompt, inputs=[nicho, sensacao, cor, estilo, beneficio, promessa, publico, provas, cta], outputs=output)

demo.launch()