import os
import sys

from browser_use.controller.service import Controller

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

import asyncio

from langchain_openai import ChatOpenAI

from browser_use import Agent
from browser_use.agent.views import AgentHistoryList

llm = ChatOpenAI(model='gpt-4o')

agent = Agent(
	task="go to google.com and type 'OpenAI' click search and give me the first url",
	llm=llm,
	controller=Controller(keep_open=False, headless=True),
)


async def main():
	history: AgentHistoryList = await agent.run()
	from pprint import pprint

	print('Final Result:')
	pprint(history.final_result(), indent=4)

	print('\nErrors:')
	pprint(history.get_errors(), indent=4)

	# e.g. xPaths the model clicked on
	print('\nModel Outputs:')
	pprint(history.all_model_outputs(), indent=4)


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