# app.py
import streamlit as st
from transformers import pipeline

# Page config
st.set_page_config(page_title="AI Text Generator", page_icon="🤖")

st.title("🤖 AI Text Generator")
st.write("Enter a prompt and get AI-generated text!")

# Prompt input
prompt = st.text_area("Your prompt here:", height=150)

# Optional sliders for advanced users
st.sidebar.header("Generation Settings")
max_length = st.sidebar.slider("Max Length", min_value=50, max_value=500, value=200, step=10)
temperature = st.sidebar.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
top_k = st.sidebar.slider("Top-k sampling", min_value=10, max_value=100, value=50, step=5)

# Generate button
if st.button("Generate"):
    if prompt.strip() == "":
        st.warning("Please enter a prompt!")
    else:
        with st.spinner("Generating text..."):
            generator = pipeline("text-generation", model="gpt2")  # You can change to any HF model
            result = generator(
                prompt,
                max_length=max_length,
                do_sample=True,
                temperature=temperature,
                top_k=top_k
            )
            st.subheader("Generated Text")
            st.write(result[0]['generated_text'])