Upload app.py

app.py

@@ -0,0 +1,75 @@
+import streamlit as st
+from transformers import pipeline
+
+# Configuración de la página
+st.set_page_config(
+    page_title="¿No sabes qué decir? Te ayudamos",
+    page_icon="💡",
+    layout="centered"
+)
+
+# Estilos personalizados en colores pasteles
+st.markdown(
+    """
+    <style>
+    body {
+        background-color: #fefaf4;
+    }
+    .title {
+        font-size: 38px;
+        font-weight: bold;
+        text-align: center;
+        color: #a3b18a;  /* Verde pastel */
+        margin-bottom: 10px;
+    }
+    .subtitle {
+        font-size: 18px;
+        text-align: center;
+        color: #588157;  /* Verde más oscuro pastel */
+    }
+    .footer {
+        text-align: center;
+        font-size: 12px;
+        margin-top: 30px;
+        color: #8d99ae; /* Gris pastel */
+    }
+    .stButton>button {
+        background-color: #ffdab9; /* Naranja pastel */
+        color: #3d405b;
+        font-size: 16px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
+# Título llamativo
+st.markdown('<p class="title">¿No sabes qué decir? 💡</p>', unsafe_allow_html=True)
+st.markdown('<p class="subtitle">¡Te ayudamos a completar tus ideas con inteligencia artificial!</p>', unsafe_allow_html=True)
+
+# Entrada del usuario
+st.subheader("📝 Escribe tu idea:")
+prompt = st.text_area("Completa tu idea:", placeholder="Ejemplo: Me gustaría escribir una carta a...")
+
+# Generar respuesta
+if st.button("✨ Generar Texto ✨"):
+    st.info("Cargando modelo... esto puede tardar unos segundos.")
+    with st.spinner("⏳ Generando texto..."):
+        generator = pipeline("text-generation", model="datificate/gpt2-small-spanish")
+        result = generator(
+            prompt,
+            max_length=20,
+            temperature=0.2,
+            top_p=0.7,
+            top_k=20,
+            repetition_penalty=1.5,
+            num_return_sequences=1,
+            eos_token_id=50256,
+            do_sample=False
+        )
+        response = result[0]["generated_text"]
+    st.subheader("🖋️ Texto Generado:")
+    st.success(response)
+
+# Footer del grupo
+st.markdown('<p class="footer">Grupo 7 - Procesamiento de datos con AI - Especialización de Inteligencia Artificial UAO</p>', unsafe_allow_html=True)