import gradio as gr
from transformers import pipeline

# Load a proper text generation model
generator = pipeline(
    "text-generation",
    model="gpt2",
    max_length=100,
    num_return_sequences=1
)

def generate_text(prompt):
    result = generator(prompt)
    return result[0]["generated_text"]

interface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(
        lines=4,
        placeholder="Enter a prompt...",
        label="Your Prompt"
    ),
    outputs=gr.Textbox(
        label="Generated Text"
    ),
    title="AI Text Generator",
    description="Enter a prompt and the AI will generate text using GPT-2."
)

interface.launch()