import streamlit as st
from transformers import pipeline

# Título de la app
st.title("Generador de Texto con GPT-2")

# Inicializar el modelo de Hugging Face
st.write("Cargando modelo GPT-2...")
generator = pipeline("text-generation", model="gpt2")

# Input del usuario
prompt = st.text_area("Escribe un texto para que GPT-2 continúe:")

# Botón para generar texto
if st.button("Generar Texto"):
    if prompt.strip() == "":
        st.warning("Por favor, ingresa un texto válido.")
    else:
        with st.spinner("Generando texto..."):
            # Generar texto usando GPT-2
            output = generator(prompt, max_length=100, num_return_sequences=1)
        
        # Mostrar resultado
        st.subheader("Texto Generado:")
        st.write(output[0]["generated_text"])