import gradio as gr
from transformers import pipeline

# Carregar modelo de geração de código
generator = pipeline("text-generation", model="Salesforce/codegen-350M-mono")

def gerar_site(prompt):
    instrucao = f"Crie um código HTML com CSS embutido para: {prompt}"
    resultado = generator(instrucao, max_length=600, temperature=0.7, do_sample=True)
    codigo = resultado[0]["generated_text"]
    return f"<iframe srcdoc='{codigo}' style='width:100%; height:500px;'></iframe>", codigo

# Interface
app = gr.Interface(
    fn=gerar_site,
    inputs=gr.Textbox(lines=3, placeholder="Ex: Site de portfólio com foto e botão de contato"),
    outputs=[gr.HTML(), gr.Code(language="html")]
)

app.launch()