Update app.py

app.py

@@ -10,13 +10,15 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener la respuesta del modelo Gemini
-def get_gemini_response(input_prompt, genre, length, language, mood):
+def get_gemini_response(input_prompt, genre, length, language, mood, target_audience, product_type):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
     model = genai.GenerativeModel('gemini-1.5-flash')
     full_prompt = f"""
-    You are a creative writer. Create a {length} {genre} in {language}. The {genre} should be {mood}. 
+    You are a creative writer. Create a {length} {genre} in {language}. 
+    The {genre} should be {mood} and targeted to {target_audience}.
+    This is for a {product_type}.
     The {genre} should be based on the following prompt:
     
     "{input_prompt}"
@@ -49,13 +51,25 @@ with col1:
     length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
     language = st.selectbox("Idioma del texto:", ["Español", "Inglés"])
     mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
+    
+    # Nuevas opciones
+    target_audience = st.selectbox("Público Objetivo:", [
+        "Jóvenes", "Adultos", "Niños", "Profesionales",
+        "Estudiantes", "Familias", "Empresarios"
+    ])
+    
+    product_type = st.selectbox("Producto o Servicio:", [
+        "Producto físico", "Servicio digital", "Curso online",
+        "Consultoría", "Software", "Servicio profesional"
+    ])
 
-    # Botón para generar contenido
     generate_button = st.button("Generar historia")
 
-# Columna derecha para la salida
 with col2:
     if generate_button:
-        response = get_gemini_response(input_prompt, genre, length, language, mood)
+        response = get_gemini_response(
+            input_prompt, genre, length, language, 
+            mood, target_audience, product_type
+        )
         st.subheader("Contenido generado:")
         st.write(response)