Update app.py

app.py

@@ -27,8 +27,10 @@ def get_random_copywriting_technique():
     return random.sample(techniques, k=random.randint(1, 3))  # Escoge entre 1 y 3 técnicas al azar
 
 # Al crear el prompt completo, incluye el tamaño y las técnicas seleccionadas
-def get_gemini_response(target_audience, product, text_type, length, mood, model_choice):
+def get_gemini_response(target_audience, product, text_type, length, mood):
     product_mention = get_random_product_mention()
+    model_choice = "gemini-1.5-flash"  # Modelo predeterminado
+
     model = genai.GenerativeModel(model_choice)
 
     # Obtener técnicas aleatorias
@@ -105,16 +107,15 @@ with col1:
     target_audience = st.text_input("¿Quién es tu público objetivo?")
     product = st.text_input("¿Qué producto tienes en mente?")
     text_type = st.selectbox("Tipo de texto", ["Historia", "Carta de venta", "Correo", "Landing page"])
-    length = st.slider("Número de palabras", min_value=50, max_value=500, value=100)
-    mood = st.selectbox("Tono de voz", ["Conversacional", "Formal", "Emocional", "Persuasivo"])
-    model_choice = st.selectbox("Modelo", ["gemini-1.5-flash", "gemini-1.5-pro"])
+    length = st.slider("Número de palabras", min_value=50, max_value=300, value=150)
+    mood = st.selectbox("Tono de voz", ["Conversacional", "Formal", "Emocional", "Persuasivo", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
 
 # Botón para generar el texto (dentro de la interfaz principal)
 if st.button("Escribir mi texto"):
     if target_audience and product:  # Verificar que se haya proporcionado el público objetivo y el producto
         try:
             # Obtener la respuesta del modelo
-            generated_text = get_gemini_response(target_audience, product, text_type, length, mood, model_choice)
+            generated_text = get_gemini_response(target_audience, product, text_type, length, mood)
             col2.subheader("Contenido generado:")
             col2.write(generated_text)  # Resultado del texto generado
         except ValueError as e: