Upload 11 files

app.py

@@ -78,82 +78,147 @@ 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:
-    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...")
+    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
+    )
     
     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"
+            label_visibility="visible",
+            key="ad_formula_key",
+            index=list(ads_formulas.keys()).index(st.session_state.ad_formula_key)
         )
+        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"
+            label_visibility="visible",
+            key="emotional_angle_key",
+            index=list(emotional_angles.keys()).index(st.session_state.emotional_angle_key)
         )
+        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"
+            label_visibility="visible",
+            key="ad_persona_key",
+            index=list(copywriter_personas.keys()).index(st.session_state.ad_persona_key)
         )
+        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"
+            label_visibility="visible",
+            key="ad_objective_key",
+            index=list(ad_objectives.keys()).index(st.session_state.ad_objective_key)
         )
+        st.session_state.ad_objective_key = ad_objective_key
         selected_objective = ad_objectives[ad_objective_key] if ad_objective_key != "ninguno" else None
 
-        # Removed language selector
-
+        # Slider de temperatura
         ad_temperature = st.slider(
             "Nivel de creatividad",
             min_value=0.0, 
             max_value=2.0, 
-            value=1.0, 
+            value=st.session_state.ad_temperature, 
             step=0.1,
-            label_visibility="visible"
+            label_visibility="visible",
+            key="ad_temperature_slider"
         )
+        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 ad_target_audience and ad_product:
-        with st.spinner('Generando anuncio...'):
-            generated_ad = generate_fb_ad(
-                ad_target_audience, 
-                ad_product, 
-                ad_temperature, 
-                ad_formula,
-                emotional_angle,
-                ad_persona,
-                input_prompt,  # Pass the new story prompt
-                selected_objective
-            )
-            if not isinstance(generated_ad, str):
-                col2.error("Error al generar el anuncio")
-            else:
-                col2.markdown(f"""
-                    <div style="{styles['results_container']}">
-                        <h3>Anuncio Generado:</h3>
-                        <p>{generated_ad}</p>
-                    </div>
-                """, unsafe_allow_html=True)
+    if st.session_state.ad_target_audience and st.session_state.ad_product:
+        # Mover el spinner a 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_formula,
+                    emotional_angle,
+                    ad_persona,
+                    st.session_state.input_prompt,
+                    selected_objective
+                )
+                if not isinstance(generated_ad, str):
+                    st.error("Error al generar el anuncio")
+                else:
+                    st.markdown(f"""
+                        <div style="{styles['results_container']}">
+                            <h3>Anuncio Generado:</h3>
+                            <p>{generated_ad}</p>
+                        </div>
+                    """, unsafe_allow_html=True)
     else:
         col2.warning("Por favor, completa todos los campos antes de generar el anuncio.")