Update app.py

app.py

@@ -78,141 +78,77 @@ st.markdown(styles["button"], unsafe_allow_html=True)
 # Crear columnas
 col1, col2 = st.columns([1, 2])  
 
-# Modificar la parte de la interfaz de usuario para usar session_state
-
-# Inicializar valores en session_state si no existen
-if 'ad_target_audience' not in st.session_state:
-    st.session_state.ad_target_audience = ""
-if 'ad_product' not in st.session_state:
-    st.session_state.ad_product = ""
-if 'input_prompt' not in st.session_state:
-    st.session_state.input_prompt = ""
-if 'ad_formula_key' not in st.session_state:
-    st.session_state.ad_formula_key = list(ads_formulas.keys())[0]
-if 'emotional_angle_key' not in st.session_state:
-    st.session_state.emotional_angle_key = list(emotional_angles.keys())[0]
-if 'ad_persona_key' not in st.session_state:
-    st.session_state.ad_persona_key = list(copywriter_personas.keys())[0]
-if 'ad_objective_key' not in st.session_state:
-    st.session_state.ad_objective_key = list(ad_objectives.keys())[0]
-if 'ad_temperature' not in st.session_state:
-    st.session_state.ad_temperature = 1.0
-
-# Funciones para actualizar el estado
-def update_target_audience():
-    st.session_state.ad_target_audience = st.session_state.target_audience_input
-
-def update_product():
-    st.session_state.ad_product = st.session_state.product_input
-
-def update_prompt():
-    st.session_state.input_prompt = st.session_state.prompt_input
-
 # Columnas de entrada
 with col1:
-    st.text_input(
-        "¿Quién es tu público objetivo?", 
-        placeholder="Ejemplo: Madres trabajadoras de 30-45 años",
-        key="target_audience_input",
-        value=st.session_state.ad_target_audience,
-        on_change=update_target_audience
-    )
-    
-    st.text_input(
-        "¿Qué producto tienes en mente?", 
-        placeholder="Ejemplo: Curso de gestión del tiempo",
-        key="product_input",
-        value=st.session_state.ad_product,
-        on_change=update_product
-    )
-    
-    st.text_area(
-        "Escribe de qué quieres que trate la historia:", 
-        placeholder="Escribe aquí tu idea...",
-        key="prompt_input",
-        value=st.session_state.input_prompt,
-        on_change=update_prompt
-    )
-    
-    # Modify the selectbox sections to avoid directly updating session_state after widget creation
+    ad_target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Madres trabajadoras de 30-45 años")
+    ad_product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de gestión del tiempo")
+    input_prompt = st.text_area("Escribe de qué quieres que trate la historia:", placeholder="Escribe aquí tu idea...")
     
     with st.expander("Opciones avanzadas"):
         # Selector de fórmula de anuncio
         ad_formula_key = st.selectbox(
             "Fórmula de anuncio",
             options=list(ads_formulas.keys()),
-            label_visibility="visible",
-            key="ad_formula_key",
-            index=list(ads_formulas.keys()).index(st.session_state.ad_formula_key)
+            label_visibility="visible"
         )
-        # Remove this line: st.session_state.ad_formula_key = ad_formula_key
         ad_formula = ads_formulas[ad_formula_key]
         
         # Selector de ángulo emocional
         emotional_angle_key = st.selectbox(
             "Ángulo emocional",
             options=list(emotional_angles.keys()),
-            label_visibility="visible",
-            key="emotional_angle_key",
-            index=list(emotional_angles.keys()).index(st.session_state.emotional_angle_key)
+            label_visibility="visible"
         )
-        # Remove this line: st.session_state.emotional_angle_key = emotional_angle_key
         emotional_angle = emotional_angles[emotional_angle_key]
         
         # Selector de personalidad de copywriter
         ad_persona_key = st.selectbox(
             "Estilo de copywriter",
             options=list(copywriter_personas.keys()),
+            index=0,
             format_func=lambda x: x.replace("_", " "),
-            label_visibility="visible",
-            key="ad_persona_key",
-            index=list(copywriter_personas.keys()).index(st.session_state.ad_persona_key)
+            label_visibility="visible"
         )
-        # Remove this line: st.session_state.ad_persona_key = ad_persona_key
         ad_persona = copywriter_personas[ad_persona_key]
         
         # Selector de objetivo de anuncio
         ad_objective_key = st.selectbox(
             "Objetivo de la campaña",
             options=list(ad_objectives.keys()),
-            label_visibility="visible",
-            key="ad_objective_key",
-            index=list(ad_objectives.keys()).index(st.session_state.ad_objective_key)
+            label_visibility="visible"
         )
-        # Remove this line: st.session_state.ad_objective_key = ad_objective_key
         selected_objective = ad_objectives[ad_objective_key] if ad_objective_key != "ninguno" else None
-    
-        # Slider de temperatura
+
+        # Removed language selector
+
         ad_temperature = st.slider(
             "Nivel de creatividad",
             min_value=0.0, 
             max_value=2.0, 
-            value=st.session_state.ad_temperature, 
+            value=1.0, 
             step=0.1,
-            label_visibility="visible",
-            key="ad_temperature_slider"
+            label_visibility="visible"
         )
-        # Remove this line: st.session_state.ad_temperature = ad_temperature
 
-    # Usar los valores de session_state para el botón de generación
     submit_ad = st.button("GENERAR ANUNCIO")
 
 # Mostrar el anuncio generado
 if submit_ad:
-    if st.session_state.ad_target_audience and st.session_state.ad_product:
-        # Mover el spinner a la segunda columna
+    if ad_target_audience and ad_product:
+        # Mostrar el spinner en la segunda columna
         with col2:
             with st.spinner('Generando anuncio...'):
                 generated_ad = generate_fb_ad(
-                    st.session_state.ad_target_audience, 
-                    st.session_state.ad_product, 
-                    st.session_state.ad_temperature, 
+                    ad_target_audience, 
+                    ad_product, 
+                    ad_temperature, 
                     ad_formula,
                     emotional_angle,
                     ad_persona,
-                    st.session_state.input_prompt,
+                    input_prompt,  # Pass the new story prompt
                     selected_objective
                 )
+                
                 if not isinstance(generated_ad, str):
                     st.error("Error al generar el anuncio")
                 else: