1

ai_pyppeteer.py

@@ -26,7 +26,7 @@ class ToWork:
         # self.browser_token = 'd8ee6138f2b7ae5e064061e5c29d611a'
         # 公司账号
         self.browser_token = '6731ae7034426c1f861479eaeb964c61'
-        self.init_prompt = '/imagine'
+        self.init_prompt = '/imagine '
         self.browser_item = browser_item
         self.browser_id = browser_item['id']
         self.lock = lock
@@ -267,6 +267,7 @@ class ToWork:
                 else:
                     logger.info('文本元素不存在')
             else:
+                # 这个先不用
                 # 开始判断
                 element = '//ol[@data-list-id="chat-messages"]/li'
                 # 消息列表
@@ -285,8 +286,8 @@ class ToWork:
                             logger.info('未找到文本')
             await asyncio.sleep(1)
 
-    # 循环输入指令
-    async def enter_prompt(self, page, tasks):
+    # 旧的enter_prompt
+    async def old_input_init_prompt(self, page, tasks):
         # task是一个列表
         for task in tasks:
             # 开始判断
@@ -323,6 +324,37 @@ class ToWork:
                         # 输入完成,循环获取生成成功后的链接
                         await self.check_img_load_process(page, tasks[0])
 
+    # 循环输入指令
+    async def enter_prompt(self, page, tasks):
+        # task是一个列表
+        for task in tasks:
+            # 开始判断
+            element = '//ol[@data-list-id="chat-messages"]/li'
+            # 消息列表
+            message_list = await self.CheckElement(page, element, 20, 'list')
+            if message_list:
+                # 输入初始指令
+                logger.info('输入初始指令')
+                await self.driver_paste(page, self.init_prompt)
+                change_prompt = task['prompt']
+                # 如果有图片链接，添加
+                prompt_img = task['prompt_img']
+                if (task['type'] == 1 or task['type'] == '1') and prompt_img and len(json.loads(prompt_img)):
+                    logger.info('带图片任务')
+                    prompt_img = json.loads(prompt_img)
+                    st = ''
+                    for item in prompt_img:
+                        # 加入图片-链接
+                        st += f' {item}'
+                    change_prompt = st + ' ' + change_prompt
+                logger.info(change_prompt)
+                await self.driver_paste(page, change_prompt)
+                # 回车
+                await page.keyboard.press('Enter')
+                await asyncio.sleep(2)
+                # 输入完成,循环获取生成成功后的链接
+                await self.check_img_load_process(page, tasks[0])
+
     # 获取任务列表
     def get_current_tasks(self):
         if True: