import os
from phi.agent import Agent
from phi.model.openai import OpenAIChat

# Load environment variables (API keys, etc.)
from dotenv import load_dotenv
load_dotenv()

#####################################################################################
#                                BNI Membership Agent                               #
#####################################################################################

bni_agent = Agent(
    name="BNI Membership Agent",
    model=OpenAIChat(id="gpt-4o"),
    description="Provides insights on how BNI membership can help a business grow.",
    show_tool_calls=True,
    markdown=True,
)

def get_bni_benefits(company_data: str) -> str:
    """
    Generates a structured overview of how BNI membership can help the user's company.
    
    Args:
        company_data (str): User's company description, industry, and business details.

    Returns:
        str: Tailored insights on how BNI can benefit the company.
    """
    query = f"""
Given the following company details:
Business Overview: {company_data}

    Provide a short and to-the-point explanation of how BNI membership can benefit this specific company. Keep it concise, company-specific, and actionable with one sentence per point:
    
    Referral Generation – How can BNI help this company get more qualified referrals? (1-3 lines)
    Networking Opportunities – What industry-specific connections would be valuable? (2 line)
    Professional Development – What key skills or business knowledge will the company gain? (2 line)
    Supportive Community – How does BNI provide peer support relevant to this company? (2 line)
    Access to Resources – What exclusive mentorship, training, or resources would benefit them? (1 line)
    Revenue Growth Potential – How can BNI help scale their business? (1 line)
    """
    response = bni_agent.run(query)
    return response.content  # Returns structured insights on BNI membership benefits