bullets/generator.py

import google.generativeai as genai
from .formulas import create_bullet_instruction

def generate_bullets(api_key, avatar_description, product_name, model_name="gemini-pro"):
    """

    Generates benefit bullets for a product/service based on the target audience.

    

    Args:

        api_key: Google API key for Gemini

        avatar_description: Description of the target audience

        product_name: Name of the product or service

        model_name: Name of the model to use

        

    Returns:

        str: Generated bullets

    """
    # Configure the API
    genai.configure(api_key=api_key)
    
    # Get the model
    model = genai.GenerativeModel(model_name)
    
    # Create the instruction
    instruction = create_bullet_instruction(avatar_description, product_name)
    
    # Generate the bullets
    response = model.generate_content(instruction)
    
    return response.text