from agency_swarm import Agency
from agency_swarm import set_openai_key
from ExpertDataAnalyst import ExpertDataAnalyst
from SalesManagerCEO import SalesManagerCEO
from dotenv import load_dotenv
import os

load_dotenv('.env')
API_KEY = os.getenv('OPENAI_API_KEY')
set_openai_key(API_KEY)

salesManagerCEO = SalesManagerCEO()
expertDataAnalyst = ExpertDataAnalyst()

agency = Agency([salesManagerCEO, [salesManagerCEO, expertDataAnalyst]],
                shared_instructions='./agency_manifesto.md')

if __name__ == '__main__':
    agency.demo_gradio()