index.js

import dotenv from 'dotenv';
import {chromium} from 'playwright';
import prompt from 'prompt';
// eslint-disable-next-line no-unused-vars
import colors from '@colors/colors';
import {Command} from 'commander';

import {ChatOpenAI} from 'langchain/chat_models/openai';
import {doActionWithAutoGPT} from './autogpt/index.js';
import {interactWithPage} from './actions/index.js';
import {createTestFile, gracefulExit, logPageScreenshot} from './util/index.js';

dotenv.config();

async function main(options) {
  const url = options.url;
  const browser = await chromium.launch({headless: options.headless});

  // Parse the viewport option
  const [width, height] = options.viewport.split(',').map(Number);

  const browserContext = await browser.newContext({
    viewport: {width, height},
  });

  const page = await browserContext.newPage();
  await page.goto(url);

  prompt.message = 'BrowserGPT'.green;
  const promptOptions = [];
  if (options.autogpt) {
    promptOptions.push('+AutoGPT');
  }
  if (options.headless) {
    promptOptions.push('+headless');
  }
  if (promptOptions.length > 0) {
    prompt.message += ` (${promptOptions.join(' ')})`.green;
  }
  prompt.delimiter = '>'.green;

  prompt.start();

  const chatApi = new ChatOpenAI({
    temperature: 0.1,
    modelName: options.model ? options.model : 'gpt-4-1106-preview',
  });

  if (options.outputFilePath) {
    createTestFile(options.outputFilePath);
  }

  process.on('exit', () => {
    gracefulExit(options);
  });

  // eslint-disable-next-line no-constant-condition
  while (true) {
    const {task} = await prompt.get({
      properties: {
        task: {
          message: ' Input a task\n',
          required: false,
        },
      },
    });

    if (task === '') {
      console.log('Please input a task or press CTRL+C to exit'.red);
    } else {
      try {
        if (options.autogpt) {
          await doActionWithAutoGPT(page, chatApi, task, options);
        } else {
          await interactWithPage(chatApi, page, task, options);
        }
        if (options.headless) {
          await logPageScreenshot(page);
        }
      } catch (e) {
        console.log('Execution failed');
        console.log(e);
      }
    }
  }
}

const program = new Command();

program
  .option('-a, --autogpt', 'run with autogpt', false)
  .option('-m, --model <model>', 'openai model to use', 'gpt-4-1106-preview')
  .option('-o, --outputFilePath <outputFilePath>', 'path to store test code')
  .option('-u, --url <url>', 'url to start on', 'https://www.google.com')
  .option('-v, --viewport <viewport>', 'viewport size to use', '1280,720')
  .option('-h, --headless', 'run in headless mode', false);

program.parse();

main(program.opts());