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

api_key = os.getenv("OPENAI_API_KEY")
set_openai_key(api_key)

salesManagerCEO = SalesManagerCEO()
expertDataAnalyst = ExpertDataAnalyst()

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

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