Update app.py

app.py

@@ -19,8 +19,8 @@ def get_gemini_response(input_prompt, formula_type, length, mood, target_audienc
     
     model = genai.GenerativeModel('gemini-2.0-flash')
     full_prompt = f"""
-    You are a creative storyteller. Write a {mood} story in Spanish about "{input_prompt}" using everyday words and natural conversation. 
-    Use exactly {length} words.
+    You are a creative storyteller. Write THREE different {mood} stories in Spanish about "{input_prompt}" using everyday words and natural conversation. 
+    Each story should use exactly {length} words.
 
     Target Audience: {target_audience}
     Call to Action: {cta_type}
@@ -41,8 +41,9 @@ def get_gemini_response(input_prompt, formula_type, length, mood, target_audienc
     - Write as a continuous narrative with natural paragraph breaks
     - The story should flow naturally between sections without any visible structure markers
     - DO NOT explain or mention the formula parts in the text
+    - Separate each story with "---"
     
-    Format: Clean story text with natural paragraphs, no labels, no explanations.
+    Format: Three clean story texts with natural paragraphs, separated by "---", no labels, no explanations.
     """
     response = model.generate_content([full_prompt])
     return response.parts[0].text if response and response.parts else "Error al generar contenido."
@@ -89,5 +90,8 @@ with col2:
             input_prompt, formula_type, length,
             mood, target_audience, cta_type
         )
-        st.subheader("Contenido generado:")
-        st.write(response)
+        stories = response.split("---")
+        for i, story in enumerate(stories, 1):
+            st.subheader(f"Historia {i}:")
+            st.write(story.strip())
+            st.divider()