Update app.py

app.py

@@ -10,6 +10,11 @@ import docx
 from PIL import Image
 import io
 
+# Import the bullet generator
+from bullets.generator import create_bullet_instruction
+# Import the bonus generator
+from bonuses.generator import create_bonus_instruction
+
 # Set page to wide mode to use full width
 st.set_page_config(layout="wide")
 
@@ -214,6 +219,24 @@ with col2:
                 """
                 instruction += creative_hook_instruction
             
+            # Add instruction for generating benefit bullets based on the promise
+            bullet_content = None
+            if hasattr(st.session_state, 'file_content') and st.session_state.input_type in ["file", "both"]:
+                bullet_content = st.session_state.file_content
+                
+            instruction += create_bullet_instruction(
+                avatar_description=avatar_description,
+                product_name=product_name,
+                uploaded_content=bullet_content
+            )
+            
+            # Add instruction for generating bonuses that complement the offer
+            instruction += create_bonus_instruction(
+                avatar_description=avatar_description,
+                product_name=product_name,
+                selected_formula_name=st.session_state.formula_type
+            )
+            
             # Add additional context based on input type
             if st.session_state.input_type == "manual":
                 additional_context = f"""