import streamlit as st
from transformers import pipeline

# Load text-generation pipeline from Hugging Face
generator = pipeline("text-generation", model="gpt2")

# Streamlit App UI
st.set_page_config(page_title="AI Text Generator", page_icon="📝")
st.title("📝 AI Text Generator with GPT-2")

# User Input
prompt = st.text_area("Enter a prompt:", "Once upon a time")

# Generate Text Button
if st.button("Generate Text"):
    with st.spinner("Generating..."):
        result = generator(prompt, max_length=100, num_return_sequences=1)
        st.subheader("Generated Text:")
        st.write(result[0]["generated_text"])