diff --git a/dsLightRag/JiMeng/Example/工人与小女孩.mp4 b/dsLightRag/JiMeng/Example/工人与小女孩.mp4 new file mode 100644 index 00000000..6941cbf0 Binary files /dev/null and b/dsLightRag/JiMeng/Example/工人与小女孩.mp4 differ diff --git a/dsLightRag/JiMeng/Example/消防员与小女孩【错误案例】.jpg b/dsLightRag/JiMeng/Example/工人与小女孩【错误案例】.jpg similarity index 100% rename from dsLightRag/JiMeng/Example/消防员与小女孩【错误案例】.jpg rename to dsLightRag/JiMeng/Example/工人与小女孩【错误案例】.jpg diff --git a/dsLightRag/JiMeng/T1_JmTxt2Img.py b/dsLightRag/JiMeng/T1_JmTxt2Img.py index fcf4e6fe..ad4aee77 100644 --- a/dsLightRag/JiMeng/T1_JmTxt2Img.py +++ b/dsLightRag/JiMeng/T1_JmTxt2Img.py @@ -1,7 +1,6 @@ +import base64 import json import os -import time -import base64 from JiMeng.Kit.JmCommon import JmCommon from JiMeng.Kit.JmErrorCode import JmErrorCode @@ -56,34 +55,14 @@ class JmTxt2Img: @staticmethod def main(): - """主函数,用于测试""" # 示例提示词 prompt = "雨天街头,一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的中国工人,背着一个中国小女孩。工人脖子上侧挂一个小熊玩偶包,拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣,戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木,地面潮湿,氛围温馨。" - # 保存图片路径 - 使用绝对路径避免相对路径问题 + # 保存图片路径 save_image_path = r"D:\Text2Img.jpg" + JmTxt2Img.generate_image(prompt, save_image_path) print(f"保存图片路径:{save_image_path}") - # 添加重试逻辑,处理API并发限制错误 - retry_count = 0 - max_retries = 5 # 减少重试次数以便更快排查问题 - retry_interval = 5000 # 重试间隔(毫秒) - - while True: - try: - JmTxt2Img.generate_image(prompt, save_image_path) - # 成功生成图片,跳出循环 - break - except Exception as e: - print(f"生成图片异常: {str(e)}") - retry_count += 1 - if retry_count < max_retries: - print(f"等待{retry_interval}毫秒后重试...") - time.sleep(retry_interval / 1000) - else: - print(f"已达到最大重试次数: {max_retries}") - raise e # 达到最大重试次数,抛出异常 - if __name__ == "__main__": JmTxt2Img.main() \ No newline at end of file