Update app.py

app.py

@@ -1,49 +1,42 @@
 import gradio as gr
 from transformers import pipeline
 
-# Carrega o modelo Falcon RW 1B
+# 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):
-    entrada = f"""
-Crie um prompt para gerar um site com IA para o seguinte negócio:
-
-Nicho: {nicho}
-Sensação desejada ao visitar o site: {sensacao}
-Cores preferidas: {cor}
-Estilo visual (moderno, divertido, minimalista etc): {estilo}
-Benefício principal: {beneficio}
-Promessa principal: {promessa}
-Público-alvo: {publico}
-Provas sociais ou de autoridade (opcional): {provas}
-Chamada para ação (CTA): {cta}
-
-O prompt final deve descrever a estrutura do site, sugerir a identidade visual, e incluir copy para as principais seções.
+    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.
 """
-    resultado = generator(entrada, max_new_tokens=500, do_sample=True, temperature=0.7)
-    return resultado[0]['generated_text']
+    result = generator(prompt_base, max_length=400, do_sample=True, temperature=0.7)
+    return result[0]["generated_text"]
 
-# Interface
 with gr.Blocks() as demo:
-    gr.Markdown("## Agente Raypages – Gere o Prompt Perfeito para seu Site com IA")
+    gr.Markdown("### ✨ Agente Raypages: Gere o prompt ideal para seu site IA")
 
-    nicho = gr.Textbox(label="Qual é o nicho do site?")
+    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="Tem alguma cor principal que deseja?")
-    estilo = gr.Textbox(label="Qual estilo visual você prefere?")
-    beneficio = gr.Textbox(label="Qual o maior benefício do produto ou serviço?")
-    promessa = gr.Textbox(label="Existe alguma promessa principal que o site deve comunicar?")
-    publico = gr.Textbox(label="Quem é o público-alvo do site?")
-    provas = gr.Textbox(label="Quer adicionar provas sociais ou de autoridade?")
-    cta = gr.Textbox(label="Qual é a chamada para ação (CTA)?")
+    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
-    )
+    btn.click(gerar_prompt, inputs=[nicho, sensacao, cor, estilo, beneficio, promessa, publico, provas, cta], outputs=output)
 
 demo.launch()