from transformers import pipeline
import gradio as gr

generator = pipeline("text2text-generation", model="google/flan-t5-small")

def generate_website_code(prompt):
    full_prompt = f"Create a stylish animated HTML, CSS, and JS website using this prompt: {prompt}"
    result = generator(full_prompt, max_new_tokens=1024)[0]["generated_text"]
    return result

gr.Interface(fn=generate_website_code, inputs="text", outputs="text", title="Website Generator").launch()