import asyncio
from browser_use import Agent, BrowserSession, ChatDeepSeek
import configparser
import os

script_dir = os.path.dirname(__file__)
config_path = os.path.join(script_dir, 'config.ini')
config = configparser.ConfigParser()
config.read(config_path)

API_KEY = config['BROWSER_USE']['BROWSER_USE_API_KEY']
AGENT_EMAIL = config['AGENT_ACCOUNT']['AGENT_EMAIL']
AGENT_PWD = config['AGENT_ACCOUNT']['AGENT_PWD']
AGENT_CONTRACT = config['AGENT_ACCOUNT']['AGENT_CONTRACT']
CUSTOMER_EMAIL = config['CUSTOMER_ACCOUNT']['ODOO_EMAIL']
CUSTOMER_PWD = config['CUSTOMER_ACCOUNT']['ODOO_PWD']
CUSTOMER_NAME = config['CUSTOMER_ACCOUNT']['ODOO_NAME']

async def main():
    browser_session = BrowserSession(headless=False)

    agent = Agent(
        task=f"""1.访问http://localhost:8070
            2.点击电子邮件框后输入{AGENT_EMAIL}，点击密码框后输入{AGENT_PWD}，点击登录按钮
            3.在侧边栏菜单中找到并点击"配置"菜单
            4.找到合约菜单下的运输服务，点击新建一个运输服务，名称设置为tests，服务类型选择FedEx Ground®
            5.单击成本合约，当下拉框成功展开后，点击"搜索更多..."，点击编号为{AGENT_CONTRACT}的合约
            6.开始时间选择今天，结束时间选择明年的今天，点击保存，若保存失败，则再次执行5
            7.点击附加费通用设置，当附加费通用设置弹窗显示后，单击加价类型，下拉框成功展开后单击选择“加价+”，点击附加费框，清除内容后输入“1”，点击确认
            8.点击更新附加费
            9.找到合约菜单下的合约管理，点击新建一个合约，等待两秒后点击客户框，下拉框成功显示后单击{CUSTOMER_NAME}
            10.点击运输服务框，下拉框成功显示后点击“搜索更多...”，单击运输服务tests
            11.点击发货地址，下拉框成功显示后选择第一个地址，勾选启用后保存
            12.点击右上角头像，退出登录后，在登录页点击电子邮件框输入{CUSTOMER_EMAIL}，密码框输入{CUSTOMER_PWD}，点击登录""",
        llm=ChatDeepSeek(api_key=API_KEY),
        browser_session=browser_session,
    )
    await agent.run()

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