import asyncio
import os
import sys

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

from dotenv import load_dotenv

load_dotenv()


from browser_use import Agent
from browser_use.browser import BrowserSession, BrowserProfile
from browser_use.llm import ChatGoogle

api_key = os.getenv('GOOGLE_API_KEY')

if not api_key:
	raise ValueError('GOOGLE_API_KEY is not set')

llm = ChatGoogle(model='gemini-2.0-flash', api_key=api_key)


async def main():
	# Create browser profile with settings
	browser_profile = BrowserProfile(
		headless=False,
		viewport={'width': 1502, 'height': 853},
		ignore_https_errors=True,
	)
	
	# Create browser session
	browser_session = BrowserSession(
		browser_profile=browser_profile,
	)
	
	# Start the browser
	await browser_session.start()

	agent = Agent(
		browser_session=browser_session,
		task='Go to https://browser-use.com/',
		llm=llm,
	)

		try:
			result = await agent.run()
			print(f'First task was {"successful" if result.is_successful else "not successful"}')

			if not result.is_successful:
				raise RuntimeError('Failed to navigate to the initial page.')

			agent.add_new_task('Navigate to the documentation page')

			result = await agent.run()
			print(f'Second task was {"successful" if result.is_successful else "not successful"}')

			if not result.is_successful:
				raise RuntimeError('Failed to navigate to the documentation page.')

			while True:
				next_task = input('Write your next task or leave empty to exit\n> ')

				if not next_task.strip():
					print('Exiting...')
					break

				agent.add_new_task(next_task)
				result = await agent.run()

				print(f"Task '{next_task}' was {'successful' if result.is_successful else 'not successful'}")

				if not result.is_successful:
					print('Failed to complete the task. Please try again.')
					continue

		finally:
			# Stop the browser session
			await browser_session.stop()


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