Update app.py

app.py

@@ -184,6 +184,74 @@ def create_base_prompt(template_name, audience, product, offer=None, extra_param
     
     return prompt
 
+def update_section_context(section_name, content):
+    """Update the context for a section after it's generated."""
+    section_key = normalize_section_key(section_name)
+    
+    # Store in content dictionary
+    if 'content' not in st.session_state.sections:
+        st.session_state.sections['content'] = {}
+    st.session_state.sections['content'][section_key] = content
+    
+    # Store in context dictionary with timestamp
+    if 'context' not in st.session_state.sections:
+        st.session_state.sections['context'] = {}
+    
+    st.session_state.sections['context'][section_key] = {
+        'content': content,
+        'timestamp': datetime.datetime.now().isoformat()
+    }
+
+def get_relevant_context(current_section):
+    """Get relevant context based on current section"""
+    context = {}
+    
+    section_key = normalize_section_key(current_section)
+    
+    if section_key == 'above_the_fold':
+        return context
+    
+    specific_relationships = {
+        'problem': ['above_the_fold'],
+        'solution_&_benefits': ['problem'],
+        'authority': ['problem', 'solution_&_benefits'],
+        'offer_&_bonus': ['solution_&_benefits', 'problem'],
+        'social_proof': ['offer_&_bonus', 'solution_&_benefits'],
+        'offer_summary': ['offer_&_bonus', 'solution_&_benefits'],
+        'guarantees': ['offer_&_bonus', 'social_proof'],
+        'faq': ['offer_&_bonus', 'guarantees', 'problem'],
+        'closing': ['offer_&_bonus', 'guarantees', 'problem', 'solution_&_benefits']
+    }
+    
+    section_order = [
+        "above_the_fold",
+        "problem",
+        "solution_&_benefits",
+        "authority",
+        "offer_&_bonus",
+        "social_proof",
+        "offer_summary",
+        "guarantees",
+        "faq",
+        "closing"
+    ]
+    
+    # First, add specific relationships if they exist
+    if section_key in specific_relationships:
+        for rel_section in specific_relationships[section_key]:
+            if rel_section in st.session_state.sections['context']:
+                context[rel_section] = st.session_state.sections['context'][rel_section]['content']
+    
+    # Then, for any section after "above_the_fold", add previous sections as context
+    if section_key in section_order:
+        current_index = section_order.index(section_key)
+        if current_index > 0:
+            for prev_section in section_order[:current_index][-3:]:
+                if prev_section in st.session_state.sections['context'] and prev_section not in context:
+                    context[prev_section] = st.session_state.sections['context'][prev_section]['content']
+    
+    return context
+
 # Función auxiliar para normalizar nombres de secciones
 def normalize_section_key(section_name):
     """Convierte nombres de sección a formato de clave normalizado."""
@@ -291,73 +359,4 @@ if submit_sales_page:
 else:
     section_key = normalize_section_key(sales_page_section)
     if section_key in st.session_state.sections['content']:
-        display_generated_content(col2, st.session_state.sections['content'][section_key], "sales_page_section", sales_page_section)
-
-
-def update_section_context(section_name, content):
-    """Update the context for a section after it's generated."""
-    section_key = normalize_section_key(section_name)
-    
-    # Store in content dictionary
-    if 'content' not in st.session_state.sections:
-        st.session_state.sections['content'] = {}
-    st.session_state.sections['content'][section_key] = content
-    
-    # Store in context dictionary with timestamp
-    if 'context' not in st.session_state.sections:
-        st.session_state.sections['context'] = {}
-    
-    st.session_state.sections['context'][section_key] = {
-        'content': content,
-        'timestamp': datetime.datetime.now().isoformat()
-    }
-
-def get_relevant_context(current_section):
-    """Get relevant context based on current section"""
-    context = {}
-    
-    section_key = normalize_section_key(current_section)
-    
-    if section_key == 'above_the_fold':
-        return context
-    
-    specific_relationships = {
-        'problem': ['above_the_fold'],
-        'solution_&_benefits': ['problem'],
-        'authority': ['problem', 'solution_&_benefits'],
-        'offer_&_bonus': ['solution_&_benefits', 'problem'],
-        'social_proof': ['offer_&_bonus', 'solution_&_benefits'],
-        'offer_summary': ['offer_&_bonus', 'solution_&_benefits'],
-        'guarantees': ['offer_&_bonus', 'social_proof'],
-        'faq': ['offer_&_bonus', 'guarantees', 'problem'],
-        'closing': ['offer_&_bonus', 'guarantees', 'problem', 'solution_&_benefits']
-    }
-    
-    section_order = [
-        "above_the_fold",
-        "problem",
-        "solution_&_benefits",
-        "authority",
-        "offer_&_bonus",
-        "social_proof",
-        "offer_summary",
-        "guarantees",
-        "faq",
-        "closing"
-    ]
-    
-    # First, add specific relationships if they exist
-    if section_key in specific_relationships:
-        for rel_section in specific_relationships[section_key]:
-            if rel_section in st.session_state.sections['context']:
-                context[rel_section] = st.session_state.sections['context'][rel_section]['content']
-    
-    # Then, for any section after "above_the_fold", add previous sections as context
-    if section_key in section_order:
-        current_index = section_order.index(section_key)
-        if current_index > 0:
-            for prev_section in section_order[:current_index][-3:]:
-                if prev_section in st.session_state.sections['context'] and prev_section not in context:
-                    context[prev_section] = st.session_state.sections['context'][prev_section]['content']
-    
-    return context
+        display_generated_content(col2, st.session_state.sections['content'][section_key], "sales_page_section", sales_page_section)