Update app.py

app.py

@@ -8,13 +8,13 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener respuesta del modelo Gemini
-def get_gemini_response(input_prompt, product, genre, length, language, mood, model_choice):
+def get_gemini_response(target_audience, product, text_type, length, language, mood, model_choice):
     model = genai.GenerativeModel(model_choice)
     
-    # Crear el prompt completo
+    # Crear el prompt completo basado en los campos del frontend
     full_prompt = f"""
-    Eres un escritor creativo famoso. Crea un {length} {genre} en {language}. 
-    El {genre} debe ser {mood}, dirigido a {input_prompt}, y relacionado con el producto: "{product}".
+    Eres un escritor creativo famoso. Crea un {length} {text_type} en {language}. 
+    El {text_type} debe ser {mood}, dirigido a este público objetivo: {target_audience}, y relacionado con el producto: "{product}".
     Asegúrate de que contenga elementos realistas y emocionales.
     """
     
@@ -63,9 +63,9 @@ 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í...")
-    genre = st.selectbox("Tipo de texto:", ["Historia", "Shayari", "Sher", "Poema", "Cita"])
+    text_type = st.selectbox("Tipo de texto:", ["Story", "Shayari", "Sher", "Poem", "Quote"])
     length = st.selectbox("Longitud del texto:", ["Corto", "Largo"])
-    language = st.selectbox("Idioma del texto:", ["Inglés", "Español"])
+    language = st.selectbox("Idioma del texto:", ["Español", "Inglés"])
     mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
     
     # Opción para seleccionar el modelo
@@ -78,7 +78,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, genre, length, language, mood, model_choice)
+            response = get_gemini_response(target_audience, product, text_type, length, language, mood, model_choice)
             col2.subheader("Contenido generado:")
             col2.write(response)
         except ValueError as e: