import os

from autogen import UserProxyAgent, config_list_from_json
from autogen.agentchat.contrib.web_surfer import WebSurferAgent
from autogen.browser_utils import (
    BingMarkdownSearch,
    PlaywrightMarkdownBrowser,
    RequestsMarkdownBrowser,
    SeleniumMarkdownBrowser,
)


def main():
    # Load LLM inference endpoints from an env variable or a file
    # See https://microsoft.github.io/autogen/docs/FAQ#set-your-api-endpoints
    # and OAI_CONFIG_LIST_sample.
    # For example, if you have created a OAI_CONFIG_LIST file in the current working directory, that file will be used.
    config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST")

    browser = RequestsMarkdownBrowser(
        # PlaywrightMarkdownBrowser(
        viewport_size=1024 * 3,
        downloads_folder=os.getcwd(),
        search_engine=BingMarkdownSearch(bing_api_key=os.environ["BING_API_KEY"]),
        # launch_args={"channel": "msedge", "headless": False},
    )

    web_surfer = WebSurferAgent(
        "web_surfer",
        llm_config={"config_list": config_list},
        summarizer_llm_config={"config_list": config_list},
        is_termination_msg=lambda x: x.get("content", "").rstrip().find("TERMINATE") >= 0,
        code_execution_config=False,
        browser=browser,
    )

    # Create the agent that represents the user in the conversation.
    user_proxy = UserProxyAgent("user", code_execution_config=False)

    # Let the assistant start the conversation.  It will end when the user types exit.
    web_surfer.initiate_chat(user_proxy, message="How can I help you today?")


if __name__ == "__main__":
    main()