Update app.py

app.py

@@ -8,8 +8,8 @@ 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, mood, model_choice):
-    model = genai.GenerativeModel(model_choice)
+def get_gemini_response(target_audience, product, product_mention, mood, length):
+    model = genai.GenerativeModel("gemini-1.5-flash")  # Modelo predeterminado
     
     # Crear la instrucción de mención basada en la opción seleccionada
     mention_instruction = ""
@@ -23,9 +23,6 @@ def get_gemini_response(target_audience, product, product_mention, mood, model_c
     # Instrucción específica para el tipo de texto (siempre será "Historia")
     format_instruction = "Ensure that the narrative is engaging and includes character development, a clear plot, and a resolution."
 
-    # Longitud del texto (siempre será "Corto")
-    length = "200 words"
-
     # 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} Historia in Spanish. The tone of the Historia 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.
@@ -84,9 +81,9 @@ with col1:
         
         # Eliminar la opción de tipo de texto y longitud del texto
         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)
+
+        # Slider para seleccionar la longitud del texto
+        length = st.slider("Longitud del texto (palabras):", min_value=100, max_value=300, value=200, step=50)
 
     # Botón para generar contenido
     submit = st.button("Escribir mi historia")
@@ -95,7 +92,7 @@ with col1:
 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, mood, model_choice)
+            response = get_gemini_response(target_audience, product, product_mention, mood, length)
             col2.subheader("Contenido generado:")
             col2.write(response)
         except ValueError as e: