Update app.py

app.py

@@ -92,7 +92,7 @@ story_formulas = {
     }
 }
 
-def generate_story(formula_type, **kwargs):
+def generate_story(formula_type, target_audience, product, action, mood, length, story_topic=None):
     """Función unificada para generar historias"""
     model = genai.GenerativeModel("gemini-1.5-flash")
 
@@ -103,17 +103,17 @@ def generate_story(formula_type, **kwargs):
 
     prompt = f"""{system_prompt}
 
-Como experto copywriter, crea una historia persuasiva en español de {kwargs['length']} palabras con un tono {kwargs['mood']}, siguiendo la estructura:
+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: {kwargs['target_audience']}
-Producto/Servicio: {kwargs['product']}
-Llamada a acción: {kwargs['action']}
+Público objetivo: {target_audience}
+Producto/Servicio: {product}
+Llamada a acción: {action}
 """
 
-    if formula_type == "GHA" and 'story_topic' in kwargs:
-        prompt += f"\nTema de la historia: {kwargs['story_topic']}"
+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:
@@ -162,16 +162,16 @@ with col2:
         if target_audience and product and action:
             try:
                 if formula == "P.A.S.A.":
-                    response = get_pasa_response(target_audience, product, action, mood, length)
+                    response = generate_story("PASA", target_audience, product, action, mood, length)
                 elif formula == "A.D.P.":
-                    response = get_adp_response(target_audience, product, action, mood, length)
+                    response = generate_story("ADP", target_audience, product, action, mood, length)
                 elif formula == "G.H.A.":
                     if story_topic:
-                        response = get_gha_response(target_audience, product, action, mood, length, story_topic)
+                        response = generate_story("GHA", 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 = ""
-                
+
                 if response:
                     st.subheader("Historia generada:")
                     st.write(response)