Update bullets/generator.py

bullets/generator.py

@@ -1,3 +1,5 @@
+import random
+
 def create_bullet_instruction():
     """
     Creates the instruction for generating benefit bullets.
@@ -186,4 +188,68 @@ def create_bullet_instruction():
     # Combine base instruction with formula instructions
     complete_instruction = base_instruction + formula_instructions
     
+    return complete_instruction
+
+
+def get_random_bullet_formula():
+    """
+    Randomly selects a bullet formula type to ensure variety in generated bullets.
+    Extracts formula names automatically from the instruction text.
+    
+    Returns:
+        str: The name of the randomly selected formula
+    """
+    # Get the full instruction text
+    full_instruction = create_bullet_instruction()
+    
+    # Extract formula names using regex pattern matching
+    import re
+    
+    # Pattern to find formula names (looks for "FORMULA X - NAME:")
+    formula_pattern = r"FORMULA\s+\d+\s+-\s+([^:]+):"
+    
+    # Find all matches in the instruction text
+    formula_matches = re.findall(formula_pattern, full_instruction)
+    
+    # If no formulas found, fallback to manual list
+    if not formula_matches:
+        formulas = [
+            "STANDARD BENEFIT",
+            "3 EN 1 (FEATURE + BENEFIT + MEANING)",
+            "ANTI-PROCRASTINACIÓN (ACTION + RESULT + TIME)",
+            "NÚMERICA SUPREMA"
+        ]
+    else:
+        formulas = formula_matches
+    
+    # Select a random formula
+    selected_formula = random.choice(formulas)
+    
+    return selected_formula
+
+
+def create_bullet_instruction_with_formula():
+    """
+    Creates the instruction for generating benefit bullets with a specific
+    randomly selected formula to ensure consistency.
+    
+    Returns:
+        str: The complete instruction for generating bullets with the selected formula
+    """
+    # Get base instruction
+    base_instruction = create_bullet_instruction()
+    
+    # Get a random formula
+    selected_formula = get_random_bullet_formula()
+    
+    # Add specific instruction to use the selected formula
+    formula_directive = f"""
+    
+    IMPORTANT OVERRIDE: For this specific task, you MUST use FORMULA {selected_formula} 
+    for ALL 5 bullets. Do not choose randomly - you must use this exact formula consistently.
+    """
+    
+    # Combine instructions
+    complete_instruction = base_instruction + formula_directive
+    
     return complete_instruction