Update app.py

app.py

@@ -168,7 +168,7 @@ 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, temperature):
+def generate_headline_for_story(story, formula_type, product, target_audience, mood, temperature):
     generation_config = {
         "temperature": temperature,
         "top_p": 0.65,
@@ -303,6 +303,16 @@ def generate_headline_for_story(model, story, formula_type, product, target_audi
 
 def generate_story(formula_type, target_audience, product, action, mood, length, temperature, story_topic=None):
     """Función unificada para generar historias"""
+
+    # Definir la configuración del modelo
+    generation_config = {
+        "temperature": temperature,
+        "top_p": 0.65,
+        "top_k": 360,
+        "max_output_tokens": 8196,
+    }
+
+    # Inicializar el modelo con la configuración
     model = genai.GenerativeModel(
         model_name="gemini-2.0-flash",
         generation_config=generation_config