From 56d32bc9f44dd07e16d9745c01c739848158c4ef Mon Sep 17 00:00:00 2001
From: MixBadGun <1059129006@qq.com>
Date: Fri, 24 Oct 2025 23:25:00 +0800
Subject: [PATCH] BA

---
 konabot/plugins/xibao_generate/__init__.py | 38 ++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/konabot/plugins/xibao_generate/__init__.py b/konabot/plugins/xibao_generate/__init__.py
index f574a5e..2a5ceff 100644
--- a/konabot/plugins/xibao_generate/__init__.py
+++ b/konabot/plugins/xibao_generate/__init__.py
@@ -75,6 +75,44 @@ async def _(msg: UniMsg, event: BaseEvent, content: Optional[str] = ""):
         other_function=lambda page: beibao_continue_handle(page, content),
         timeout=30
     )
+    await evt.send(
+        await UniMessage().image(raw=screenshot).export()
+    )
+
+async def continue_handle_3(page: Page, arg1: str, arg2: str) -> None:
+    # 这里可以添加一些预处理逻辑
+    # 找到 id 为 textL 的 input，id 为 textR 的 input
+    input1 = await page.query_selector("#textL")
+    input2 = await page.query_selector("#textR")
+    if input1:
+        await input1.fill(arg1)
+    if input2:
+        await input2.fill(arg2)
+    # 等待 0.3 秒钟
+    await page.wait_for_timeout(300)
+    # 等待 id 为 loading 的元素不可见
+    loading = await page.query_selector("#loading")
+    if loading:
+        await loading.wait_for_element_state("hidden")
+
+evt = on_alconna(
+    Alconna(
+        f"BA生成",
+        Args["arg1", str],
+        Args["arg2", str]
+    ),
+    use_cmd_start=True,
+    use_cmd_sep=False,
+    skip_for_unmatch=True,
+)
+@evt.handle()
+async def _(msg: UniMsg, event: BaseEvent, arg1: str, arg2: str):
+    screenshot = await WebRenderer.render(
+        url="https://tmp.nulla.top/ba-logo/",
+        target="#canvas",
+        other_function=lambda page: continue_handle_3(page, arg1, arg2),
+        timeout=30
+    )
     await evt.send(
         await UniMessage().image(raw=screenshot).export()
     )
\ No newline at end of file