Update app.py

app.py

@@ -12,7 +12,7 @@ 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, formula_type, length, mood, target_audience, cta_type):
+def get_gemini_response(input_prompt, formula_type, length, mood, target_audience, cta_type, temperature):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
@@ -52,7 +52,7 @@ def get_gemini_response(input_prompt, formula_type, length, mood, target_audienc
     
     Format: Three clean story texts with natural paragraphs, separated by "---", no labels, no explanations.
     """
-    response = model.generate_content([full_prompt])
+    response = model.generate_content([full_prompt], generation_config={"temperature": temperature})
     return response.parts[0].text if response and response.parts else "Error al generar contenido."
 
 # Configurar la aplicación Streamlit
@@ -83,14 +83,22 @@ with col1:
             options=list(story_formulas.keys()),
             format_func=lambda x: x
         )
-        # Remove or comment out this line
-        # st.info(story_formulas[formula_type])
         
         length = st.slider("Longitud del texto (palabras):", 
                           min_value=100, 
                           max_value=150, 
                           value=125,
                           step=5)
+                          
+        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."
+        )
+        
         mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
     generate_button = st.button("Generar historia")
 # Al inicio de la app
@@ -101,7 +109,8 @@ with col2:
     if generate_button:
         response = get_gemini_response(
             input_prompt, formula_type, length,
-            mood, target_audience, cta_type
+            mood, target_audience, cta_type,
+            temperature
         )
         stories = response.split("---")
         for i, story in enumerate(stories, 1):