Update app.py

app.py

@@ -8,7 +8,7 @@ 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):
+def get_gemini_response(target_audience, product, product_mention, mood, model_choice):
     model = genai.GenerativeModel(model_choice)
     
     # Crear la instrucción de mención basada en la opción seleccionada
@@ -20,20 +20,15 @@ def get_gemini_response(target_audience, product, product_mention, text_type, le
     elif product_mention == "Metafórica":
         mention_instruction = f"Refer to the product '{product}' through a metaphor without explicitly mentioning it."
 
-    # Instrucción específica para el tipo de texto
-    format_instruction = ""
-    if text_type == "Historia":
-        format_instruction = "Ensure that the narrative is engaging and includes character development, a clear plot, and a resolution."
-    elif text_type == "Carta de venta":
-        format_instruction = "Craft a compelling sales pitch that highlights the product's benefits and persuades the reader to take action."
-    elif text_type == "Correo":
-        format_instruction = "Write a concise and engaging email that captures the reader's attention and encourages a response."
-    elif text_type == "Landing page":
-        format_instruction = "Design an informative and persuasive landing page that effectively showcases the product and drives conversions."
+    # 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 = "Corto"
 
     # 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.
+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.
     """
     
     response = model.generate_content([full_prompt])
@@ -87,9 +82,7 @@ with col1:
         # 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"])
+        # 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
@@ -102,7 +95,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, text_type, length, mood, model_choice)
+            response = get_gemini_response(target_audience, product, product_mention, mood, model_choice)
             col2.subheader("Contenido generado:")
             col2.write(response)
         except ValueError as e: