Update app.py

app.py

@@ -193,29 +193,49 @@ st.markdown("<h3 style='text-align: center;'>Teje historias inolvidables en segu
 
 # Acordeón para elegir la fórmula
 with st.expander("Selecciona la fórmula para generar tu historia"):
-    formula = st.radio("Selecciona la fórmula para generar tu historia:", ["P.A.S.A.", "A.D.P.", "G.H.A."])
+    # Obtener las fórmulas disponibles del diccionario story_formulas
+    available_formulas = list(story_formulas.keys())
+    
+
+formula = st.radio(
+        "Selecciona la fórmula para generar tu historia:",
+        formula_display_names
+    )
+    # Convertir el nombre de presentación de vuelta a la clave del diccionario
+    selected_formula = ''.join(formula.split('.'))
 
 # Crear las columnas para el diseño (40% para la izquierda, 60% para la derecha)
 col1, col2 = st.columns([2, 3])
 
-# Cambiar el comportamiento del slider cuando se seleccione "G.H.A."
+# Cambiar el comportamiento del slider cuando se seleccione "GHA"
 with col1:
-    target_audience = st.text_input("Público objetivo", placeholder="¿A quién está dirigido tu mensaje?")
-    product = st.text_input("Producto/Servicio", placeholder="¿Qué estás ofreciendo?")
-    action = st.text_area("Llamado a la acción", placeholder="¿Qué acción específica debe tomar tu audiencia?")
+    target_audience = st.text_input("Público objetivo",
+                                   placeholder="¿A quién está dirigido tu mensaje?")
+    product = st.text_input("Producto/Servicio",
+                           placeholder="¿Qué estás ofreciendo?")
+    action = st.text_area("Llamado a la acción",
+                         placeholder="¿Qué acción específica debe tomar tu audiencia?")
 
     # Personalización adicional
     with st.expander("Personaliza tu historia"):
-        mood = st.selectbox("Tono de la historia:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
+        mood = st.selectbox("Tono de la historia:",
+                           ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
 
         # Ajustar el slider según la fórmula seleccionada
-        if formula == "G.H.A.":
-            length = st.slider("Longitud de la historia (palabras):", min_value=50, max_value=200, value=150, step=10)
+        if selected_formula == "GHA":
+            length = st.slider("Longitud de la historia (palabras):",
+                             min_value=50, max_value=200, value=150, step=10)
         else:
-            length = st.slider("Longitud de la historia (palabras):", min_value=50, max_value=150, value=100, step=10)
+            length = st.slider("Longitud de la historia (palabras):",
+                             min_value=50, max_value=150, value=100, step=10)
 
-    if formula == "G.H.A.":
-        story_topic = st.text_area("De qué quieres que trate la historia", placeholder="Explica si hay algo específico sobre lo que te gustaría contar (puede ser una vivencia personal, película, cuento, personaje ficticio, etc.).")
+    # Mostrar campo adicional solo para GHA
+    story_topic = None
+    if selected_formula == "GHA":
+        story_topic = st.text_area(
+            "De qué quieres que trate la historia",
+            placeholder="Explica si hay algo específico sobre lo que te gustaría contar (puede ser una vivencia personal, película, cuento, personaje ficticio, etc.)."
+        )
 
     # Botón para generar contenido
     submit = st.button("Generar mi historia")
@@ -225,16 +245,17 @@ with col2:
     if submit:
         if target_audience and product and action:
             try:
-                if formula == "P.A.S.A.":
-                    response = generate_story("PASA", target_audience, product, action, mood, length)
-                elif formula == "A.D.P.":
-                    response = generate_story("ADP", target_audience, product, action, mood, length)
-                elif formula == "G.H.A.":
+                # Verificar si es GHA y necesita story_topic
+                if selected_formula == "GHA":
                     if story_topic:
-                        response = generate_story("GHA", target_audience, product, action, mood, length, story_topic)
+                        response = generate_story(selected_formula, target_audience, product,
+                                               action, mood, length, 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)
 
                 if response:
                     st.subheader("Historia generada:")