import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import asyncio
from dotenv import load_dotenv
from browser_use import Agent, Controller
from browser_use.browser import BrowserProfile, BrowserSession
from browser_use.llm import ChatOpenAI
# 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.')
# Use the default controller with built-in Google Sheets actions
# The controller already includes all the necessary Google Sheets actions:
# - select_cell_or_range: Select specific cells or ranges (Ctrl+G navigation)
# - get_range_contents: Get contents of cells using clipboard
# - get_sheet_contents: Get entire sheet contents
# - clear_selected_range: Clear selected cells
# - input_selected_cell_text: Input text into selected cells
# - update_range_contents: Batch update ranges with TSV data
controller = Controller()
# For more Google Sheets keyboard shortcuts and automation ideas, see:
# - https://github.com/philc/sheetkeys/blob/master/content_scripts/sheet_actions.js
# - https://github.com/philc/sheetkeys/blob/master/content_scripts/commands.js
# - https://support.google.com/docs/answer/181110?hl=en&co=GENIE.Platform%3DDesktop#zippy=%2Cmac-shortcuts
# Tip: LLM is bad at spatial reasoning, don't make it navigate with arrow keys relative to current cell
# if given arrow keys, it will try to jump from G1 to A2 by pressing Down, without realizing needs to go Down+LeftLeftLeftLeft
async def main():
browser_session = BrowserSession(
browser_profile=BrowserProfile(
executable_path='/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
user_data_dir='~/.config/browseruse/profiles/default',
keep_alive=True,
),
)
await browser_session.start()
try:
model = ChatOpenAI(model='gpt-4.1')
# eraser = Agent(
# task="""
# Clear all the existing values in columns A through M in this Google Sheet:
# https://docs.google.com/spreadsheets/d/1INaIcfpYXlMRWO__de61SHFCaqt1lfHlcvtXZPItlpI/edit
# """,
# llm=model,
# browser_session=browser_session,
# controller=controller,
# )
# await eraser.run()
researcher = Agent(
task="""
Open this Google Sheet and read it to understand the structure: https://docs.google.com/spreadsheets/d/1INaIcfpYXlMRWO__de61SHFCaqt1lfHlcvtXZPItlpI/edit
Make sure column headers are present and all existing values in the sheet are formatted correctly.
Columns should be labeled using the top row of cells:
A: "Company Name"
B: "CEO Full Name"
C: "CEO Country of Birth"
D: "Source URL where the information was found"
Then Google to find the full name and nationality of each CEO of the top Fortune 100 companies and for each company,
append a row to this existing Google Sheet. You can do a few searches at a time,
but make sure to check the sheet for errors after inserting a new batch of rows.
At the end, double check the formatting and structure and fix any issues by updating/overwriting cells.
""",
llm=model,
browser_session=browser_session,
controller=controller,
)
await researcher.run()
finally:
await browser_session.stop()
# improvised_continuer = Agent(
# task="""
# Read the Google Sheet https://docs.google.com/spreadsheets/d/1INaIcfpYXlMRWO__de61SHFCaqt1lfHlcvtXZPItlpI/edit
# Add 3 more rows to the bottom continuing the existing pattern, make sure any data you add is sourced correctly.
# """,
# llm=model,
# browser_session=browser_session,
# controller=controller,
# )
# await improvised_continuer.run()
# final_fact_checker = Agent(
# task="""
# Read the Google Sheet https://docs.google.com/spreadsheets/d/1INaIcfpYXlMRWO__de61SHFCaqt1lfHlcvtXZPItlpI/edit
# Fact-check every entry, add a new column F with your findings for each row.
# Make sure to check the source URL for each row, and make sure the information is correct.
# """,
# llm=model,
# browser_session=browser_session,
# controller=controller,
# )
# await final_fact_checker.run()
if __name__ == '__main__':
asyncio.run(main())
