from transformers import pipeline
import gradio as gr

generator = pipeline("text-generation", model="Salesforce/codegen2-1B")

def generate_code(prompt):
    output = generator(prompt, max_length=2048, do_sample=True)[0]["generated_text"]
    return output

iface = gr.Interface(
    fn=generate_code,
    inputs=gr.Textbox(lines=5, placeholder="Enter your website prompt here..."),
    outputs="text",
    title="Website Code Generator",
    description="Enter a prompt like: 'Create a modern café website using beige and brown with animations.'"
)

iface.launch()