'commit'

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,33 +55,13 @@ class JmTxt2Img:
 
     @staticmethod
     def main():
-        """主函数，用于测试"""
         # 示例提示词
         prompt = "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的中国工人，背着一个中国小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。"
 
-        # 保存图片路径 - 使用绝对路径避免相对路径问题
+        # 保存图片路径
         save_image_path =  r"D:\Text2Img.jpg"
-        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  # 达到最大重试次数，抛出异常
+        print(f"保存图片路径：{save_image_path}")
 
 
 if __name__ == "__main__":