import asyncio
from autogen_ext.models import OpenAIChatCompletionClient
from autogen_ext.agents import MultimodalWebSurfer

async def main() -> None:
    model_client = OpenAIChatCompletionClient(model="gpt-4o-2024-08-06")


    web_surfer = MultimodalWebSurfer(
        name="web_surfer",
        model_client = model_client,
    )

    await web_surfer.init(
        to_save_screenshots = True,
    )

    while True:
        # Run the team and stream messages
        user_input = await asyncio.get_event_loop().run_in_executor(None, input, ">: ")
        response = await web_surfer.run(user_input)
        print(response)

asyncio.run(main())