Update app.py

app.py

@@ -168,7 +168,19 @@ Inscríbete hoy y recibe un módulo extra sobre cómo adaptar tu mensaje sin per
     }
 }
 
-def generate_headline_for_story(model, story, formula_type, product, target_audience, mood):
+def generate_headline_for_story(model, story, formula_type, product, target_audience, mood, temperature):
+    generation_config = {
+        "temperature": temperature,
+        "top_p": 0.65,
+        "top_k": 360,
+        "max_output_tokens": 8196,
+    }
+
+    model = genai.GenerativeModel(
+        model_name="gemini-2.0-flash",
+        generation_config=generation_config
+    )
+    
     """Función para generar un titular basado en la historia"""
 
     system_prompt = """Eres un experto copywriter especializado en crear titulares persuasivos.
@@ -289,9 +301,12 @@ def generate_headline_for_story(model, story, formula_type, product, target_audi
         return headline_text
     return None
 
-def generate_story(formula_type, target_audience, product, action, mood, length, story_topic=None):
+def generate_story(formula_type, target_audience, product, action, mood, length, temperature, story_topic=None):
     """Función unificada para generar historias"""
-    model = genai.GenerativeModel("gemini-2.0-flash")
+    model = genai.GenerativeModel(
+        model_name="gemini-2.0-flash",
+        generation_config=generation_config
+    )
 
     if formula_type not in story_formulas:
         raise ValueError("Fórmula no válida")
@@ -651,6 +666,15 @@ with col1:
             length = st.slider("Longitud de la historia (palabras):",
                              min_value=50, max_value=150, value=100, step=10)
 
+        temperature = st.slider(
+            "Nivel de creatividad:",
+            min_value=0.0,
+            max_value=2.0,
+            value=1.0,
+            step=0.1,
+            help="Valores más altos generan historias más creativas pero menos predecibles. Valores más bajos producen historias más consistentes."
+        )    
+
     # Campo story_topic fuera del expander pero dependiente de la fórmula seleccionada
     story_topic = None
     if selected_formula == "GHA":
@@ -667,17 +691,16 @@ with col2:
     if submit:
         if target_audience and product and action:
             try:
-                # Verificar si es GHA y necesita story_topic
                 if selected_formula == "GHA":
                     if story_topic:
                         response = generate_story(selected_formula, target_audience, product,
-                                               action, mood, length, story_topic)
+                                               action, mood, length, temperature, story_topic)
                     else:
                         st.error("Por favor, completa todos los campos requeridos para la fórmula G.H.A.")
                         response = ""
                 else:
                     response = generate_story(selected_formula, target_audience, product,
-                                           action, mood, length)
+                                           action, mood, length, temperature)
 
                 if response:
                     st.subheader("Historia generada:")