Update bullets/generator.py

bullets/generator.py

@@ -1,6 +1,6 @@
 import random
 
-def create_bullet_instruction(avatar_description=None, product_name=None, uploaded_content=None):
+def create_bullet_instruction(avatar_description=None, product_name=None, uploaded_content=None, target_audience=None):
     """
     Creates the instruction for generating benefit bullets.
     The model will randomly choose between different bullet formulas.
@@ -9,6 +9,7 @@ def create_bullet_instruction(avatar_description=None, product_name=None, upload
         avatar_description: Description of the target audience
         product_name: Name of the product or service
         uploaded_content: Content from uploaded files (if any)
+        target_audience: Specific target audience for the bullets
     
     Returns:
         str: The complete instruction for generating bullets