Update app.py

app.py

@@ -292,3 +292,72 @@ 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