Update bullets/generator.py

bullets/generator.py

@@ -1,4 +1,11 @@
 def create_bullet_instruction():
+    """
+    Creates the instruction for generating benefit bullets.
+    The model will randomly choose between different bullet formulas.
+    
+    Returns:
+        str: The complete instruction for generating bullets
+    """
     # Base instruction that applies to all formulas
     base_instruction = """
     IMPORTANT: After creating the main offer, add a section with 5 powerful benefit bullets that reinforce the promise.
@@ -173,4 +180,10 @@ def create_bullet_instruction():
     
     • Los 5 movimientos efectivos para fortalecer tu core sin equipamiento, que activan un 78% más de fibras musculares que los ejercicios tradicionales.
     
-    Remember to choose just ONE formula and apply it consistently to all 5 bullets.
+    Remember to choose just ONE formula and apply it consistently to all 5 bullets.
+    """
+    
+    # Combine base instruction with formula instructions
+    complete_instruction = base_instruction + formula_instructions
+    
+    return complete_instruction