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 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]
+
 # Función para obtener respuesta del modelo Gemini
-def get_gemini_response(target_audience, product, product_mention, text_type, length, mood, model_choice):
+def get_gemini_response(target_audience, product, text_type, length, mood, model_choice):
+    product_mention = get_random_product_mention()  # Obtener mención de producto aleatoria
     model = genai.GenerativeModel(model_choice)
     
     # Crear la instrucción de mención basada en la opción seleccionada
@@ -132,4 +143,4 @@ if submit:
         except ValueError as e:
             col2.error(f"Error: {str(e)}")
     else:
-        col2.error("Por favor, proporciona el público objetivo y el producto.")
+        col2.error("Por favor, proporciona el público objetivo y el producto.")