Update app.py

app.py

@@ -2,7 +2,7 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
-import random # Nueva importación para la selección aleatoria de mención
+import random  # Nueva importación para la selección aleatoria de mención
 
 load_dotenv()
 
@@ -19,7 +19,7 @@ def get_random_product_mention():
 
 def get_gemini_response(target_audience, product, text_type, length, mood, model_choice):
     # Generar la mención aleatoria
-    product_mention = random_product_mention()  # Asumiendo que tienes esta función definida
+    product_mention = get_random_product_mention()  # Llamada correcta a la función
     model = genai.GenerativeModel(model_choice)
     
     # Crear la instrucción de mención basada en la opción seleccionada
@@ -80,65 +80,26 @@ def get_gemini_response(target_audience, product, text_type, length, mood, model
         raise ValueError("Lo sentimos, intenta con una combinación diferente de entradas.")
 
 # Inicializar la aplicación Streamlit
-st.set_page_config(page_title="Poetic Vision", page_icon=":pencil:", layout="wide")  # Configurar el diseño en ancho
-
-# Componentes principales de la UI
-st.markdown("<h1 style='text-align: center;'>VisionTales</h1>", unsafe_allow_html=True)
-st.markdown("<h3 style='text-align: center;'>Convierte tus ideas en historias inolvidables.</h3>", unsafe_allow_html=True)
-
-# Añadir CSS personalizado para el botón
-st.markdown("""
-    <style>
-    div.stButton > button {
-        background-color: #FFCC00; /* Color llamativo */
-        color: black;  /* Texto en negro */
-        width: 90%;
-        height: 60px;
-        font-weight: bold;
-        font-size: 22px; /* Tamaño más grande */
-        text-transform: uppercase; /* Texto en mayúsculas */
-        border: 1px solid #000000; /* Borde negro de 1px */
-        border-radius: 8px;
-        display: block;
-        margin: 0 auto;  /* Centramos el botón */
-    }
-    div.stButton > button:hover {
-        background-color: #FFD700; /* Color al pasar el mouse */
-        color: black;  /* Texto sigue en negro */
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-# Crear dos columnas para el diseño (40% y 60%)
-col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 partes en total
-
-with col1:
-    # Entradas del usuario
-    target_audience = st.text_input("Público objetivo:", placeholder="Especifica tu público aquí...")
-    product = st.text_input("Producto:", placeholder="Especifica el producto aquí...")
-    
-    # Agrupar todas las opciones en una acordeón
-    with st.expander("Personaliza tu texto"):
-        
-        # Opciones actualizadas para el tipo de texto
-        text_type = st.selectbox("Tipo de texto:", ["Historia", "Carta de venta", "Correo", "Landing page"])
-        length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
-        mood = st.selectbox("Tono del Texto:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
-        
-        # Opción para seleccionar el modelo
-        model_choice = st.selectbox("Selecciona el modelo:", ["gemini-1.5-flash", "gemini-1.5-pro"], index=0)
-
-    # Botón para generar contenido
-    submit = st.button("Escribir mi historia")
-
-# Manejo del clic en el botón
-if submit:
-    if target_audience and product:  # Verificar que se haya proporcionado el público objetivo y el producto
-        try:
-response = get_gemini_response(target_audience, product, text_type, length, mood, model_choice)
-            col2.subheader("Contenido generado:")
-            col2.write(response)
-        except ValueError as e:
-            col2.error(f"Error: {str(e)}")
-    else:
-        col2.error("Por favor, proporciona el público objetivo y el producto.")
+st.set_page_config(page_title="VisionTales", layout="wide")
+st.title("VisionTales: Crea historias inolvidables con IA")
+
+with st.sidebar:
+    st.header("Personaliza tu texto")
+    target_audience = st.text_input("¿Quién es tu público objetivo?")
+    product = st.text_input("Nombre del producto:")
+    text_type = st.selectbox("Tipo de texto:", ["Historia", "Carta de venta", "Correo", "Landing page"])
+    length = st.slider("Longitud del texto:", 50, 1000, 300)
+    mood = st.selectbox("Estado de ánimo:", ["Feliz", "Triste", "Motivador", "Reflexivo"])
+    model_choice = st.selectbox("Selecciona el modelo:", ["gemini-1.5-flash", "gemini-1.5-pro"])
+
+# Botón para generar el texto
+if st.button("Generar texto"):
+    try:
+        generated_text = get_gemini_response(target_audience, product, text_type, length, mood, model_choice)
+        st.subheader("Texto Generado:")
+        st.write(generated_text)
+    except Exception as e:
+        st.error(f"Error: {e}")
+
+st.markdown("---")
+st.write("Este generador de texto está diseñado para ayudarte a crear contenido atractivo y efectivo con la ayuda de la inteligencia artificial.")