Update app.py

app.py

@@ -69,19 +69,25 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
     response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
     return response.text  # Regresar la respuesta directamente
 
-# Configurar la interfaz de usuario con Gradio
-iface = gr.Interface(
-    fn=generate_headlines,
-    inputs=[
-        gr.Dropdown(choices=[str(i) for i in range(1, 11)], label="Número de Titulares", value="5"),
-        gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios"),
-        gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés"),
-        gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="Creatividad")
-    ],
-    outputs=gr.Markdown(label="Titulares Generados"), 
-    title="Generador de Titulares",
-    description="Usa el poder de Gemini AI para crear titulares atractivos. Ajusta los parámetros para generar titulares que capturen la atención de tu audiencia."
-)
+# Configurar la interfaz de usuario con Streamlit
+st.set_page_config(page_title="Generador de Titulares", layout="wide")
 
-# Lanza la interfaz
-iface.launch()
+# Centrar el título y el subtítulo
+st.markdown("<h1 style='text-align: center;'>Generador de Titulares</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>Usa el poder de Gemini AI para crear titulares atractivos.</h4>", unsafe_allow_html=True)
+
+# Campos de entrada
+target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
+product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
+number_of_headlines = st.selectbox("Número de Titulares", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
+temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+
+# Botón de enviar
+if st.button("Generar Titulares"):
+    if target_audience and product:
+        # Obtener la respuesta del modelo
+        generated_headlines = generate_headlines(number_of_headlines, target_audience, product, temperature)
+        st.markdown(f"**Estos son tus encabezados para enganchar a {target_audience}:**")
+        st.markdown(generated_headlines)
+    else:
+        st.error("Por favor, proporciona el público objetivo y el producto.")