diff --git a/dsLightRag/JiMeng/Async_SubmitTask.py b/dsLightRag/JiMeng/Async_SubmitTask.py new file mode 100644 index 00000000..4203d113 --- /dev/null +++ b/dsLightRag/JiMeng/Async_SubmitTask.py @@ -0,0 +1,30 @@ +from __future__ import print_function +from JiMeng.Kit.JmCommon import JmCommon +from volcengine.visual.VisualService import VisualService + +if __name__ == '__main__': + visual_service = VisualService() + + visual_service.set_ak(JmCommon.ak) + visual_service.set_sk(JmCommon.sk) + + # 请求Body(查看接口文档请求参数-请求示例，将请求参数内容复制到此) + form = { + "req_key": "jimeng_high_aes_general_v21_L", + # "prompt": "少女，光影，瘦，白皙，干净，美丽", + "prompt": "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的工人，背着一个小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。", + "width": 1024, + "height": 1024, + "seed": -1, + "ddim_steps": 20, + "scale": 7.0, + "return_url": True, + "logo_info": { + "add_logo": True, + "position": 2, + "language": 0, + "opacity": 1 + } + } + resp = visual_service.cv_submit_task(form) + print(resp) \ No newline at end of file diff --git a/dsLightRag/JiMeng/Async_Txt2Img.py b/dsLightRag/JiMeng/Async_Txt2Img.py deleted file mode 100644 index f67b5909..00000000 --- a/dsLightRag/JiMeng/Async_Txt2Img.py +++ /dev/null @@ -1,16 +0,0 @@ -# uv pip install volcengine -from volcengine.visual.VisualService import VisualService - -from JiMeng.JmTxt2Img import JmTxt2Img -from JiMeng.Kit.JmCommon import JmCommon - -if __name__ == '__main__': - visual_service = VisualService() - - visual_service.set_ak(JmCommon.ak) - visual_service.set_sk(JmCommon.sk) - - prompt = "鸿门宴,室内场景图..." - req = {"req_key": JmTxt2Img.req_key, "prompt": prompt} - resp = visual_service.cv_submit_task(req) - print(resp) \ No newline at end of file diff --git a/dsLightRag/JiMeng/Async_text2img_xl_sft.py b/dsLightRag/JiMeng/Async_text2img_xl_sft.py new file mode 100644 index 00000000..1ed296a6 --- /dev/null +++ b/dsLightRag/JiMeng/Async_text2img_xl_sft.py @@ -0,0 +1,30 @@ +# uv pip install volcengine +from volcengine.visual.VisualService import VisualService +from JiMeng.Kit.JmCommon import JmCommon + +if __name__ == '__main__': + visual_service = VisualService() + + visual_service.set_ak(JmCommon.ak) + visual_service.set_sk(JmCommon.sk) + + # https://www.volcengine.com/docs/6791/1330203 + form = { + "req_key": "t2i_xl_sft", + # "prompt": "少女，光影，瘦，白皙，干净，美丽", + "prompt": "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的工人，背着一个小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。", + "width": 1024, + "height": 1024, + "seed": -1, + "ddim_steps": 20, + "scale": 7.0, + "return_url": True, + "logo_info": { + "add_logo": True, + "position": 2, + "language": 0, + "opacity": 1 + } + } + resp = visual_service.text2img_xl_sft(form) + print(resp) diff --git a/dsLightRag/JiMeng/Example/消防员与小女孩【错误案例】.jpg b/dsLightRag/JiMeng/Example/消防员与小女孩【错误案例】.jpg new file mode 100644 index 00000000..5a14e432 Binary files /dev/null and b/dsLightRag/JiMeng/Example/消防员与小女孩【错误案例】.jpg differ diff --git a/dsLightRag/JiMeng/JmTxt2Img.py b/dsLightRag/JiMeng/JmTxt2Img.py index b34fda3f..fcf4e6fe 100644 --- a/dsLightRag/JiMeng/JmTxt2Img.py +++ b/dsLightRag/JiMeng/JmTxt2Img.py @@ -52,101 +52,16 @@ class JmTxt2Img: print(f"保存图片失败: {str(e)}") raise Exception(f"保存图片失败: {str(e)}") - @staticmethod - # 新增一个方法用于查询任务状态 - @staticmethod - def query_task_status(task_id): - """查询任务状态""" - req = {"task_id": task_id} - response_body = JmCommon.do_request("POST", {}, json.dumps(req).encode('utf-8'), "CVProcessStatus") - jo = json.loads(response_body) - - code = jo.get("code") - if not JmErrorCode.is_success(code): - error_msg = JmErrorCode.get_message_by_code(code) - print(f"查询任务状态失败: 错误码={code}, 错误信息={error_msg}") - raise Exception(f"查询任务状态失败: {error_msg}") - - return jo.get("data", {}) - - # 修改generate_image方法，使其支持异步模式 - @staticmethod - def generate_image_async(prompt): - """异步生成图片，返回任务ID""" - req = {"req_key": JmTxt2Img.req_key, "prompt": prompt} - response_body = JmCommon.do_request("POST", {}, json.dumps(req).encode('utf-8'), JmTxt2Img.action) - jo = json.loads(response_body) - - code = jo.get("code") - if not JmErrorCode.is_success(code): - error_msg = JmErrorCode.get_message_by_code(code) - print(f"提交生成图片任务失败: 错误码={code}, 错误信息={error_msg}") - raise Exception(f"提交生成图片任务失败: {error_msg}") - - return jo.get("data", {}).get("task_id") - - # 修改main方法，实现异步轮询 - @staticmethod - def main(): - prompt = "鸿门宴,室内场景图..." # 详细提示词略 - save_image_path = os.path.abspath("../../../../../Text2Img.jpg") - print(f"保存图片路径：{save_image_path}") - - # 1. 提交生成任务 - task_id = JmTxt2Img.generate_image_async(prompt) - print(f"图片生成任务已提交，任务ID: {task_id}") - - # 2. 轮询任务状态 - max_polling_times = 60 # 最多轮询60次 - polling_interval = 5000 # 每5秒查询一次 - polling_count = 0 - - while polling_count < max_polling_times: - try: - polling_count += 1 - print(f"第{polling_count}次查询任务状态...") - task_status = JmTxt2Img.query_task_status(task_id) - - # 检查任务是否完成 - if task_status.get("status") == "completed": - print("图片生成完成！") - # 获取并保存图片 - img_base64 = task_status.get("binary_data_base64", [""])[0] - os.makedirs(os.path.dirname(save_image_path), exist_ok=True) - - if img_base64.startswith('data:image'): - img_base64 = img_base64.split(',')[1] - bytes_data = base64.b64decode(img_base64) - with open(save_image_path, 'wb') as f: - f.write(bytes_data) - print(f"文件保存成功！文件位置: {save_image_path}") - return - elif task_status.get("status") == "failed": - print(f"图片生成失败: {task_status.get('error_msg', '未知错误')}") - return - else: - print(f"任务进行中，当前状态: {task_status.get('status')}") - time.sleep(polling_interval / 1000) - except Exception as e: - print(f"查询任务状态异常: {str(e)}") - time.sleep(polling_interval / 1000) - - print(f"已达到最大轮询次数: {max_polling_times}") + @staticmethod def main(): """主函数，用于测试""" # 示例提示词 - prompt = "鸿门宴,室内场景图。非真人，是一幅画。远景镜头，让人一看就知道是在讲鸿门宴，" - prompt += "1、要亮！" - prompt += "2、不要有人出现" - prompt += "3、有大厅、墙面，桌子,不要有台阶，否则项庄没机会刺杀刘邦" - prompt += "4、西周时期特点" - prompt += "5、宏大，雄伟,颜色鲜明,不要灰色调的，不要有雾霾之类的" - prompt += "6、超高清画质" + prompt = "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的中国工人，背着一个中国小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。" # 保存图片路径 - 使用绝对路径避免相对路径问题 - save_image_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", "..", "..", "..", "..", "Text2Img.jpg")) + save_image_path = r"D:\Text2Img.jpg" print(f"保存图片路径：{save_image_path}") # 添加重试逻辑，处理API并发限制错误 diff --git a/dsLightRag/JiMeng/Kit/JmCommon.py b/dsLightRag/JiMeng/Kit/JmCommon.py index bc1a73f2..fac938af 100644 --- a/dsLightRag/JiMeng/Kit/JmCommon.py +++ b/dsLightRag/JiMeng/Kit/JmCommon.py @@ -100,8 +100,8 @@ class JmCommon: # 发送请求 try: - print(f"请求URL: {url}") - print(f"请求头: {headers}") + #print(f"请求URL: {url}") + #print(f"请求头: {headers}") response = requests.request( method=method, url=url, @@ -109,8 +109,8 @@ class JmCommon: data=body, timeout=(30, 30) # 连接超时和读取超时 ) - print(f"响应状态码: {response.status_code}") - print(f"响应内容: {response.text}") + #print(f"响应状态码: {response.status_code}") + #print(f"响应内容: {response.text}") response.raise_for_status() # 如果状态码不是200，抛出异常 return response.text except requests.exceptions.RequestException as e: diff --git a/dsLightRag/JiMeng/即梦文档.txt b/dsLightRag/JiMeng/即梦文档.txt index 82a98bfa..abc352b8 100644 --- a/dsLightRag/JiMeng/即梦文档.txt +++ b/dsLightRag/JiMeng/即梦文档.txt @@ -1,2 +1,5 @@ 即梦AI视频教程：（二）结合AI写提示词 -https://baijiahao.baidu.com/s?id=1832704271457492977&wfr=spider&for=pc \ No newline at end of file +https://baijiahao.baidu.com/s?id=1832704271457492977&wfr=spider&for=pc + +# 官方例子 +https://github.com/volcengine/volc-sdk-python/blob/main/volcengine/example/visual/cv_submit_task.py \ No newline at end of file