Update app.py

app.py

@@ -8,20 +8,21 @@ 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, text_type, length, product_mention, mood, model_choice):
+def get_gemini_response(target_audience, product, product_mention, text_type, length, mood, model_choice):
     model = genai.GenerativeModel(model_choice)
     
-    # Crear el prompt completo basado en los campos del frontend
+    # Crear la instrucción de mención basada en la opción seleccionada
     mention_instruction = ""
-    if product_mention == "Direct":
-        mention_instruction = f"Make sure to mention the product '{product}' directly in the text."
-    elif product_mention == "Subtle":
+    if product_mention == "Directa":
+        mention_instruction = f"Mention the product '{product}' directly in the text."
+    elif product_mention == "Indirecta":
         mention_instruction = f"Ensure that subtle and realistic references to the product '{product}' are included without explicitly mentioning it."
-    elif product_mention == "Metaphor":
-        mention_instruction = f"Refer to the product '{product}' through a metaphor, without mentioning it explicitly."
+    elif product_mention == "Metafórica":
+        mention_instruction = f"Refer to the product '{product}' through a metaphor without explicitly mentioning it."
 
+    # Crear el prompt completo basado en los campos del frontend
     full_prompt = f"""
-Eres un escritor creativo experto en el arte de la persuasión. Escribe una {length} {text_type} en español. El tono del {text_type} debe ser {mood} y estar diseñado específicamente para resonar emocionalmente con un público {target_audience}. {mention_instruction} Usa técnicas persuasivas para guiar al lector hacia una comprensión intuitiva de los beneficios del producto, enfocándote en crear una fuerte conexión emocional con la audiencia.
+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} 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.
     """
     
     response = model.generate_content([full_prompt])
@@ -69,16 +70,18 @@ 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í...")
-    text_type = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
-    length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
-    
-    # Nueva opción para cómo se debe mencionar el producto
-    product_mention = st.selectbox("Mención del producto:", ["Direct", "Subtle", "Metaphor"])
-    
-    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)
+    # 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"])
+        
+        text_type = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
+        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")
@@ -87,7 +90,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, text_type, length, product_mention, mood, model_choice)
+            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: