import os
import sys
import asyncio

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from langchain_openai import ChatOpenAI
from dotenv import load_dotenv

from browser_use import Agent, Browser

# Load environment variables
load_dotenv()
if not os.getenv('OPENAI_API_KEY'):
    raise ValueError('OPENAI_API_KEY is not set. Please add it to your environment variables.')

async def main():
    browser = Browser()
    async with await browser.new_context() as context:
        model = ChatOpenAI(model='gpt-4o')

        # Initialize browser agent
        agent1 = Agent(
            task='search iphone 17 news on google and pick 3 sources.',
            llm=model,
            browser_context=context,
        )
        agent2 = Agent(
            task='compare all the content found and summarize the key information about iPhone 17.',
            llm=model,
            browser_context=context,
        )
        await agent1.run()
        await agent2.run()

if __name__ == "__main__":
    asyncio.run(main())