Update app.py

app.py

@@ -2,13 +2,24 @@ 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
 
 load_dotenv()
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Función para obtener respuesta del modelo Gemini
-def get_gemini_response(target_audience, product, product_mention, text_type, length, mood, model_choice):
+# Función para obtener una mención del producto de manera probabilística
+def get_random_product_mention():
+    # Definir las opciones de mención y sus probabilidades
+    mentions = ["Directa", "Indirecta", "Metafórica"]
+    probabilities = [0.34, 0.33, 0.33]  # Probabilidades de cada mención (ajusta según desees)
+    
+    # Elegir una opción al azar basándose en las probabilidades
+    return random.choices(mentions, probabilities)[0]
+
+def get_gemini_response(target_audience, product, text_type, length, mood, model_choice):
+    # Generar la mención aleatoria
+    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
@@ -48,7 +59,7 @@ def get_gemini_response(target_audience, product, product_mention, text_type, le
     # Crear el prompt completo basado en los campos del frontend
     full_prompt = f"""
     You are a creative writer skilled in the art of persuasion. Write a {length} {text_type} in Spanish. The tone of the {text_type} should be {mood} and carefully crafted to emotionally resonate with a {target_audience}. {mention_instruction} {format_instruction} Use persuasive techniques to guide the reader towards an intuitive understanding of the product's benefits, focusing on creating a strong emotional connection with the audience.
-    Create an opening paragraph for a {text_type.lower()} that makes {target_audience} aware they have a problem by explaining it with real-life situations, using a natural or conversational tone. The goal of this paragraph is to make them want to keep reading and find out what {product} is about. Use persuasion effectively in every word, mastering advanced techniques like using powerful headlines and intriguing openings.
+    Create an opening paragraph for a {text_type} that makes {target_audience} aware they have a problem by explaining it with real-life situations, using a natural or conversational tone. The goal of this paragraph is to make them want to keep reading and find out what {product} is about. Use persuasion effectively in every word, mastering advanced techniques like using powerful headlines and intriguing openings.
 
     To write this opening paragraph, use one or more of these Copywriting Techniques, choosing the most appropriate ones:
 
@@ -69,67 +80,26 @@ def get_gemini_response(target_audience, product, product_mention, text_type, le
         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
+st.set_page_config(page_title="VisionTales", layout="wide")
+st.title("VisionTales: Crea historias inolvidables con IA")
 
-# 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)
+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"])
 
-# 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)
+# 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}")
 
-# 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"):
-        # Variable para cómo se debe mencionar el producto
-        product_mention = st.selectbox("Mención del producto:", ["Directa", "Indirecta", "Metafórica"])
-        
-        # 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, product_mention, 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.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.")