Update app.py

app.py

@@ -118,28 +118,7 @@ def generate_story(formula_type, target_audience, product, action, mood, length,
     if formula_type not in story_formulas:
         raise ValueError("Fórmula no válida")
 
-    formula_description = story_formulas[formula_type]["description"]
-
-    prompt = f"""{system_prompt}
-
-Como experto copywriter, crea una historia persuasiva en español de {length} palabras con un tono {mood}, siguiendo la estructura:
-
-{formula_description}
-
-Público objetivo: {target_audience}
-Producto/Servicio: {product}
-Llamada a acción: {action}
-"""
-
-    if formula_type == "GHA" and story_topic:
-        prompt += f"\nTema de la historia: {story_topic}"
-
-    response = model.generate_content([prompt])
-    if response and response.parts:
-        return response.parts[0].text.strip()
-    raise ValueError("No se pudo generar la historia")
-
-if formula_type == "PASA":
+    if formula_type == "PASA":
         prompt = f"""
         As a master copywriter and storyteller, craft a compelling {length}-word story in Spanish that follows the P.A.S.A. formula while maintaining a {mood} tone.
 
@@ -184,11 +163,26 @@ if formula_type == "PASA":
         - Ensure call-to-action feels like a logical next step
         - Use language that resonates with {target_audience}
         """
+    else:
+        formula_description = story_formulas[formula_type]["description"]
+        prompt = f"""{system_prompt}
+
+Como experto copywriter, crea una historia persuasiva en español de {length} palabras con un tono {mood}, siguiendo la estructura:
+
+{formula_description}
 
-        response = model.generate_content([prompt])
-        if response and response.parts:
-            return response.parts[0].text.strip()
-        raise ValueError("No se pudo generar la historia")
+Público objetivo: {target_audience}
+Producto/Servicio: {product}
+Llamada a acción: {action}
+"""
+
+        if formula_type == "GHA" and story_topic:
+            prompt += f"\nTema de la historia: {story_topic}"
+
+    response = model.generate_content([prompt])
+    if response and response.parts:
+        return response.parts[0].text.strip()
+    raise ValueError("No se pudo generar la historia")
     
 # Inicializar la aplicación Streamlit
 st.set_page_config(page_title="Generador de Historias", page_icon=":pencil:", layout="wide")